tests/test_vocabular.py

"""Тест поиска терминов с учётом падежей"""

import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).parent.parent))

from vocabulary.parse_vocabulary import VOCABULARY_MANAGER

TESTS_CASES = {
    'Газель газель Газель газели Газелей': set(('ГАЗель', )),
    'Икаруса-280 Икарус-260 Икарусом-260 Икарусов-280': set(('Икарус-260', 'Икарус-280')),
    'Купили ещё автобусов': set(),
    'Икарус-2600': set(),
    'Иккарус-260': set(),
    'Ока': set(('Ока', )),
    'Окой': set(('Ока', )),
    'Окружная дорога': set(),
    }

for text, terms in TESTS_CASES.items():
    assert VOCABULARY_MANAGER.find_terms(text) == set((el.lower() for el in terms)), breakpoint()