def test_graph(): """Construct a small graph. Note that this graph is strictly speaking nonsensical, because it doesn't use any actual primitive operations. """ g = Graph() x = Parameter(g) assert x.value is PARAMETER one = Constant(1) add = Constant('add') return_ = Constant('return') value = Apply([add, x, one], g) return_ = Apply([return_, value], g) g.return_ = return_ g.parameters.append(x)
def test_str_coverage(): """Just a coverage test for __str__ and __repr__ Doesn't check that they take particular values since that could change easily. """ g = Graph() p = Parameter(g) p.name = 'param' objects = [g, Apply([], g), p, Parameter(g), Constant(0), Constant(g)] for o in objects: str(o) repr(o) o.debug.debug_name
def test_graph(): """Construct a small graph. Note that this graph is strictly speaking nonsensical, because it doesn't use any actual primitive operations. """ g = Graph() x = Parameter(g) assert x.value is PARAMETER one = Constant(1) add = Constant("add") return_ = Constant("return") value = Apply([add, x, one], g) return_ = Apply([return_, value], g) g.return_ = return_ g.parameters.append(x) assert g.abstract is None g.parameters[0].abstract = 123 assert g.abstract is None g.parameters[0].abstract = None g.return_.abstract = 456 assert g.abstract is None g.parameters[0].abstract = 123 assert g.abstract == AbstractFunctionUnique([123], 456)
def test_incoming(): in0 = Constant(0) value = Apply([in0], Graph()) assert list(value.incoming) == [in0] assert list(in0.incoming) == []