def test_create_data_for_gantt_plotly(): """test_create_data_for_gantt_plotly.""" c = BaseComponent("c") c.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) c.create_data_for_gantt_plotly(init_datetime, timedelta)
def test_create_data_for_gantt_plotly(): c = BaseComponent("c") task1 = BaseTask("task1") task2 = BaseTask("task2") c.extend_targeted_task_list([task1, task2]) # Set test case (start time = 0, finish time = 5) task1.start_time_list = [0, 2] task1.ready_time_list = [0, 2] task1.finish_time_list = [3, 5] task2.start_time_list = [1] task2.ready_time_list = [2] task2.finish_time_list = [5] init_datetime = datetime.datetime(2020, 4, 1, 8, 0, 0) # timedelta = 1day timedelta = datetime.timedelta(days=1) df = c.create_data_for_gantt_plotly(init_datetime, timedelta) assert df[0]["Start"] == (init_datetime + 0 * timedelta).strftime( "%Y-%m-%d %H:%M:%S" ) assert df[0]["Finish"] == (init_datetime + (5 + 1.0) * timedelta).strftime( "%Y-%m-%d %H:%M:%S" ) assert df[0]["Type"] == "Component"