Ross
85 lines
2.7 KiB
Python
85 lines
2.7 KiB
Python
import pytest
|
|
from rota.shifts import RotaBuilder, SingleShift, days
|
|
from rota.workers import Worker
|
|
import datetime
|
|
|
|
@pytest.fixture
|
|
def demo_rota_clear():
|
|
weeks_to_rota = 10
|
|
start_date = datetime.date(2022, 3, 7)
|
|
|
|
Rota = RotaBuilder(
|
|
start_date,
|
|
weeks_to_rota=weeks_to_rota,
|
|
balance_offset_modifier=1,
|
|
)
|
|
Rota.constraint_options["max_shifts_per_month"] = 20
|
|
|
|
workers = [
|
|
Worker(name="worker1", site="group1", grade=1),
|
|
Worker(name="worker2", site="group1", grade=1),
|
|
Worker(name="worker3", site="group1", grade=1),
|
|
Worker(name="worker4", site="group1", grade=1),
|
|
Worker(name="worker5", site="group2", grade=1, fte=60),
|
|
Worker(name="worker6", site="group2", grade=1, fte=40),
|
|
Worker(name="worker7", site="group2", grade=1),
|
|
Worker(name="worker8", site="group2", grade=1),
|
|
]
|
|
Rota.add_workers(workers)
|
|
return Rota
|
|
|
|
def test_pre(demo_rota_clear):
|
|
Rota = demo_rota_clear
|
|
Rota.add_shifts(
|
|
SingleShift(
|
|
sites=("group1", "group2"),
|
|
name="a",
|
|
length=12.5,
|
|
days=days[:5],
|
|
workers_required=2,
|
|
constraint=[{"name": "pre", "options": 1}],
|
|
),
|
|
SingleShift(
|
|
sites=("group1", "group2"),
|
|
name="b",
|
|
length=12.5,
|
|
days=days[:5],
|
|
workers_required=2,
|
|
),
|
|
SingleShift(
|
|
sites=("group1", "group2"),
|
|
name="c",
|
|
length=12.5,
|
|
days=days[3],
|
|
workers_required=1,
|
|
constraint=[{"name": "pre", "options": 1}, {"name": "post", "options": 1}],
|
|
),
|
|
)
|
|
|
|
Rota.build_shifts()
|
|
Rota.build_workers()
|
|
Rota.build_model()
|
|
|
|
solver_options = {"seconds": 1000, "threads": 10}
|
|
Rota.solve_model(options=solver_options)
|
|
Rota.export_rota_to_html("test4")
|
|
|
|
assert Rota.results.solver.status == "ok"
|
|
assert Rota.results.solver.termination_condition == "optimal"
|
|
|
|
for worker in Rota.workers:
|
|
shifts = Rota.get_worker_shift_list(worker)
|
|
shifts_string = "".join([i if i != "" else "-" for i in shifts])
|
|
|
|
# check for patterns that should not occur
|
|
assert "ba" not in shifts_string
|
|
assert "ca" not in shifts_string
|
|
assert "ac" not in shifts_string
|
|
assert "ca" not in shifts_string
|
|
assert "bc" not in shifts_string
|
|
assert "cb" not in shifts_string
|
|
# Uncomment if you want to check for these as well:
|
|
# assert "a-c" not in shifts_string
|
|
# assert "c-a" not in shifts_string
|
|
# assert "b-c" not in shifts_string
|
|
# assert "c-b" not in shifts_string |