def test_workflow_flush() -> None:
"""
Trivial setup.
"""
def mock_postproc(w) -> None:
w.output = 10
def mock_preproc(w) -> None:
w.input = 20
workflow = Workflow(preprocessors=[mock_preproc],
postprocessors=[mock_postproc])
# initially the value of workflow.input is = 2
workflow.run(2)
# once the run method is complete, the value of input and output is modified by the
# mock_postproc and mock_preproc functions.
assert workflow.input == 20, "workflow.get_input() should be 20."
assert workflow.output == 10, "workflow.get_output() should be 10."
workflow.flush()
assert workflow.input == None, "workflow.get_input() should be None."
assert workflow.output == None, "workflow.get_output() should be None."
def test_safe_flush():
workflow = Workflow()
i, o = workflow.flush()
assert i == {}
assert o == {}