File size: 1,459 Bytes
2fc8dc5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
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()
|