Update test data

main.py

@@ -3,61 +3,63 @@
 import sys
 
 from src.myontology import MyOntology
+from src.nlp import NLP
+from src.parse_tree.parse_tree import ParseTree
+from src.syntax import Syntax
 
-data = [
-    ['получить кредит обучение'],
-    ['получить кредит образование'],
-    ['купить товар рассрочка'],
-    ['кредит ставка ипотека'],
-    ['самая низкий ставка кредиту'],
-    ['оформить кредит'],
-    ['ставка кредит'],
-    ['ипотека кредит ставка', 'какой ставка'],
-    ['получить деньга карта', 'заблокировать'],
-    ['снять деньга карта'],
-    ['какой баланс карта'],
-    ['лимит карта'],
-    ['проходить оплата', 'быть деньга'],
-    ['деньга карта'],
-    ['деньга счёт'],
-    ['разблокировать карта'],
-    ['заблокировать карта'],
-    ['какой условие ипотека'],
-    ['какой условие кредит жильё'],
-    ['какой срок кредит жильё'],
-    ['какой срок ипотека'],
-    ['потерять карта'],
-    ['застрять банкомат'],
-    ['украсть карта'],
-    ['какой остаток карта'],
-    ['купить дом'],
-    ['купить дом'],
-    ['купить товар кредит'],
-    ['срок ипотека'],
-    ['купить дом кредит'],
-    ['снизить ставка кредиту'],
-    ['рефинансировать кредит'],
-    ['снизить ставка кредиту'],
-    ['ли друг банка банк, перевести банк'],
-    ['провести рефинансирование кредит'],
-    ['рефинансировать ипотека']
+test_data = [
+    'как получить кредит на обучение',
+    'Как получить кредит на образование?',
+    'Можно ли купить товар в рассрочку?',
+    'какая ставка кредита на ипотеку?',
+    'Самая низкая ставка по кредиту?',
+    'Как оформить кредит?',
+    'Ставка кредита',
+    'Какая ставка кредита на ипотеку?',
+    'Как получить деньги с заблокированной карты?',
+    'Как снять деньги с заблокированной карты',
+    'Какой баланс карты',
+    'Какой лимит карты',
+    'Почему не проходит оплата, хотя деньги есть',
+    'Сколько денег на карте',
+    'Сколько денег на счете',
+    'Как разблокировать карту?',
+    'Как заблокировать карту?',
+    'Какие условия по ипотеке?',
+    'Какие условия кредита на жилье',
+    'Какой срок кредита на жилье',
+    'Какой срок ипотеки',
+    'Я потеряла карту. Что делать?',
+    'Карта застряла в банкомате. Что делать?',
+    'У меня украли карту. Что делать?',
+    'Какой остаток по карте?',
+    'Хочу купить дом',
+    'Хочу купить дом . Что делать?',
+    'Можно ли купить товар в кредит?',
+    'Какой срок по ипотеке',
+    'Хочу купить дом в кредит Что делать?',
+    'Можно ли снизить ставку по кредиту?',
+    'Можно ли рефинансировать кредит?',
+    'Можно ли снизить ставку по кредиту другого банка?',
+    'У меня кредит в другом банке. Можно ли его перевести в ваш банк?',
+    'Можно провести рефинансирование кредита?',
+    'Можно ли рефинансировать ипотеку?'
 ]
 
 
 def _main(wav_file: str):
     # text: str = Speech().run_recognition(wav_file)
     # text: str = 'Как получить деньги с заблокированной карты?'
-    # print(f'Text: {text}')
-    # parse_tree: ParseTree = Syntax().get_parse_tree(text)
-    # print(f'Parse tree:\n{parse_tree}')
-    # terms = NLP().get_terms(parse_tree)
-    for terms in data:
-        print(f'Extracted terms:\n{", ".join(terms)}')
-        result: str = MyOntology().get_event_description(terms)
+    for text in test_data:
+        print(f'Text: {text}')
+        parse_tree: ParseTree = Syntax().get_parse_tree(text)
+        print(f'Parse tree:\n{parse_tree}')
+        terms = NLP().get_terms(parse_tree)
+        print(f'Extracted terms:\n\t{", ".join(terms)}')
+        result: str = MyOntology().get_events(terms)
         print(f'Result:\n{result}')
         print()
 
-
 if __name__ == '__main__':
     if len(sys.argv) < 2:
         print(f'Usage: {sys.argv[0]} FILE')