def test_walker_is_complete(self):
tree_walker = TreeWalker()
self.assertTrue(tree_walker.is_complete(verbose=True))
tree_walker.functions = dict((x, tree_walker.walk_skip)
for x in op.ALL_TYPES if x is not op.AND)
self.assertFalse(tree_walker.is_complete())
def test_undefined_node(self):
varA = Symbol("At", INT)
dag_walker = DagWalker()
with self.assertRaises(UnsupportedOperatorError):
dag_walker.walk(varA)
tree_walker = TreeWalker()
with self.assertRaises(UnsupportedOperatorError):
tree_walker.walk(varA)
def test_undefined_node(self):
varA = Symbol("At", INT)
dag_walker = DagWalker()
with self.assertRaises(NotImplementedError):
dag_walker.walk(varA)
tree_walker = TreeWalker()
with self.assertRaises(NotImplementedError):
tree_walker.walk(varA)
def test_walker_is_complete(self):
op.ALL_TYPES.append(-1)
with self.assertRaises(AssertionError):
TreeWalker()
op.ALL_TYPES.remove(-1)
