def test_inputs_target2(drain_setup):
inputs = [Step(value=2), Step(value=3)]
inputs[0].target = True
step = Step(value=1)
step.inputs = inputs
step.target = True
assert get_inputs(step, target=True) == set([Step(value=2)])
def test_inputs_no_target(drain_setup):
inputs = [Step(value=2), Step(value=3, inputs=[Step(value=4)])]
inputs[0].target = True
step = Step(value=1, inputs=inputs)
step.target = True
assert get_inputs(step, target=False) ==\
set([Step(value=3, inputs=[Step(value=4)]), Step(value=4)])
