def test_func_node_return_value(): # no outputs def func(): pass n = FuncNode(func) assert n() == () # function explicitly returns None def func(a, b): return None n = FuncNode(func) assert n((1, 2)) == (None, ) # single value output def func(): return 1 n = FuncNode(func) assert n() == (1, ) def func(): tup = (1, 2) return tup n = FuncNode(func) assert n() == ((1, 2), ) # multi outputs def func(): return 'a', 1 n = FuncNode(func) assert n() == ('a', 1)
def test_func_node_output_names_are_unique(): def func(): a = 1 return a, a, 2 n = FuncNode(func) assert tuple(n.outputs()) == ("a1", "a2", "res")
def test_func_node_output_names_are_unique(): def func(): a = 1 return a, a, 2 n = FuncNode(func) assert tuple(n.outputs()) == ('a1', 'a2', 'res')
def test_func_node_no_args_or_kwds(): def func1(a, *args): return None assert_raises(TypeError, lambda: FuncNode(func1)) def func2(**kwds): return None assert_raises(TypeError, lambda: FuncNode(func2))
def test_func_node_return_none_differ_from_no_return(): def func(): return None n = FuncNode(func) assert tuple(n.outputs()) == ("None",) def func(): print None n = FuncNode(func) assert tuple(n.outputs()) == ()
def test_func_node_args_name_detection(): def func(): return None n = FuncNode(func) assert tuple(n.inputs()) == () def func(a, b): return None n = FuncNode(func) assert tuple(n.inputs()) == ("a", "b")
def test_func_node_return_type_detection(): def func(a): return a n = FuncNode(func) assert n.output("a").type is None def func(a): return a func.__rettypes__ = ["IInt"] n = FuncNode(func) assert n.output("a").type == "IInt" @rettype("IInt") def func(a): return a n = FuncNode(func) assert n.output("a").type == "IInt" @rettype("IInt", "IFloat") def func(a, b): return a, b n = FuncNode(func) assert n.output("a").type == "IInt" assert n.output("b").type == "IFloat"
def test_func_node_args_type_detection(): def func(a): return a n = FuncNode(func) assert n.input("a").type is None def func(a): return a func.__argtypes__ = ["IInt"] n = FuncNode(func) assert n.input("a").type == "IInt" @argtype("IInt") def func(a): return a n = FuncNode(func) assert n.input("a").type == "IInt" @argtype("IInt", "IFloat") def func(a, b): return a + b n = FuncNode(func) assert n.input("a").type == "IInt" assert n.input("b").type == "IFloat"
def test_evaluation_do_not_reevaluate_same_node(): visited = [] def func(): visited.append("yes") pg = PortGraph() n = FuncNode(func) pg.add_actor(n, 0) pg.add_actor(n, 1) env = EvaluationEnvironment() algo = BruteEvaluation(pg) ws = WorkflowState(pg) algo.eval(env, ws, 0) assert len(visited) == 1 algo.eval(env, ws, 1) assert len(visited) == 2 algo.eval(env, ws, 0) assert len(visited) == 2 env.new_execution() algo.eval(env, ws, 0) assert len(visited) == 3
def test_func_node_inline(): def tata(a, b): c = a + b return c n = FuncNode(tata) assert n((1, 2)) == (3, )
def test_evaluation_propagated_upstream(): visited = [] def func(txt): visited.append(txt) return txt pg = PortGraph() n = FuncNode(func) pg.add_actor(n, 0) pg.add_actor(n, 1) pg.connect(pg.out_port(0, 'txt'), pg.in_port(1, 'txt')) algo = BruteEvaluation(pg) ws = WorkflowState(pg) env = EvaluationEnvironment() ws.store_param(pg.in_port(0, 'txt'), "txt", 0) algo.eval(env, ws, 0) assert len(visited) == 1 algo.eval(env, ws, 1) assert len(visited) == 2 env.new_execution() algo.eval(env, ws, 1) assert len(visited) == 4
def test_evaluation_needs_ready_to_evaluate_state(): def func(a, b): c = a + b return c pg = PortGraph() n = FuncNode(func) pg.add_actor(n, 0) algo = BruteEvaluation(pg) ws = WorkflowState(pg) assert_raises(EvaluationError, lambda: algo.eval(None, ws))
def test_func_node_return_outputs(): def func(): return 1 n = FuncNode(func) assert tuple(n.outputs()) == ('res', ) def func(): a = 1 return a n = FuncNode(func) assert tuple(n.outputs()) == ('a', ) def func(): a = 1 return a, 1 n = FuncNode(func) assert tuple(n.outputs()) == ('a', 'res')
def test_func_node_args_name_detection(): def func(): return None n = FuncNode(func) assert tuple(n.inputs()) == () def func(a, b): return None n = FuncNode(func) assert tuple(n.inputs()) == ('a', 'b')
def test_func_node_return_none_differ_from_no_return(): def func(): return None n = FuncNode(func) assert tuple(n.outputs()) == ('None', ) def func(): print None n = FuncNode(func) assert tuple(n.outputs()) == ()
def test_lazy_evaluate_at_least_once_each_node(): evaluated = [] def func(): evaluated.append('bla') pg = PortGraph() vid = pg.add_actor(FuncNode(func)) assert pg.actor(vid).is_lazy() algo = LazyEvaluation(pg) env = EvaluationEnvironment() ws = WorkflowState(pg) algo.eval(env, ws) assert len(evaluated) == 1 algo.eval(env, ws) assert len(evaluated) == 1
def test_evaluation_eval_all_nodes(): visited = [] def func(): visited.append("yes") pg = PortGraph() n = FuncNode(func) pg.add_actor(n, 0) pg.add_actor(n, 1) algo = BruteEvaluation(pg) env = EvaluationEnvironment() ws = WorkflowState(pg) algo.eval(env, ws) assert len(visited) == 2
def test_evaluation_clear(): def func(): pass pg = PortGraph() n = FuncNode(func) pg.add_actor(n, 0) pg.add_actor(n, 1) algo = BruteEvaluation(pg) env = EvaluationEnvironment() ws = WorkflowState(pg) algo.eval(env, ws) assert not algo.requires_evaluation(env, ws) ws.clear() assert algo.requires_evaluation(env, ws)
def test_lazy_do_not_reevaluate_node_if_same_execution(): evaluated = [] def func(txt): evaluated.append(txt) return txt pg = PortGraph() vid = pg.add_actor(FuncNode(func)) assert pg.actor(vid).is_lazy() algo = LazyEvaluation(pg) env = EvaluationEnvironment() ws = WorkflowState(pg) ws.store_param(pg.in_port(vid, 'txt'), 'toto', env.current_execution()) algo.eval(env, ws) assert len(evaluated) == 1 algo.eval_node(env, ws, vid) assert len(evaluated) == 1
def test_lazy_always_reevaluate_non_lazy_nodes(): evaluated = [] def func(txt): evaluated.append(txt) return txt pg = PortGraph() vid = pg.add_actor(FuncNode(func)) pg.actor(vid).set_lazy(False) algo = LazyEvaluation(pg) env = EvaluationEnvironment() ws = WorkflowState(pg) ws.store_param(pg.in_port(vid, 'txt'), 'toto', env.current_execution()) algo.eval(env, ws) assert len(evaluated) == 1 env.new_execution() algo.eval(env, ws) assert len(evaluated) == 2
def test_func_node_return_outputs(): def func(): return 1 n = FuncNode(func) assert tuple(n.outputs()) == ("res",) def func(): a = 1 return a n = FuncNode(func) assert tuple(n.outputs()) == ("a",) def func(): a = 1 return a, 1 n = FuncNode(func) assert tuple(n.outputs()) == ("a", "res")
def test_func_node_return_type_detection(): def func(a): return a n = FuncNode(func) assert n.output('a').type is None def func(a): return a func.__rettypes__ = ["IInt"] n = FuncNode(func) assert n.output('a').type == "IInt" @rettype("IInt") def func(a): return a n = FuncNode(func) assert n.output('a').type == "IInt" @rettype("IInt", "IFloat") def func(a, b): return a, b n = FuncNode(func) assert n.output('a').type == "IInt" assert n.output('b').type == "IFloat"
def test_func_node_args_type_detection(): def func(a): return a n = FuncNode(func) assert n.input('a').type is None def func(a): return a func.__argtypes__ = ["IInt"] n = FuncNode(func) assert n.input('a').type == "IInt" @argtype("IInt") def func(a): return a n = FuncNode(func) assert n.input('a').type == "IInt" @argtype("IInt", "IFloat") def func(a, b): return a + b n = FuncNode(func) assert n.input('a').type == "IInt" assert n.input('b').type == "IFloat"
def test_evaluation_affect_output_to_right_ports(): def func(a, b): c = a + b d = a * 2 return c, d # simple order pg = PortGraph() n = FuncNode(func) pg.add_vertex(0) pg.add_in_port(0, 'a', 0) pg.add_in_port(0, 'b', 1) pg.add_out_port(0, 'c', 2) pg.add_out_port(0, 'd', 3) pg.set_actor(0, n) algo = BruteEvaluation(pg) env = EvaluationEnvironment() ws = WorkflowState(pg) ws.store_param(0, 'a', 0) ws.store_param(1, 'b', 0) algo.eval(env, ws) assert ws.get(2) == 'ab' assert ws.get(3) == 'aa' # reverse input orders pg = PortGraph() n = FuncNode(func) pg.add_vertex(0) pg.add_in_port(0, 'b', 0) pg.add_in_port(0, 'a', 1) pg.add_out_port(0, 'c', 2) pg.add_out_port(0, 'd', 3) pg.set_actor(0, n) algo = BruteEvaluation(pg) env = EvaluationEnvironment() ws = WorkflowState(pg) ws.store_param(0, 'a', 0) ws.store_param(1, 'b', 0) algo.eval(env, ws) assert ws.get(2) == 'ba' assert ws.get(3) == 'bb' # reverse output order pg = PortGraph() n = FuncNode(func) pg.add_vertex(0) pg.add_in_port(0, 'a', 0) pg.add_in_port(0, 'b', 1) pg.add_out_port(0, 'd', 2) pg.add_out_port(0, 'c', 3) pg.set_actor(0, n) algo = BruteEvaluation(pg) env = EvaluationEnvironment() ws = WorkflowState(pg) ws.store_param(0, 'a', 0) ws.store_param(1, 'b', 0) algo.eval(env, ws) assert ws.get(3) == 'ab' assert ws.get(2) == 'aa'