ross/proc-rota
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update tests

Browse Source
This commit is contained in:
Ross
2025-12-09 14:29:54 +00:00
parent 069da08f5a
commit dfd37a7a62
11 changed files with 78 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_rota_nights.py
+12 -12
View File
@@ -3,7 +3,7 @@ import pytest
from pytest import approx
from rota_generator.shifts import PostShiftConstraint, RotaBuilder, SingleShift, days
from rota_generator.workers import Worker
from web.rota.shifts import PreShiftConstraint
from rota_generator.shifts import PreShiftConstraint
@pytest.fixture
def demo_rota_nights():
@@ -16,17 +16,17 @@ def demo_rota_nights():
balance_offset_modifier=1,
)
Rota.constraint_options["max_shifts_per_week"] = 5
Rota.constraint_options["max_shifts_per_month"] = 31
Rota.constraint_options["limit_to_1_st2_on_nights"] = False
Rota.constraint_options["constrain_time_off_after_nights"] = False
Rota.constraint_options["balance_nights_across_site"] = False
Rota.constraint_options["balance_shifts"] = True
Rota.constraint_options["balance_shifts_over_workers"] = True
Rota.constraint_options["balance_nights"] = False
Rota.constraint_options["minimise_shift_diffs"] = False
Rota.constraint_options["balance_blocks"] = False
Rota.constraint_options["max_night_frequency"] = 0
Rota.constraint_options_model.max_shifts_per_week = 5
Rota.constraint_options_model.max_shifts_per_month = 31
# legacy option `limit_to_1_st2_on_nights` removed; test omits it
Rota.constraint_options_model.constrain_time_off_after_nights = False
Rota.constraint_options_model.balance_nights_across_sites = False
Rota.constraint_options_model.balance_shifts = True
Rota.constraint_options_model.balance_shifts_over_workers = True
Rota.constraint_options_model.balance_nights = False
Rota.constraint_options_model.minimise_shift_diffs = False
Rota.constraint_options_model.balance_blocks = False
Rota.constraint_options_model.max_night_frequency = 0
# Add workers
workers = [