lang-model/naive-nlu/tree_nlu/tests/gac_extension.py

27 lines
719 B
Python

from ..knowledge_base import KnowledgeBase
from ..session.org_mode import global_session as session
from . import gac_100
def ask_then_learn_test(knowledge: KnowledgeBase):
with session().log("is icecream blue?"):
ret, _, _ = knowledge.process("is icecream blue?")
assert(ret is False)
with session().log("icecream is blue"):
ret, _, _ = knowledge.process("icecream is blue")
with session().log("is icecream blue?"):
ret, _, _ = knowledge.process("is icecream blue?")
assert(ret is True)
return knowledge
def main():
knowledge = gac_100.main()
knowledge.knowledge['blue'] = {'groups': {'property'}}
knowledge = ask_then_learn_test(knowledge)