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_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)])
