def test_create_data_for_cost_history_plotly():
factory = BaseFactory("factory")
w1 = BaseFacility("w1", cost_per_time=10.0)
w1.cost_list = [0, 0, 10, 10, 0, 10]
w2 = BaseFacility("w2", cost_per_time=5.0)
w2.cost_list = [5, 5, 0, 0, 5, 5]
factory.facility_list = [w1, w2]
factory.cost_list = list(map(sum, zip(w1.cost_list, w2.cost_list)))
init_datetime = datetime.datetime(2020, 4, 1, 8, 0, 0)
timedelta = datetime.timedelta(days=1)
data = factory.create_data_for_cost_history_plotly(init_datetime,
timedelta)
x = [(init_datetime + time * timedelta).strftime("%Y-%m-%d %H:%M:%S")
for time in range(len(factory.cost_list))]
# w1
assert data[0].name == w1.name
assert data[0].x == tuple(x)
assert data[0].y == tuple(w1.cost_list)
# w2
assert data[1].name == w2.name
assert data[1].x == tuple(x)
assert data[1].y == tuple(w2.cost_list)
def test_add_labor_cost():
factory = BaseFactory("factory")
w1 = BaseFacility("w1", cost_per_time=10.0)
w2 = BaseFacility("w2", cost_per_time=5.0)
factory.facility_list = [w2, w1]
w1.state = BaseFacilityState.WORKING
w2.state = BaseFacilityState.FREE
factory.add_labor_cost()
assert w1.cost_list == [10.0]
assert w2.cost_list == [0.0]
assert factory.cost_list == [10.0]
factory.add_labor_cost(only_working=False)
assert factory.cost_list == [10.0, 15.0]
assert w1.cost_list == [10.0, 10.0]
assert w2.cost_list == [0.0, 5.0]
def test_create_gantt_plotly():
factory = BaseFactory("factory")
w1 = BaseFacility("w1", cost_per_time=10.0)
w1.start_time_list = [0, 5]
w1.finish_time_list = [2, 8]
w2 = BaseFacility("w2", cost_per_time=5.0)
w2.start_time_list = [9]
w2.finish_time_list = [11]
factory.facility_list = [w1, w2]
init_datetime = datetime.datetime(2020, 4, 1, 8, 0, 0)
timedelta = datetime.timedelta(days=1)
factory.create_gantt_plotly(init_datetime,
timedelta,
save_fig_path="test.png")
if os.path.exists("test.png"):
os.remove("test.png")
def test_initialize():
factory = BaseFactory("factory")
factory.cost_list = [9.0, 7.2]
w = BaseFacility("w1")
factory.facility_list = [w]
w.state = BaseFacilityState.WORKING
w.cost_list = [9.0, 7.2]
w.start_time_list = [0]
w.finish_time_list = [1]
w.assigned_task_list = [BaseTask("task")]
factory.initialize()
assert factory.cost_list == []
assert w.state == BaseFacilityState.FREE
assert w.cost_list == []
assert w.start_time_list == []
assert w.finish_time_list == []
assert w.assigned_task_list == []
def test_create_cost_history_plotly():
factory = BaseFactory("factory")
w1 = BaseFacility("w1", cost_per_time=10.0)
w1.cost_list = [0, 0, 10, 10, 0, 10]
w2 = BaseFacility("w2", cost_per_time=5.0)
w2.cost_list = [5, 5, 0, 0, 5, 5]
factory.facility_list = [w1, w2]
factory.cost_list = list(map(sum, zip(w1.cost_list, w2.cost_list)))
init_datetime = datetime.datetime(2020, 4, 1, 8, 0, 0)
timedelta = datetime.timedelta(days=1)
factory.create_cost_history_plotly(init_datetime, timedelta)
factory.create_cost_history_plotly(init_datetime,
timedelta,
title="bbbbbbb",
save_fig_path="test.png")
if os.path.exists("test.png"):
os.remove("test.png")
def test_create_data_for_gantt_plotly():
factory = BaseFactory("factory")
w1 = BaseFacility("w1", cost_per_time=10.0)
w1.start_time_list = [0, 5]
w1.finish_time_list = [2, 8]
w2 = BaseFacility("w2", cost_per_time=5.0)
w2.start_time_list = [9]
w2.finish_time_list = [11]
factory.facility_list = [w1, w2]
init_datetime = datetime.datetime(2020, 4, 1, 8, 0, 0)
timedelta = datetime.timedelta(days=1)
df = factory.create_data_for_gantt_plotly(init_datetime, timedelta)
# w1 part1
assert df[0]["Task"] == factory.name + ": " + w1.name
assert df[0]["Start"] == (
init_datetime +
w1.start_time_list[0] * timedelta).strftime("%Y-%m-%d %H:%M:%S")
assert df[0]["Finish"] == (init_datetime + (w1.finish_time_list[0] + 1.0) *
timedelta).strftime("%Y-%m-%d %H:%M:%S")
assert df[0]["Type"] == "Facility"
# w1 part2
assert df[1]["Task"] == factory.name + ": " + w1.name
assert df[1]["Start"] == (
init_datetime +
w1.start_time_list[1] * timedelta).strftime("%Y-%m-%d %H:%M:%S")
assert df[1]["Finish"] == (init_datetime + (w1.finish_time_list[1] + 1.0) *
timedelta).strftime("%Y-%m-%d %H:%M:%S")
assert df[1]["Type"] == "Facility"
# w2
assert df[2]["Start"] == (
init_datetime +
w2.start_time_list[0] * timedelta).strftime("%Y-%m-%d %H:%M:%S")
assert df[2]["Finish"] == (init_datetime + (w2.finish_time_list[0] + 1.0) *
timedelta).strftime("%Y-%m-%d %H:%M:%S")
assert df[2]["Type"] == "Facility"
