#!/usr/bin/python
# -*- coding: utf-8 -*-
"""Tests for BaseFacility.
This module contains unit tests for the BaseFacility class and related functionality.
"""
import pytest
from pDESy.model.base_facility import BaseFacility, BaseFacilityState
from pDESy.model.base_task import BaseTask
from pDESy.model.base_workplace import BaseWorkplace
[docs]
@pytest.fixture(name="dummy_facility")
def fixture_dummy_facility():
"""Fixture for a dummy BaseFacility.
Returns:
BaseFacility: A dummy facility instance.
"""
w = BaseFacility("dummy", workplace_id="---")
return w
[docs]
def test_init(dummy_facility):
"""Test initialization of BaseFacility.
Args:
dummy_facility (BaseFacility): The dummy facility fixture.
"""
# team = Team("team")
assert dummy_facility.name == "dummy"
assert dummy_facility.workplace_id == "---"
assert dummy_facility.cost_per_time == 0.0
assert not dummy_facility.solo_working
assert dummy_facility.workamount_skill_mean_map == {}
assert dummy_facility.workamount_skill_sd_map == {}
# assert dummy_facility.quality_skill_mean_map == {}
assert dummy_facility.state == BaseFacilityState.FREE
assert dummy_facility.cost_record_list == []
assert dummy_facility.assigned_task_worker_id_tuple_set == set()
w = BaseFacility(
"w1",
solo_working=True,
state=BaseFacilityState.WORKING,
cost_record_list=[10, 10],
state_record_list=["a"],
assigned_task_worker_id_tuple_set={(BaseTask("task").ID, "dummy_worker")},
assigned_task_worker_id_tuple_set_record_list=[
{(BaseTask("task").ID, "dummy_worker")},
{(BaseTask("task").ID, "dummy_worker")},
],
)
assert w.name == "w1"
assert w.workplace_id is None
assert w.cost_per_time == 0.0
assert w.solo_working
assert w.workamount_skill_mean_map == {}
assert w.workamount_skill_sd_map == {}
# assert w.quality_skill_mean_map == {}
assert w.state == BaseFacilityState.WORKING
assert w.cost_record_list == [10, 10]
[docs]
def test_str():
"""Test string representation of BaseFacility."""
print(BaseFacility("w1"))
[docs]
def test_initialize():
"""Test initialization/reset of BaseFacility."""
team = BaseWorkplace("team")
w = BaseFacility("w1", workplace_id=team.ID)
w.state = BaseFacilityState.WORKING
w.cost_record_list = [9.0, 7.2]
w.assigned_task_worker_id_tuple_set = {(BaseTask("task").ID, "dummy_worker")}
w.initialize()
assert w.state == BaseFacilityState.FREE
assert w.cost_record_list == []
assert w.assigned_task_worker_id_tuple_set == set()
[docs]
def test_remove_insert_absence_time_list():
"""Test removing and inserting absence time list for BaseFacility."""
w = BaseFacility("w1", "----")
w.cost_record_list = [1.0, 0.0, 1.0, 0.0, 0.0, 1.0]
w.assigned_task_worker_id_tuple_set_record_list = [
"aa",
"bb",
"cc",
"dd",
"ee",
"ff",
]
w.state_record_list = [2, 1, 2, 1, 1, 2]
absence_time_list = [0, 1, 4]
w.remove_absence_time_list(absence_time_list)
assert w.cost_record_list == [1.0, 0.0, 1.0]
assert w.assigned_task_worker_id_tuple_set_record_list == ["cc", "dd", "ff"]
assert w.state_record_list == [2, 1, 2]
w.insert_absence_time_list(absence_time_list)
assert w.cost_record_list == [0.0, 0.0, 1.0, 0.0, 0.0, 1.0]
assert w.assigned_task_worker_id_tuple_set_record_list == [
None,
None,
"cc",
"dd",
"dd",
"ff",
]
assert w.state_record_list == [
BaseFacilityState.FREE,
BaseFacilityState.FREE,
2,
1,
BaseFacilityState.FREE,
2,
]
[docs]
def test_has_workamount_skill():
"""Test has_workamount_skill method of BaseFacility."""
w = BaseFacility("w1", "----")
w.workamount_skill_mean_map = {"task1": 1.0, "task2": 0.0}
assert w.has_workamount_skill("task1")
assert not w.has_workamount_skill("task2")
assert not w.has_workamount_skill("task3")
[docs]
def test_check_update_state_from_absence_time_list():
"""Test check_update_state_from_absence_time_list method of BaseFacility."""
w = BaseFacility("w1", "----", absence_time_list=[1, 2, 4])
w.state = BaseFacilityState.FREE
w.check_update_state_from_absence_time_list(0)
assert w.state == BaseFacilityState.FREE
w.check_update_state_from_absence_time_list(1)
assert w.state == BaseFacilityState.ABSENCE
w.state = BaseFacilityState.WORKING
w.assigned_task_worker_id_tuple_set = set()
w.check_update_state_from_absence_time_list(2)
assert w.state == BaseFacilityState.ABSENCE
w.check_update_state_from_absence_time_list(3)
assert w.state == BaseFacilityState.FREE
task = BaseTask("task")
w.state = BaseFacilityState.WORKING
w.assigned_task_worker_id_tuple_set = {(task.ID, "dummy_worker")}
w.check_update_state_from_absence_time_list(2)
assert w.state == BaseFacilityState.ABSENCE
w.check_update_state_from_absence_time_list(3)
assert w.state == BaseFacilityState.WORKING
[docs]
def test_get_time_list_for_gantt_chart():
"""Test get_time_list_for_gantt_chart method of BaseFacility."""
w = BaseFacility("w1", "----")
w.state_record_list = [
BaseFacilityState.FREE,
BaseFacilityState.FREE,
BaseFacilityState.WORKING,
]
(
ready_time_list,
working_time_list,
absence_time_list,
) = w.get_time_list_for_gantt_chart()
assert ready_time_list == [(0, 2)]
assert working_time_list == [(2, 1)]
assert absence_time_list == [(0, 0)]
w.state_record_list = [
BaseFacilityState.WORKING,
BaseFacilityState.WORKING,
BaseFacilityState.FREE,
]
(
ready_time_list,
working_time_list,
absence_time_list,
) = w.get_time_list_for_gantt_chart()
assert ready_time_list == [(2, 1)]
assert working_time_list == [(0, 2)]
assert absence_time_list == [(0, 0)]
w.state_record_list = [
BaseFacilityState.WORKING,
BaseFacilityState.WORKING,
BaseFacilityState.WORKING,
]
(
ready_time_list,
working_time_list,
absence_time_list,
) = w.get_time_list_for_gantt_chart()
assert ready_time_list == [(0, 0)]
assert working_time_list == [(0, 3)]
assert absence_time_list == [(0, 0)]
# for backward
w.state_record_list = [
BaseFacilityState.FREE,
BaseFacilityState.WORKING,
BaseFacilityState.WORKING,
BaseFacilityState.WORKING,
BaseFacilityState.FREE,
BaseFacilityState.FREE,
BaseFacilityState.FREE,
BaseFacilityState.WORKING,
]
(
ready_time_list,
working_time_list,
absence_time_list,
) = w.get_time_list_for_gantt_chart()
assert ready_time_list == [(0, 1), (4, 3)]
assert working_time_list == [(1, 3), (7, 1)]
assert absence_time_list == [(0, 0)]
[docs]
def test_print_mermaid_diagram(dummy_facility):
"""Test the print_mermaid_diagram method.
Args:
dummy_facility (BaseFacility): The dummy facility fixture.
"""
dummy_facility.print_mermaid_diagram(
subgraph=True,
)