def test_viz_saves_graph_object_if_filename(self):
import graphviz
f = Flow(name="test")
f.add_task(Task(name="a_nice_task"))
with tempfile.NamedTemporaryFile() as tmp:
graph = f.visualize(filename=tmp.name)
with open(tmp.name, "r") as f:
contents = f.read()
assert "label=a_nice_task" in contents
assert "shape=ellipse" in contents
def test_to_environment_file_writes_data(self):
f = Flow(name="test")
with tempfile.NamedTemporaryFile() as tmp:
f.to_environment_file(tmp.name)
with open(tmp.name, "r") as f:
assert f.read()
