rename home to remote

2022-02-24 21:50:37 +00:00
parent 7807e3b1e8
commit 9e873ac5f7
11 changed files with 806 additions and 145 deletions
@@ -7,8 +7,140 @@ import datetime

 from rota.workers import Worker

+def weeks_from_list(lst):
+    """Yield successive n-sized chunks from a lst."""
+    for i in range(0, len(lst), 7):
+        yield lst[i : i + 7]

 class TestDemoRota:
+    def test_balance_weekend(self):
+        # Set up rota
+        weeks_to_rota = 10
+        start_date = datetime.date(2022, 3, 7)
+
+        self.Rota = RotaBuilder(
+            start_date,
+            weeks_to_rota=weeks_to_rota,
+        )
+        self.Rota.constraint_options["balance_weekends"] = True
+
+        # Add a few workers
+        worker1 = Worker(self.Rota, "worker1", "group1", 1)
+        worker2 = Worker(self.Rota, "worker2", "group1", 1)
+        worker3 = Worker(self.Rota, "worker3", "group1", 1)
+        worker4 = Worker(self.Rota, "worker4", "group1", 1)
+
+        self.Rota.add_workers((worker1, worker2, worker3, worker4))
+
+        # Add a weekday and weekend shift
+        self.Rota.add_shifts(
+            SingleShift(
+                ("group1",),
+                "a",
+                12.5,
+                days,
+                balance_offset=1,
+                workers_required=1,
+            ),
+            SingleShift(
+                ("group1",),
+                "b",
+                12.5,
+                days[5:],
+                balance_offset=0,
+                workers_required=1
+            ),
+        )
+
+        self.Rota.build_and_solve(options={"ratio": 0.00})
+
+        self.Rota.export_rota_to_html("balance_weekend")
+
+        assert self.Rota.results.solver.status == "ok"
+        assert self.Rota.results.solver.termination_condition == "optimal"
+
+        for worker in self.Rota.workers:
+            shifts = self.Rota.get_worker_shift_list(worker)
+
+            # Convert shift to a string representation
+            shifts_string = "".join([i if i != "" else "-" for i in shifts])
+
+            assert shifts_string.count("b") == 5
+            assert shifts_string.count("a") in (17,18)
+
+            weekend_count = 0
+            for week in weeks_from_list(shifts_string):
+                if week[5:] == "--":
+                    continue
+                weekend_count = weekend_count + 1
+
+            assert weekend_count == 5 == self.Rota.model.worker_weekend_count[worker.id].value
+
+    def test_balance_weekend2(self):
+        # Set up rota
+        weeks_to_rota = 10
+        start_date = datetime.date(2022, 3, 7)
+
+        self.Rota = RotaBuilder(
+            start_date,
+            weeks_to_rota=weeks_to_rota,
+        )
+        self.Rota.constraint_options["balance_weekends"] = True
+
+        # Add a few workers
+        worker1 = Worker(self.Rota, "worker1", "group1", 1)
+        worker2 = Worker(self.Rota, "worker2", "group1", 1)
+        worker3 = Worker(self.Rota, "worker3", "group1", 1)
+        worker4 = Worker(self.Rota, "worker4", "group1", 1)
+        worker5 = Worker(self.Rota, "worker5", "group1", 1)
+
+        self.Rota.add_workers((worker1, worker2, worker3, worker4, worker5))
+
+        # Add a weekday and weekend shift
+        self.Rota.add_shifts(
+            SingleShift(
+                ("group1",),
+                "a",
+                12.5,
+                days,
+                balance_offset=5,
+                workers_required=1,
+                force_as_block=True
+            ),
+            SingleShift(
+                ("group1",),
+                "b",
+                12.5,
+                days[5:],
+                balance_offset=0,
+                workers_required=2
+            ),
+        )
+
+        self.Rota.build_and_solve(options={"ratio": 0.00})
+
+        self.Rota.export_rota_to_html("balance_weekend2")
+
+        assert self.Rota.results.solver.status == "ok"
+        assert self.Rota.results.solver.termination_condition == "optimal"
+
+        for worker in self.Rota.workers:
+            shifts = self.Rota.get_worker_shift_list(worker)
+
+            # Convert shift to a string representation
+            shifts_string = "".join([i if i != "" else "-" for i in shifts])
+
+            assert shifts_string.count("b") == 8
+            assert shifts_string.count("a") == 14
+
+            weekend_count = 0
+            for week in weeks_from_list(shifts_string):
+                if week[5:] == "--":
+                    continue
+                weekend_count = weekend_count + 1
+
+            assert weekend_count == 6 == self.Rota.model.worker_weekend_count[worker.id].value
+
    def test_post_mon_tue(self):
        # Set up rota
        weeks_to_rota = 10