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