def test_visitor(self): visited = [] visitor = lambda n: visited.append(n.name) n = Node("LOL") n.visit(visitor) self.assertIn(n.name, visited) self.assertEqual(len(visited), 1)
def test_init(self): n = Node("LOL") self.assertIsInstance(n, Node) self.assertEqual(n.name, "LOL") self.assertEqual(str(n), "<PythonNode LOL>") n.name = "MEGALOL" self.assertEqual(n.name, "MEGALOL") self.assertEqual(str(n), "<PythonNode MEGALOL>")