def test_remove_function(self):
code = "def foo(a, b):\n" " x, y = a, b\n" " return x, y\n" "c, d = foo(1, 2)\n" "e, f = foo(3, 4)"
block = Block(code)
info = find_functions(block.ast)
foo1_func = FunctionCall.from_ast(block.sub_blocks[-1].ast, info)
foo2_func = FunctionCall.from_ast(block.sub_blocks[-2].ast, info)
assert foo1_func != foo2_func
model = ExecutionModel()
model.add_function(foo1_func)
model.add_function(foo2_func)
model.remove_function(foo2_func)
assert not (foo2_func in model.statements)
assert foo1_func in model.statements
def test_import_and_rename(self):
code = "from blockcanvas.debug.my_operator import add as add1\n" \
"a = add1(1,2)\n" \
"b = add1(a,3)"
foo_block = Block(code)
info = find_functions(foo_block.ast)
foo_call = FunctionCall.from_ast(foo_block.sub_blocks[1].ast, info)
desired = 'result = add(1, 2)'
self.assertEqual(foo_call.call_signature, desired)
def test_import_and_rename(self):
code = "from blockcanvas.debug.my_operator import add as add1\n" \
"a = add1(1,2)\n" \
"b = add1(a,3)"
foo_block = Block(code)
info = find_functions(foo_block.ast)
foo_call = FunctionCall.from_ast(foo_block.sub_blocks[1].ast, info)
desired = 'result = add(1, 2)'
self.assertEqual(foo_call.call_signature, desired)
def collect_func_call(statement_info,info):
functions = []
for names, node in statement_info.call_funcs:
function = FunctionCall.from_ast(node, info)
# update the output bindings if we can, else just leave empty
if len(function.outputs) == len(names):
for output, name in zip(function.outputs, names):
output.binding = name
functions.append(function)
return functions
def test_local_def(self):
code = "def foo(a):\n" \
" b = a\n" \
" return b\n" \
"y = foo(2)\n"
foo_block = Block(code)
info = find_functions(foo_block.ast)
foo_call = FunctionCall.from_ast(foo_block.sub_blocks[-1].ast, info)
desired = 'b = foo(2)'
self.assertEqual(foo_call.call_signature, desired)
def test_local_def(self):
code = "def foo(a):\n" \
" b = a\n" \
" return b\n" \
"y = foo(2)\n"
foo_block = Block(code)
info = find_functions(foo_block.ast)
foo_call = FunctionCall.from_ast(foo_block.sub_blocks[-1].ast, info)
desired = 'b = foo(2)'
self.assertEqual(foo_call.call_signature, desired)
def test_remove_function(self):
code = "def foo(a, b):\n" \
" x, y = a, b\n" \
" return x, y\n" \
"c, d = foo(1, 2)\n" \
"e, f = foo(3, 4)"
block = Block(code)
info = find_functions(block.ast)
foo1_func = FunctionCall.from_ast(block.sub_blocks[-1].ast, info)
foo2_func = FunctionCall.from_ast(block.sub_blocks[-2].ast, info)
assert foo1_func != foo2_func
model = ExecutionModel()
model.add_function(foo1_func)
model.add_function(foo2_func)
model.remove_function(foo2_func)
assert not (foo2_func in model.statements)
assert foo1_func in model.statements
def test_import(self):
code = "from blockcanvas.debug.my_operator import add, mul\n" \
"c = add(a,b)\n" \
"d = mul(c, 2)\n" \
"e = mul(c, 3)\n" \
"f = add(d,e)"
foo_block = Block(code)
info = find_functions(foo_block.ast)
foo_call = FunctionCall.from_ast(foo_block.sub_blocks[1].ast, info)
desired = 'result = add(a, b)'
self.assertEqual(foo_call.call_signature, desired)
def test_import(self):
code = "from blockcanvas.debug.my_operator import add, mul\n" \
"c = add(a,b)\n" \
"d = mul(c, 2)\n" \
"e = mul(c, 3)\n" \
"f = add(d,e)"
foo_block = Block(code)
info = find_functions(foo_block.ast)
foo_call = FunctionCall.from_ast(foo_block.sub_blocks[1].ast, info)
desired = 'result = add(a, b)'
self.assertEqual(foo_call.call_signature, desired)
def test_add_function(self):
code = "def foo(a, b):\n" " x, y = a, b\n" " return x, y\n" "c, d = foo(1, 2)"
block = Block(code)
info = find_functions(block.ast)
func = FunctionCall.from_ast(block.sub_blocks[-1].ast, info)
model = ExecutionModel()
model.add_function(func)
assert len(model.sorted_statements) == 1
desired = "\ndef foo(a, b): \n x, y = (a, b)\n return x, y\n\nx_, y_ = foo(1, 2)"
self.assertEqual(desired, model.code)
def test_add_function(self):
code = "def foo(a, b):\n" \
" x, y = a, b\n" \
" return x, y\n" \
"c, d = foo(1, 2)"
block = Block(code)
info = find_functions(block.ast)
func = FunctionCall.from_ast(block.sub_blocks[-1].ast, info)
model = ExecutionModel()
model.add_function(func)
assert len(model.sorted_statements) == 1
desired = '\ndef foo(a, b): \n x, y = (a, b)\n return x, y\n\nx_, y_ = foo(1, 2)'
self.assertEqual(desired, model.code)
