def test_output(): step_code = 'c=a+b' m = Model('sum') m.inputs_info = [InputObj('a'), InputObj('b')] m.outputs_info = [OutputObj('c')] m.step_code = step_code assert m.run(a=1, b=2) == 3
def test_steps(): step_code = 'a += 10' m = Model('sum') m.inputs_info = [InputObj('a=0')] m.outputs_info = [OutputObj('a')] m.step_code = step_code assert m.run() == 10 assert m.run(nstep=10) == 100