def test_check_removing_placed_workplace():
"""test_check_removing_placed_workplace."""
c1 = BaseComponent("c1")
task1 = BaseTask("task1")
c1.append_targeted_task(task1)
c2 = BaseComponent("c2")
task2 = BaseTask("task2")
c2.append_targeted_task(task2)
product = BaseProduct([c1, c2])
f1 = BaseWorkplace("f1")
f2 = BaseWorkplace("f2")
c1.placed_workplace = f1
c2.placed_workplace = f2
f1.set_placed_component(c1)
f2.set_placed_component(c2)
# case1
task1.state = BaseTaskState.WORKING
task2.state = BaseTaskState.FINISHED
product.check_removing_placed_workplace()
assert c1.placed_workplace.name == "f1"
assert c2.placed_workplace is None
# case2
task1.state = BaseTaskState.FINISHED
task2.state = BaseTaskState.FINISHED
c1.append_child_component(c2)
c1.placed_workplace = f1
c2.placed_workplace = f1
f1.placed_component_list = [c1, c2]
product.check_removing_placed_workplace()
assert c1.placed_workplace is None
assert c2.placed_workplace is None
def test_get_available_space_size():
"""test_get_available_space_size."""
max_space_size = 5.0
workplace = BaseWorkplace("f", max_space_size=max_space_size)
assert workplace.get_available_space_size() == max_space_size
c1_space_size = 3.0
workplace.set_placed_component(BaseComponent("c1", space_size=c1_space_size))
assert workplace.get_available_space_size() == max_space_size - c1_space_size
def test_remove_placed_component():
"""test_remove_placed_component."""
c = BaseComponent("c")
c1 = BaseComponent("c1")
c2 = BaseComponent("c2")
c.append_child_component(c1)
c1.append_child_component(c2)
workplace = BaseWorkplace("workplace")
workplace.set_placed_component(c)
assert workplace.placed_component_list == [c, c1, c2]
workplace.remove_placed_component(c)
assert workplace.placed_component_list == []
def test_can_put():
"""test_can_put."""
c1 = BaseComponent("c1", space_size=2.0)
c2 = BaseComponent("c2", space_size=2.0)
workplace = BaseWorkplace("f", max_space_size=1.0)
assert workplace.can_put(c1) is False
assert workplace.can_put(c2) is False
workplace.max_space_size = 3.0
assert workplace.can_put(c1) is True
assert workplace.can_put(c2) is True
workplace.set_placed_component(c1)
assert workplace.can_put(c2) is False
workplace.max_space_size = 4.0
assert workplace.can_put(c2) is True
