further tests and fixes
This commit is contained in:
Ross
@@ -0,0 +1,499 @@
|
||||
import datetime
|
||||
import pytest
|
||||
from rota.shifts import InvalidShift, NoWorkers, RotaBuilder, SingleShift, days
|
||||
|
||||
from rota.workers import Worker
|
||||
|
||||
import itertools
|
||||
|
||||
|
||||
def generate_basic_rota(weeks_to_rota=10):
|
||||
start_date = datetime.date(2022, 3, 7)
|
||||
|
||||
Rota = RotaBuilder(
|
||||
start_date,
|
||||
weeks_to_rota=weeks_to_rota,
|
||||
)
|
||||
|
||||
# Add a few workers
|
||||
Rota.add_workers(
|
||||
[
|
||||
Worker(Rota, "worker1", "group1", 1),
|
||||
Worker(Rota, "worker2", "group1", 1),
|
||||
]
|
||||
)
|
||||
|
||||
return Rota
|
||||
|
||||
|
||||
def date_generator(from_date, days):
|
||||
n = 0
|
||||
while True:
|
||||
yield from_date
|
||||
|
||||
n = n + 1
|
||||
if n >= days:
|
||||
break
|
||||
from_date = from_date + datetime.timedelta(days=1)
|
||||
|
||||
|
||||
class TestWorkerRequests:
|
||||
def test_preferences_not_to_work(self):
|
||||
Rota = generate_basic_rota()
|
||||
|
||||
Rota.add_worker(
|
||||
Worker(
|
||||
Rota,
|
||||
"worker3",
|
||||
"group1",
|
||||
1,
|
||||
pref_not_to_work=(list(date_generator(datetime.date(2022, 3, 7), 5))),
|
||||
),
|
||||
)
|
||||
|
||||
Rota.add_shifts(
|
||||
SingleShift(
|
||||
("group1", "group2"),
|
||||
"a",
|
||||
12.5,
|
||||
days,
|
||||
workers_required=2,
|
||||
force_as_block=False,
|
||||
),
|
||||
)
|
||||
|
||||
Rota.build_and_solve(options={"ratio": 0.000})
|
||||
Rota.export_rota_to_html("preferences_not_to_work")
|
||||
|
||||
shift_summary = Rota.get_shift_summary_dict()
|
||||
for worker_name in shift_summary:
|
||||
worker_shifts = shift_summary[worker_name]
|
||||
|
||||
worker = Rota.get_worker_by_name(worker_name)
|
||||
|
||||
assert worker_shifts["a"] in (46, 47)
|
||||
|
||||
shift_string = Rota.get_worker_shift_list_string(worker)
|
||||
|
||||
if worker.name == "worker3":
|
||||
assert shift_string.startswith("-----")
|
||||
else:
|
||||
assert shift_string.startswith("aaaaa")
|
||||
|
||||
def test_preferences_not_to_work2(self):
|
||||
"""_summary_"""
|
||||
Rota = generate_basic_rota()
|
||||
|
||||
Rota.add_worker(
|
||||
Worker(
|
||||
Rota,
|
||||
"worker3",
|
||||
"group1",
|
||||
1,
|
||||
pref_not_to_work=(list(date_generator(datetime.date(2022, 3, 7), 55))),
|
||||
),
|
||||
)
|
||||
|
||||
Rota.add_shifts(
|
||||
SingleShift(
|
||||
("group1", "group2"),
|
||||
"a",
|
||||
12.5,
|
||||
days,
|
||||
workers_required=2,
|
||||
force_as_block=False,
|
||||
),
|
||||
)
|
||||
|
||||
Rota.build_and_solve(options={"ratio": 0.000})
|
||||
Rota.export_rota_to_html("preferences_not_to_work2")
|
||||
|
||||
shift_summary = Rota.get_shift_summary_dict()
|
||||
for worker_name in shift_summary:
|
||||
worker_shifts = shift_summary[worker_name]
|
||||
|
||||
worker = Rota.get_worker_by_name(worker_name)
|
||||
|
||||
assert worker_shifts["a"] in (46, 47)
|
||||
|
||||
shift_string = Rota.get_worker_shift_list_string(worker)
|
||||
|
||||
if worker.name == "worker3":
|
||||
assert shift_string.endswith("a" * 15)
|
||||
|
||||
def test_preferences_not_to_work_all_shifts(self):
|
||||
"""_summary_"""
|
||||
Rota = generate_basic_rota()
|
||||
|
||||
Rota.add_worker(
|
||||
Worker(
|
||||
Rota,
|
||||
"worker3",
|
||||
"group1",
|
||||
1,
|
||||
pref_not_to_work=(list(date_generator(datetime.date(2022, 3, 7), 70))),
|
||||
),
|
||||
)
|
||||
|
||||
Rota.add_shifts(
|
||||
SingleShift(
|
||||
("group1", "group2"),
|
||||
"a",
|
||||
12.5,
|
||||
days,
|
||||
workers_required=2,
|
||||
force_as_block=False,
|
||||
),
|
||||
)
|
||||
|
||||
Rota.build_and_solve(options={"ratio": 0.000})
|
||||
Rota.export_rota_to_html("preferences_not_to_work_all_shifts")
|
||||
|
||||
shift_summary = Rota.get_shift_summary_dict()
|
||||
for worker_name in shift_summary:
|
||||
worker_shifts = shift_summary[worker_name]
|
||||
|
||||
worker = Rota.get_worker_by_name(worker_name)
|
||||
|
||||
assert worker_shifts["a"] in (46, 47)
|
||||
|
||||
# shift_string = Rota.get_worker_shift_list_string(worker)
|
||||
|
||||
# if worker.name == "worker3":
|
||||
# assert shift_string.startswith("-----")
|
||||
# else:
|
||||
# assert shift_string.startswith("aaaaa")
|
||||
|
||||
def test_unavailable_to_work(self):
|
||||
"""_summary_"""
|
||||
Rota = generate_basic_rota()
|
||||
|
||||
Rota.add_worker(
|
||||
Worker(
|
||||
Rota,
|
||||
"worker3",
|
||||
"group1",
|
||||
1,
|
||||
not_available_to_work=(
|
||||
[
|
||||
(i, "NOT AROUND")
|
||||
for i in list(date_generator(datetime.date(2022, 3, 7), 23))
|
||||
]
|
||||
),
|
||||
),
|
||||
)
|
||||
|
||||
Rota.add_shifts(
|
||||
SingleShift(
|
||||
("group1", "group2"),
|
||||
"a",
|
||||
12.5,
|
||||
days,
|
||||
workers_required=2,
|
||||
force_as_block=False,
|
||||
),
|
||||
)
|
||||
|
||||
Rota.build_and_solve(options={"ratio": 0.000})
|
||||
Rota.export_rota_to_html("unavailable_to_work")
|
||||
|
||||
shift_summary = Rota.get_shift_summary_dict()
|
||||
for worker_name in shift_summary:
|
||||
worker_shifts = shift_summary[worker_name]
|
||||
|
||||
worker = Rota.get_worker_by_name(worker_name)
|
||||
|
||||
assert worker_shifts["a"] in (46, 47)
|
||||
|
||||
# shift_string = Rota.get_worker_shift_list_string(worker)
|
||||
|
||||
# if worker.name == "worker3":
|
||||
# assert shift_string.startswith("-----")
|
||||
# else:
|
||||
# assert shift_string.startswith("aaaaa")
|
||||
|
||||
def test_unavailable_to_work2(self):
|
||||
"""_summary_"""
|
||||
Rota = generate_basic_rota()
|
||||
|
||||
Rota.add_worker(
|
||||
Worker(
|
||||
Rota,
|
||||
"worker3",
|
||||
"group1",
|
||||
1,
|
||||
not_available_to_work=(
|
||||
[
|
||||
(i, "NOT AROUND")
|
||||
for i in list(date_generator(datetime.date(2022, 3, 7), 25))
|
||||
]
|
||||
),
|
||||
),
|
||||
)
|
||||
|
||||
Rota.add_shifts(
|
||||
SingleShift(
|
||||
("group1", "group2"),
|
||||
"a",
|
||||
12.5,
|
||||
days,
|
||||
workers_required=2,
|
||||
balance_offset=1,
|
||||
force_as_block=False,
|
||||
),
|
||||
)
|
||||
|
||||
Rota.build_and_solve(options={"ratio": 0.000})
|
||||
Rota.export_rota_to_html("unavailable_to_work2")
|
||||
|
||||
assert Rota.results.solver.status == "warning"
|
||||
assert Rota.results.solver.termination_condition == "infeasible"
|
||||
|
||||
def test_unavailable_to_work3(self):
|
||||
"""_summary_"""
|
||||
Rota = generate_basic_rota()
|
||||
|
||||
Rota.constraint_options["balance_bank_holidays"] = False
|
||||
|
||||
Rota.add_worker(
|
||||
Worker(
|
||||
Rota,
|
||||
"worker3",
|
||||
"group1",
|
||||
1,
|
||||
not_available_to_work=(
|
||||
[
|
||||
(i, "NOT AROUND")
|
||||
for i in list(date_generator(datetime.date(2022, 3, 7), 23))
|
||||
]
|
||||
),
|
||||
pref_not_to_work=[datetime.date(2022, 5, 15)],
|
||||
),
|
||||
)
|
||||
|
||||
Rota.add_shifts(
|
||||
SingleShift(
|
||||
("group1", "group2"),
|
||||
"a",
|
||||
12.5,
|
||||
days,
|
||||
workers_required=2,
|
||||
balance_offset=1,
|
||||
force_as_block=False,
|
||||
),
|
||||
)
|
||||
|
||||
Rota.build_and_solve(options={"ratio": 0.000})
|
||||
Rota.export_rota_to_html("unavailable_to_work3")
|
||||
|
||||
assert Rota.results.solver.status == "ok"
|
||||
assert Rota.results.solver.termination_condition == "optimal"
|
||||
|
||||
shift_summary = Rota.get_shift_summary_dict()
|
||||
for worker_name in shift_summary:
|
||||
worker_shifts = shift_summary[worker_name]
|
||||
|
||||
worker = Rota.get_worker_by_name(worker_name)
|
||||
|
||||
assert worker_shifts["a"] in (46, 47)
|
||||
|
||||
shift_string = Rota.get_worker_shift_list_string(worker)
|
||||
|
||||
if worker.name == "worker3":
|
||||
assert shift_string.startswith("-"*23)
|
||||
assert shift_string.endswith("-")
|
||||
|
||||
def test_unavailable_to_work2(self):
|
||||
"""_summary_"""
|
||||
Rota = generate_basic_rota()
|
||||
|
||||
Rota.add_workers(
|
||||
[Worker(
|
||||
Rota,
|
||||
"worker3",
|
||||
"group1",
|
||||
1,
|
||||
not_available_to_work=(
|
||||
[
|
||||
(i, "NOT AROUND")
|
||||
for i in list(date_generator(datetime.date(2022, 3, 7), 25))
|
||||
]
|
||||
),
|
||||
),
|
||||
Worker(
|
||||
Rota,
|
||||
"worker4",
|
||||
"group1",
|
||||
1,
|
||||
not_available_to_work=(
|
||||
[
|
||||
(i, "NOT AROUND")
|
||||
for i in list(date_generator(datetime.date(2022, 3, 7), 25))
|
||||
]
|
||||
),
|
||||
),]
|
||||
)
|
||||
|
||||
Rota.add_shifts(
|
||||
SingleShift(
|
||||
("group1", "group2"),
|
||||
"a",
|
||||
12.5,
|
||||
days,
|
||||
workers_required=3,
|
||||
balance_offset=1,
|
||||
force_as_block=False,
|
||||
),
|
||||
)
|
||||
|
||||
Rota.build_and_solve(options={"ratio": 0.000})
|
||||
Rota.export_rota_to_html("unavailable_to_work4")
|
||||
|
||||
assert Rota.results.solver.status == "warning"
|
||||
assert Rota.results.solver.termination_condition == "infeasible"
|
||||
|
||||
def test_work_requests(self):
|
||||
"""_summary_"""
|
||||
Rota = generate_basic_rota()
|
||||
Rota.constraint_options["balance_bank_holidays"] = False
|
||||
|
||||
Rota.add_workers(
|
||||
[Worker(
|
||||
Rota,
|
||||
"worker3",
|
||||
"group1",
|
||||
1,
|
||||
work_requests=(
|
||||
[
|
||||
(i, "a")
|
||||
for i in list(date_generator(datetime.date(2022, 3, 7), 35))
|
||||
]
|
||||
),
|
||||
),
|
||||
Worker(
|
||||
Rota,
|
||||
"worker4",
|
||||
"group1",
|
||||
1,
|
||||
work_requests=(
|
||||
[
|
||||
(i, "a")
|
||||
for i in list(date_generator(datetime.date(2022, 3, 7), 35))
|
||||
]
|
||||
),
|
||||
),]
|
||||
)
|
||||
|
||||
Rota.add_shifts(
|
||||
SingleShift(
|
||||
("group1", "group2"),
|
||||
"a",
|
||||
12.5,
|
||||
days,
|
||||
workers_required=2,
|
||||
balance_offset=1,
|
||||
force_as_block=False,
|
||||
),
|
||||
)
|
||||
|
||||
Rota.build_and_solve(options={"ratio": 0.000})
|
||||
Rota.export_rota_to_html("work_requests")
|
||||
|
||||
shift_summary = Rota.get_shift_summary_dict()
|
||||
for worker_name in shift_summary:
|
||||
worker_shifts = shift_summary[worker_name]
|
||||
|
||||
worker = Rota.get_worker_by_name(worker_name)
|
||||
|
||||
assert worker_shifts["a"] == 35
|
||||
|
||||
shift_string = Rota.get_worker_shift_list_string(worker)
|
||||
|
||||
if worker.name in ("worker3", "worker4"):
|
||||
assert shift_string.startswith("a"*35)
|
||||
|
||||
def test_work_requests_invalid(self):
|
||||
"""_summary_"""
|
||||
Rota = generate_basic_rota()
|
||||
Rota.constraint_options["balance_bank_holidays"] = False
|
||||
|
||||
Rota.add_worker(
|
||||
Worker(
|
||||
Rota,
|
||||
"worker3",
|
||||
"group1",
|
||||
1,
|
||||
work_requests=(
|
||||
[
|
||||
(i, "b")
|
||||
for i in list(date_generator(datetime.date(2022, 3, 7), 35))
|
||||
]
|
||||
),
|
||||
),
|
||||
)
|
||||
|
||||
Rota.add_shifts(
|
||||
SingleShift(
|
||||
("group1", "group2"),
|
||||
"a",
|
||||
12.5,
|
||||
days,
|
||||
workers_required=1,
|
||||
balance_offset=1,
|
||||
force_as_block=False,
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
with pytest.raises(InvalidShift):
|
||||
Rota.build_and_solve(options={"ratio": 0.000})
|
||||
|
||||
Rota.add_shifts(
|
||||
SingleShift(
|
||||
("group1", "group2"),
|
||||
"c",
|
||||
12.5,
|
||||
days,
|
||||
workers_required=1,
|
||||
balance_offset=1,
|
||||
force_as_block=False,
|
||||
),
|
||||
)
|
||||
|
||||
with pytest.raises(InvalidShift):
|
||||
Rota.build_and_solve(options={"ratio": 0.000})
|
||||
|
||||
Rota.add_shifts(
|
||||
SingleShift(
|
||||
("group1", "group2"),
|
||||
"b",
|
||||
12.5,
|
||||
days,
|
||||
workers_required=1,
|
||||
balance_offset=1,
|
||||
force_as_block=False,
|
||||
),
|
||||
)
|
||||
|
||||
Rota.build_and_solve(options={"ratio": 0.000})
|
||||
|
||||
assert Rota.results.solver.status == "ok"
|
||||
assert Rota.results.solver.termination_condition == "optimal"
|
||||
|
||||
Rota.add_worker(
|
||||
Worker(
|
||||
Rota,
|
||||
"worker4",
|
||||
"group1",
|
||||
1,
|
||||
work_requests=(
|
||||
[
|
||||
(i, "d")
|
||||
for i in list(date_generator(datetime.date(2022, 3, 7), 35))
|
||||
]
|
||||
),
|
||||
),
|
||||
)
|
||||
|
||||
with pytest.raises(InvalidShift):
|
||||
Rota.build_and_solve(options={"ratio": 0.000})
|
||||
Reference in New Issue
Block a user