plik do testów

Test_nlu.py

@@ -0,0 +1,36 @@
+import jsgf
+
+book_grammar = jsgf.parse_grammar_file('book.jsgf')
+book_grammar
+
+utterance = 'chciałbym zarezerwować stolik na jutro na godzinę dwunastą trzydzieści na cztery osoby'
+matched = book_grammar.find_matching_rules(utterance)
+matched
+
+def get_dialog_act(rule):
+    slots = []
+    get_slots(rule.expansion, slots)
+    return {'act': rule.grammar.name, 'slots': slots}
+
+def get_slots(expansion, slots):
+    if expansion.tag != '':
+        slots.append((expansion.tag, expansion.current_match))
+        return
+
+    for child in expansion.children:
+        get_slots(child, slots)
+
+    if not expansion.children and isinstance(expansion, jsgf.NamedRuleRef):
+        get_slots(expansion.referenced_rule.expansion, slots)
+
+get_dialog_act(matched[0])
+
+def nlu(utterance):
+    matched = book_grammar.find_matching_rules(utterance)
+
+    if matched:
+        return get_dialog_act(matched[0])
+    else:
+        return {'act': 'null', 'slots': []}
+
+nlu('chciałbym zarezerwować stolik na jutro na godzinę dziesiątą dla trzech osób')