add custom display char for shift

test/test_shifts.py

@@ -708,4 +708,25 @@ class TestShiftWorkerRequirements:
         total_shifts = Rota.get_workers_total_shifts()
 
         for worker in Rota.get_workers():
-            assert total_shifts[worker.name] == 30
+            assert total_shifts[worker.name] == 30
+
+class TestDisplayChar:
+    def test_display_char_in_output(monkeypatch):
+        Rota = generate_basic_rota(workers=1)
+        shift = SingleShift(
+            sites=("group1",),
+            name="testshift",
+            length=8,
+            days=days[:1],
+            display_char="Z"
+        )
+        Rota.add_shifts(shift)
+
+        assert shift.display_char == "Z"
+
+        Rota.build_and_solve(options={"ratio": 0.1})
+
+        assert Rota.results.solver.status == "ok"
+
+        for worker in Rota.get_workers():
+            assert Rota.get_worker_shift_list_string(worker).startswith("Z------" * 5)