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)])
def test_drake_data4(drain_setup):
steps = [
Step(a=1, inputs=[Step(b=1, target=True)]),
Step(a=2, inputs=[Step(b=1, target=True)])
]
data = get_drake_data(steps)
assert data == {
Step(b=1): set(),
steps[0]: {Step(b=1)},
steps[1]: {Step(b=1)}
}
def test_drakefile(drain_setup):
inputs = [Step(b=1)]
inputs[0].target = True
steps = [Step(a=1, inputs=inputs), Step(a=2, inputs=inputs)]
print(to_drakefile(steps, preview=True))
def test_drake_data3(drain_setup):
inputs = [Step(b=1)]
inputs[0].target = True
step = Step(a=1, inputs=inputs)
data = get_drake_data([step])
assert data == {Step(b=1): set(), step: {Step(b=1)}}
def test_inputs_target(drain_setup):
assert get_inputs(Step(value=1, inputs=[Step(value=2)]),
target=True) == set()
def test_drake_data2(drain_setup):
step = Step(a=1, inputs=[Step(b=1)])
data = get_drake_data([step])
assert data == {step: set()}
def test_drake_data(drain_setup):
step = Step(a=1)
data = get_drake_data([step])
assert data == {Step(a=1): set()}
def test_input_targets2(drain_setup):
assert get_input_targets(
Step(value=1,
target=True,
inputs=[Step(value=2, target=True),
Step(value=3)])) == set([Step(value=2)])
def test_drakefile(drain_setup):
steps = [
Step(a=1, inputs=[Step(b=1, target=True)]),
Step(a=2, inputs=[Step(b=1, target=True)])
]
print to_drakefile(steps, preview=True)
def test_input_targets(drain_setup):
assert get_input_targets(Step(value=1, inputs=[Step(value=2)])) == set()
