def test_clinker_dups_inner(): # Testing that duplicates are allowed inside the graph x, y, z = inputs() e = add(mul(y, y), add(x, z)) lnk = CLinker().accept(FunctionGraph([x, y, z], [e])) fn = lnk.make_function() assert fn(1.0, 2.0, 3.0) == 8.0
def test_clinker_not_used_inputs(): # Testing that unused inputs are allowed. x, y, z = inputs() e = add(x, y) lnk = CLinker().accept(FunctionGraph([x, y, z], [e])) fn = lnk.make_function() assert fn(2.0, 1.5, 1.0) == 3.5
def test_clinker_dups(): # Testing that duplicate inputs are allowed. x, y, z = inputs() e = add(x, x) lnk = CLinker().accept(FunctionGraph([x, x], [e])) fn = lnk.make_function() assert fn(2.0, 2.0) == 4
def test_clinker_literal_inlining(): x, y, z = inputs() z = Constant(tdouble, 4.12345678) e = add(mul(add(x, y), div(x, y)), bad_sub(bad_sub(x, y), z)) lnk = CLinker().accept(FunctionGraph([x, y], [e])) fn = lnk.make_function() assert abs(fn(2.0, 2.0) + 0.12345678) < 1e-9 code = lnk.code_gen() # print "=== Code generated ===" # print code assert "4.12345678" in code # we expect the number to be inlined
def test_clinker_single_node(): x, y, z = inputs() node = add.make_node(x, y) lnk = CLinker().accept(FunctionGraph(node.inputs, node.outputs)) fn = lnk.make_function() assert fn(2.0, 7.0) == 9
def test_clinker_straightforward(): x, y, z = inputs() e = add(mul(add(x, y), div(x, y)), bad_sub(bad_sub(x, y), z)) lnk = CLinker().accept(FunctionGraph([x, y, z], [e])) fn = lnk.make_function() assert fn(2.0, 2.0, 2.0) == 2.0