e

2022-07-27 07:49:34 +01:00
parent ecd81236a2
commit 7ef0c191d7
5 changed files with 411 additions and 51 deletions
@@ -318,3 +318,76 @@ class TestWorkers:
        assert Rota.results.solver.termination_condition == "optimal"

        Rota.export_rota_to_html("worker_oops")
+
+    def test_worker_pairs(self):
+        Rota = setup_basic_rota()
+
+        Rota.add_shifts(
+            SingleShift(
+                sites=("group1",),
+                name="a",
+                length=12.5,
+                days=days[:5],
+                balance_offset=20,
+                workers_required=1,
+            ),
+            SingleShift(
+                sites=("group1",),
+                name="b",
+                length=12.5,
+                days=days[:5],
+                balance_offset=20,
+                workers_required=1,
+            ),
+        )
+
+        Rota.add_workers(
+            [
+                Worker(name="worker1", site="group1", grade=1, fte=100, pair="1"),
+                Worker(name="worker2", site="group1", grade=1, fte=100, pair="1"),
+            ]
+        )
+
+        Rota.constraint_options["hard_constrain_pair_separation"] = True
+
+        Rota.build_and_solve()
+
+        # With two paired workers we shouldn't be able to solve
+        assert Rota.results.solver.status == "warning"
+
+        worker3 = Worker(name="worker3", site="group1", grade=1, fte=100)
+        Rota.add_worker(worker3)
+
+        # An additional worker can take half the shifts
+        Rota.build_and_solve()
+
+        assert Rota.results.solver.status == "ok"
+
+        assert Rota.get_workers_total_shifts()["worker3"] == 50
+        assert (
+            Rota.get_workers_total_shifts()["worker1"]
+            + Rota.get_workers_total_shifts()["worker2"]
+            == 50
+        )
+
+        worker3.pair = "1"
+
+        Rota.build_and_solve()
+
+        assert Rota.results.solver.status == "warning"
+
+        worker3.pair = 0
+
+        Rota.build_and_solve()
+        Rota.export_rota_to_html("worker_pairs")
+
+        assert Rota.results.solver.status == "ok"
+
+        worker4 = Worker(name="worker4", site="group1", grade=1, fte=100, pair="1")
+        Rota.add_worker(worker4)
+
+        Rota.build_and_solve()
+        Rota.export_rota_to_html("worker_pairs")
+
+        assert Rota.results.solver.status == "ok"
+        assert Rota.get_workers_total_shifts()["worker3"] == 50