def testExpandContexts(self): Brain.init(TEST_BRAIN_FILE, order=2) brain = Brain(TEST_BRAIN_FILE) tokens = [ "this", Brain.SPACE_TOKEN_ID, "is", Brain.SPACE_TOKEN_ID, "a", Brain.SPACE_TOKEN_ID, "test" ] self.assertEqual(list(brain._to_edges(tokens)), [((1, 1), False), ((1, "this"), False), (("this", "is"), True), (("is", "a"), True), (("a", "test"), True), (("test", 1), False), ((1, 1), False)]) tokens = ["this", "is", "a", "test"] self.assertEqual(list(brain._to_edges(tokens)), [((1, 1), False), ((1, "this"), False), (("this", "is"), False), (("is", "a"), False), (("a", "test"), False), (("test", 1), False), ((1, 1), False)])
def testExpandGraph(self): Brain.init(TEST_BRAIN_FILE, order=2) brain = Brain(TEST_BRAIN_FILE) tokens = ["this", Brain.SPACE_TOKEN_ID, "is", Brain.SPACE_TOKEN_ID, "a", Brain.SPACE_TOKEN_ID, "test"] self.assertEquals(list(brain._to_graph(brain._to_edges(tokens))), [((1, 1), False, (1, "this")), ((1, "this"), True, ("this", "is")), (("this", "is"), True, ("is", "a")), (("is", "a"), True, ("a", "test")), (("a", "test"), False, ("test", 1)), (("test", 1), False, (1, 1))])