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