tests/test_filter.py

import sys
import os
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from knowledge_base import KnowledgeBase

def test_itmo_query_filter():
    kb = KnowledgeBase()
    
    test_cases = [
        ('Какие дисциплины по NLP в 1 семестре программы ИИ?', True),
        ('Расскажи о программе AI Product', True),
        ('Сколько кредитов за курс машинного обучения?', True),
        ('Какая погода в Санкт-Петербурге?', False),
        ('Как приготовить борщ?', False),
        ('Расскажи о программе ИИ в ITMO', True),
        ('Какие курсы по глубокому обучению?', True),
        ('Как добраться до метро?', False),
        ('Учебный план магистратуры', True),
        ('Дисциплины по компьютерному зрению', True)
    ]
    
    print('Тестирование фильтра релевантности...')
    
    for query, expected in test_cases:
        result = kb.is_itmo_query(query)
        status = '✓' if result == expected else '✗'
        print(f'{status} "{query}" -> {result} (ожидалось {expected})')
    
    print('\nТест завершен')

if __name__ == '__main__':
    test_itmo_query_filter()