def test_create_data_for_gantt_plotly(): """test_create_data_for_gantt_plotly.""" c1 = Component("c1") c1.state_record_list = [ BaseComponentState.WORKING, BaseComponentState.FINISHED, BaseComponentState.FINISHED, BaseComponentState.FINISHED, BaseComponentState.FINISHED, ] product = Product([c1]) init_datetime = datetime.datetime(2020, 4, 1, 8, 0, 0) timedelta = datetime.timedelta(days=1) product.create_data_for_gantt_plotly(init_datetime, timedelta)
def test_plot_simple_gantt(tmpdir): """test_plot_simple_gantt.""" c1 = Component("c1") c1.state_record_list = [ BaseComponentState.WORKING, BaseComponentState.FINISHED, BaseComponentState.FINISHED, BaseComponentState.FINISHED, BaseComponentState.FINISHED, ] init_datetime = datetime.datetime(2020, 4, 1, 8, 0, 0) timedelta = datetime.timedelta(days=1) c1.create_data_for_gantt_plotly(init_datetime, timedelta) product = Product([c1]) product.plot_simple_gantt( save_fig_path=os.path.join(str(tmpdir), "test.png"))