From f1c4f5ad41cfa9da5ed5c6b7471aad8c228a6c68 Mon Sep 17 00:00:00 2001
From: Aleksey Filippov <al.filippov@ulstu.ru>
Date: Fri, 2 Jun 2023 17:48:33 +0400
Subject: [PATCH] Add test data

---
 main.py | 60 ++++++++++++++++++++++++++++++++++++++++++++++-----------
 1 file changed, 49 insertions(+), 11 deletions(-)

diff --git a/main.py b/main.py
index 67d7538..5e98963 100644
--- a/main.py
+++ b/main.py
@@ -3,21 +3,59 @@
 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 = [
+    ['получить кредит обучение'],
+    ['получить кредит образование'],
+    ['купить товар рассрочка'],
+    ['кредит ставка ипотека'],
+    ['самая низкий ставка кредиту'],
+    ['оформить кредит'],
+    ['ставка кредит'],
+    ['ипотека кредит ставка', 'какой ставка'],
+    ['получить деньга карта', 'заблокировать'],
+    ['снять деньга карта'],
+    ['какой баланс карта'],
+    ['лимит карта'],
+    ['проходить оплата', 'быть деньга'],
+    ['деньга карта'],
+    ['деньга счёт'],
+    ['разблокировать карта'],
+    ['заблокировать карта'],
+    ['какой условие ипотека'],
+    ['какой условие кредит жильё'],
+    ['какой срок кредит жильё'],
+    ['какой срок ипотека'],
+    ['потерять карта'],
+    ['застрять банкомат'],
+    ['украсть карта'],
+    ['какой остаток карта'],
+    ['купить дом'],
+    ['купить дом'],
+    ['купить товар кредит'],
+    ['срок ипотека'],
+    ['купить дом кредит'],
+    ['снизить ставка кредиту'],
+    ['рефинансировать кредит'],
+    ['снизить ставка кредиту'],
+    ['ли друг банка банк, перевести банк'],
+    ['провести рефинансирование кредит'],
+    ['рефинансировать ипотека']
+]
 
 
 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)
-    print(f'Extracted terms:\n{", ".join(terms)}')
-    result: str = MyOntology().get_event_description(terms)
-    print(f'Result:\n{result}')
+    # 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)
+        print(f'Result:\n{result}')
+        print()
 
 
 if __name__ == '__main__':