def test_add_labor_cost():
"""test_add_labor_cost."""
team = BaseTeam("team")
w1 = BaseWorker("w1", cost_per_time=10.0)
w2 = BaseWorker("w2", cost_per_time=5.0)
team.worker_list = [w2, w1]
w1.state = BaseWorkerState.WORKING
w2.state = BaseWorkerState.FREE
team.add_labor_cost()
assert w1.cost_list == [10.0]
assert w2.cost_list == [0.0]
assert team.cost_list == [10.0]
team.add_labor_cost(only_working=False)
assert team.cost_list == [10.0, 15.0]
assert w1.cost_list == [10.0, 10.0]
assert w2.cost_list == [0.0, 5.0]
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 == []