def test_entityList(self):
context = ConvertData.createContext()
Beatscript.addEntityList(context, "person§Karl, person§Hugo,\tobject§Table,place§Home,\t")
self.assertTrue("Table" in context["Entities"] and context["Entities"]["Table"].type == Config.OBJECT)
self.assertTrue("Karl" in context["Entities"] and context["Entities"]["Karl"].type == Config.PERSON)
self.assertTrue("Hugo" in context["Entities"] and context["Entities"]["Hugo"].type == Config.PERSON)
self.assertTrue("Home" in context["Entities"] and context["Entities"]["Home"].type == Config.PLACE)
self.assertFalse("Peter" in context["Entities"])
self.assertTrue(len(context["Entities"]) == 4)
def test_initialContext(self):
context = ConvertData.createContext()
Beatscript.addInitialContext(context, "person§Peter, person§Hugo,\tobject§Table,place§Home, person§Lord Harthorne,")
Beatscript.addEntityList(context, "person§Karl, person§Hugo,\tobject§Table,place§Home, ")
self.assertTrue(context["Entities"]["Table"] in context["KnownEntities"])
self.assertTrue(context["Entities"]["Hugo"] in context["KnownEntities"])
self.assertTrue(context["Entities"]["Home"] in context["KnownEntities"])
self.assertFalse(context["Entities"]["Karl"] in context["KnownEntities"])
self.assertTrue(len(context["KnownEntities"]) == 5)
for entity in context["KnownEntities"]:
if entity.name not in context["Entities"]:
self.assertTrue(False)
for entityName in context["Entities"]:
if context["Entities"][entityName] not in context["KnownEntities"]:
self.assertTrue(entityName=="Karl" and context["Entities"][entityName].type == Config.PERSON)
