Example #1
0
 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)
Example #2
0
 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>")