def test_simple_folding_non_const_inputs(self):
"""Test that nodes with non constant inputs cannot fold to become
constant.
"""
with DeltaGraph() as graph:
n1 = add_const(4, forward_non_const(3))
n2 = add_non_const(4, forward_non_const(2))
# nodes producing constants are turned to constant nodes
for node in graph.find_node_by_name('node'):
self.assertIsInstance(node.body, PyConstBody)
# this one is a non-constant one now
self.assertIsInstance(n1.body, PyFuncBody)
# this one is still a non-constant
self.assertIsInstance(n2.body, PyFuncBody)
self.assertTrue(graph.check())
def test_forked_folding_one_non_const(self):
"""Check the case with a forked return.
One forked output goes to a non-const node ->
the rest of destinations stay const.
"""
with DeltaGraph() as graph:
nums = return_1_2_const()
p1 = forward_const(nums.x)
p2 = forward_non_const(nums.y)
self.assertIsInstance(nums.body, PyConstBody)
self.assertIsInstance(p1.body, PyConstBody)
self.assertIsInstance(p2.body, PyFuncBody)
self.assertTrue(graph.check())
def test_simple_folding_chain_splitting(self):
"""Test that if an output of a constant node can be split between
a constant and non-constan nodes, i.e. the source stays constant and
only one of the destinations is constant.
"""
with DeltaGraph() as graph:
n0 = return_2_const()
n1 = forward_const(n0)
n2 = forward_const(n1)
n3 = forward_non_const(n1)
self.assertIsInstance(n0.body, PyConstBody)
self.assertIsInstance(n1.body, PyConstBody)
self.assertIsInstance(n2.body, PyConstBody)
self.assertIsInstance(n3.body, PyFuncBody)
self.assertTrue(graph.check())
def test_forked_folding_one_non_const_splitting(self):
"""Check the case with a forked return.
Same as above but with splitting ->
no funny business.
"""
with DeltaGraph():
nums = return_1_2_const()
p1 = forward_const(nums.x)
p2 = forward_non_const(nums.y)
p3 = forward_const(nums.x)
p4 = forward_const(nums.y)
self.assertIsInstance(nums.body, PyConstBody)
self.assertIsInstance(p1.body, PyConstBody)
self.assertIsInstance(p2.body, PyFuncBody)
self.assertIsInstance(p3.body, PyConstBody)
self.assertIsInstance(p4.body, PyConstBody)
def setUp(self):
s = dl.lib.StateSaver(int, condition=lambda x: x == 10)
gen = dl.lib.make_generator(range(11))
with dl.DeltaGraph() as self.test_graph:
s.save_and_exit_if(forward_non_const(gen.call()))