def test_remove_insert_absence_time_list():
"""test_remove_insert_absence_time_list."""
w1 = BaseWorker("w1", "----")
w1.cost_list = [1.0, 0.0, 1.0, 0.0, 0.0, 1.0]
w1.assigned_task_id_record = ["aa", "bb", "cc", "dd", "ee", "ff"]
w1.state_record_list = [2, 1, 2, 1, 1, 2]
w2 = BaseWorker("w1", "----")
w2.cost_list = [1.0, 0.0, 1.0, 0.0, 0.0, 1.0]
w2.assigned_task_id_record = ["aa", "bb", "cc", "dd", "ee", "ff"]
w2.state_record_list = [2, 1, 2, 1, 1, 2]
team = BaseTeam("aa", worker_list=[w1, w2])
team.cost_list = [2.0, 0.0, 2.0, 0.0, 0.0, 2.0]
absence_time_list = [1, 3, 4]
team.remove_absence_time_list(absence_time_list)
assert team.cost_list == [2.0, 2.0, 2.0]
assert w1.cost_list == [1.0, 1.0, 1.0]
assert w1.assigned_task_id_record == ["aa", "cc", "ff"]
assert w1.state_record_list == [2, 2, 2]
assert w2.cost_list == [1.0, 1.0, 1.0]
assert w2.assigned_task_id_record == ["aa", "cc", "ff"]
assert w2.state_record_list == [2, 2, 2]
team.insert_absence_time_list(absence_time_list)
assert team.cost_list == [2.0, 0.0, 2.0, 0.0, 0.0, 2.0]
assert w1.cost_list == [1.0, 0.0, 1.0, 0.0, 0.0, 1.0]
assert w1.assigned_task_id_record == ["aa", "aa", "cc", "cc", "cc", "ff"]
assert w1.state_record_list == [2, 0, 2, 0, 0, 2]
assert w2.cost_list == [1.0, 0.0, 1.0, 0.0, 0.0, 1.0]
assert w2.assigned_task_id_record == ["aa", "aa", "cc", "cc", "cc", "ff"]
assert w2.state_record_list == [2, 0, 2, 0, 0, 2]
def test_create_data_for_cost_history_plotly():
"""test_create_data_for_cost_history_plotly."""
team = BaseTeam("team")
w1 = BaseWorker("w1", cost_per_time=10.0)
w1.cost_list = [0, 0, 10, 10, 0, 10]
w2 = BaseWorker("w2", cost_per_time=5.0)
w2.cost_list = [5, 5, 0, 0, 5, 5]
team.worker_list = [w1, w2]
team.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 = team.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(team.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_create_cost_history_plotly(tmpdir):
"""test_create_cost_history_plotly."""
team = BaseTeam("team")
w1 = BaseWorker("w1", cost_per_time=10.0)
w1.cost_list = [0, 0, 10, 10, 0, 10]
w2 = BaseWorker("w2", cost_per_time=5.0)
w2.cost_list = [5, 5, 0, 0, 5, 5]
team.worker_list = [w1, w2]
team.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)
team.create_cost_history_plotly(init_datetime, timedelta)
for ext in ["png", "html", "json"]:
save_fig_path = os.path.join(str(tmpdir), "test." + ext)
team.create_cost_history_plotly(
init_datetime, timedelta, title="bbbbbbb", save_fig_path=save_fig_path
)
def test_initialize():
"""test_initialize."""
team = BaseTeam("team")
w = BaseWorker("w1", team_id=team.ID)
w.state = BaseWorkerState.WORKING
w.cost_list = [9.0, 7.2]
w.assigned_task_list = [BaseTask("task")]
w.initialize()
assert w.state == BaseWorkerState.FREE
assert w.cost_list == []
assert w.assigned_task_list == []
def test_initialize():
team = BaseTeam("team")
w = BaseWorker("w1", team_id=team.ID)
w.state = BaseResourceState.WORKING
w.cost_list = [9.0, 7.2]
w.start_time_list = [0]
w.finish_time_list = [1]
w.assigned_task_list = [BaseTask("task")]
w.initialize()
assert w.state == BaseResourceState.FREE
assert w.cost_list == []
assert w.start_time_list == []
assert w.finish_time_list == []
assert w.assigned_task_list == []
def test_initialize():
"""test_initialize."""
team = BaseTeam("team")
team.cost_list = [9.0, 7.2]
w = BaseWorker("w1")
team.worker_list = [w]
w.state = BaseWorkerState.WORKING
w.cost_list = [9.0, 7.2]
w.assigned_task_list = [BaseTask("task")]
team.initialize()
assert team.cost_list == []
assert w.state == BaseWorkerState.FREE
assert w.cost_list == []
assert w.assigned_task_list == []
def test_remove_insert_absence_time_list():
"""test_remove_insert_absence_time_list."""
w = BaseWorker("w1", "----")
w.cost_list = [1.0, 0.0, 1.0, 0.0, 0.0, 1.0]
w.assigned_task_id_record = ["aa", "bb", "cc", "dd", "ee", "ff"]
w.state_record_list = [2, 1, 2, 1, 1, 2]
absence_time_list = [1, 3, 4]
w.remove_absence_time_list(absence_time_list)
assert w.cost_list == [1.0, 1.0, 1.0]
assert w.assigned_task_id_record == ["aa", "cc", "ff"]
assert w.state_record_list == [2, 2, 2]
w.insert_absence_time_list(absence_time_list)
assert w.cost_list == [1.0, 0.0, 1.0, 0.0, 0.0, 1.0]
assert w.assigned_task_id_record == ["aa", "aa", "cc", "cc", "cc", "ff"]
assert w.state_record_list == [2, 0, 2, 0, 0, 2]
def test_remove_insert_absence_time_list():
"""test_remove_insert_absence_time_list."""
f1 = BaseFacility("w1", "----")
f1.cost_list = [1.0, 0.0, 1.0, 0.0, 0.0, 1.0]
f1.assigned_task_id_record = ["aa", "bb", "cc", "dd", "ee", "ff"]
f1.state_record_list = [2, 1, 2, 1, 1, 2]
f2 = BaseFacility("w1", "----")
f2.cost_list = [1.0, 0.0, 1.0, 0.0, 0.0, 1.0]
f2.assigned_task_id_record = ["aa", "bb", "cc", "dd", "ee", "ff"]
f2.state_record_list = [2, 1, 2, 1, 1, 2]
workplace = BaseWorkplace("aa", facility_list=[f1, f2])
workplace.cost_list = [2.0, 0.0, 2.0, 0.0, 0.0, 2.0]
w1 = BaseWorker("w1", "----")
w1.cost_list = [1.0, 0.0, 1.0, 0.0, 0.0, 1.0]
w1.assigned_task_id_record = ["aa", "bb", "cc", "dd", "ee", "ff"]
w1.state_record_list = [2, 1, 2, 1, 1, 2]
w2 = BaseWorker("w1", "----")
w2.cost_list = [1.0, 0.0, 1.0, 0.0, 0.0, 1.0]
w2.assigned_task_id_record = ["aa", "bb", "cc", "dd", "ee", "ff"]
w2.state_record_list = [2, 1, 2, 1, 1, 2]
team = BaseTeam("aa", worker_list=[w1, w2])
team.cost_list = [2.0, 0.0, 2.0, 0.0, 0.0, 2.0]
organization = BaseOrganization(team_list=[team],
workplace_list=[workplace])
organization.cost_list = [4.0, 0.0, 4.0, 0.0, 0.0, 4.0]
absence_time_list = [1, 3, 4]
organization.remove_absence_time_list(absence_time_list)
assert organization.cost_list == [4.0, 4.0, 4.0]
assert workplace.cost_list == [2.0, 2.0, 2.0]
assert f1.cost_list == [1.0, 1.0, 1.0]
assert f1.assigned_task_id_record == ["aa", "cc", "ff"]
assert f1.state_record_list == [2, 2, 2]
assert f2.cost_list == [1.0, 1.0, 1.0]
assert f2.assigned_task_id_record == ["aa", "cc", "ff"]
assert f2.state_record_list == [2, 2, 2]
assert team.cost_list == [2.0, 2.0, 2.0]
assert w1.cost_list == [1.0, 1.0, 1.0]
assert w1.assigned_task_id_record == ["aa", "cc", "ff"]
assert w1.state_record_list == [2, 2, 2]
assert w2.cost_list == [1.0, 1.0, 1.0]
assert w2.assigned_task_id_record == ["aa", "cc", "ff"]
assert w2.state_record_list == [2, 2, 2]
organization.insert_absence_time_list(absence_time_list)
assert organization.cost_list == [4.0, 0.0, 4.0, 0.0, 0.0, 4.0]
assert team.cost_list == [2.0, 0.0, 2.0, 0.0, 0.0, 2.0]
assert w1.cost_list == [1.0, 0.0, 1.0, 0.0, 0.0, 1.0]
assert w1.assigned_task_id_record == ["aa", "aa", "cc", "cc", "cc", "ff"]
assert w1.state_record_list == [2, 0, 2, 0, 0, 2]
assert w2.cost_list == [1.0, 0.0, 1.0, 0.0, 0.0, 1.0]
assert w2.assigned_task_id_record == ["aa", "aa", "cc", "cc", "cc", "ff"]
assert w2.state_record_list == [2, 0, 2, 0, 0, 2]
assert workplace.cost_list == [2.0, 0.0, 2.0, 0.0, 0.0, 2.0]
assert f1.cost_list == [1.0, 0.0, 1.0, 0.0, 0.0, 1.0]
assert f1.assigned_task_id_record == ["aa", "aa", "cc", "cc", "cc", "ff"]
assert f1.state_record_list == [2, 0, 2, 0, 0, 2]
assert f2.cost_list == [1.0, 0.0, 1.0, 0.0, 0.0, 1.0]
assert f2.assigned_task_id_record == ["aa", "aa", "cc", "cc", "cc", "ff"]
assert f2.state_record_list == [2, 0, 2, 0, 0, 2]