def test_draw_plotly_network():
task0 = BaseTask("auto", auto_task=True)
task1 = BaseTask("task1")
task1.start_time_list = [1]
task1.ready_time_list = [0]
task1.finish_time_list = [3]
task2 = BaseTask("task2")
task2.start_time_list = [4]
task2.ready_time_list = [4]
task2.finish_time_list = [6]
task2.append_input_task(task1)
w = BaseWorkflow([task1, task2, task0])
w.draw_plotly_network(save_fig_path="test.png")
if os.path.exists("test.png"):
os.remove("test.png")
def test_draw_plotly_network(tmpdir):
"""test_draw_plotly_network."""
task0 = BaseTask("auto", auto_task=True)
task1 = BaseTask("task1")
task1.state_record_list = [
BaseTaskState.READY,
BaseTaskState.READY,
BaseTaskState.WORKING,
BaseTaskState.FINISHED,
BaseTaskState.FINISHED,
]
task2 = BaseTask("task2")
task2.state_record_list = [
BaseTaskState.READY,
BaseTaskState.READY,
BaseTaskState.WORKING,
BaseTaskState.FINISHED,
BaseTaskState.FINISHED,
]
task2.append_input_task(task1)
w = BaseWorkflow([task1, task2, task0])
for ext in ["png", "html", "json"]:
save_fig_path = os.path.join(str(tmpdir), "test." + ext)
w.draw_plotly_network(save_fig_path=save_fig_path)