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

start pydanitification

Browse Source
This commit is contained in:
Ross
2022-05-26 15:38:35 +01:00
parent a8cf3e3460
commit c9ba2a098c
14 changed files with 767 additions and 459 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_worker_requests.py
+27 -63
View File
@@ -18,8 +18,8 @@ def generate_basic_rota(weeks_to_rota=10):
# Add a few workers
Rota.add_workers(
[
Worker(Rota, "worker1", "group1", 1),
Worker(Rota, "worker2", "group1", 1),
Worker(name="worker1", site="group1", grade=1),
Worker(name="worker2", site="group1", grade=1),
]
)
@@ -43,11 +43,8 @@ class TestWorkerRequests:
Rota.add_worker(
Worker(
Rota,
"worker3",
"group1",
1,
pref_not_to_work=(list(date_generator(datetime.date(2022, 3, 7), 5))),
name="worker3", site="group1", grade=1,
pref_not_to_work=[{"date": d} for d in date_generator(datetime.date(2022, 3, 7), 5)],
),
)
@@ -86,11 +83,8 @@ class TestWorkerRequests:
Rota.add_worker(
Worker(
Rota,
"worker3",
"group1",
1,
pref_not_to_work=(list(date_generator(datetime.date(2022, 3, 7), 55))),
name="worker3", site="group1", grade=1,
pref_not_to_work=[{"date": d} for d in date_generator(datetime.date(2022, 3, 7), 55)],
),
)
@@ -127,11 +121,8 @@ class TestWorkerRequests:
Rota.add_worker(
Worker(
Rota,
"worker3",
"group1",
1,
pref_not_to_work=(list(date_generator(datetime.date(2022, 3, 7), 70))),
name="worker3", site="group1", grade=1,
pref_not_to_work=[{"date": d} for d in date_generator(datetime.date(2022, 3, 7), 70)],
),
)
@@ -170,13 +161,10 @@ class TestWorkerRequests:
Rota.add_worker(
Worker(
Rota,
"worker3",
"group1",
1,
name="worker3", site="group1", grade=1,
not_available_to_work=(
[
(i, "NOT AROUND")
{"date": i, "reason": "NOT AROUND"}
for i in list(date_generator(datetime.date(2022, 3, 7), 23))
]
),
@@ -218,13 +206,10 @@ class TestWorkerRequests:
Rota.add_worker(
Worker(
Rota,
"worker3",
"group1",
1,
name="worker3", site="group1", grade=1,
not_available_to_work=(
[
(i, "NOT AROUND")
{"date": i, "reason": "NOT AROUND"}
for i in list(date_generator(datetime.date(2022, 3, 7), 25))
]
),
@@ -257,17 +242,14 @@ class TestWorkerRequests:
Rota.add_worker(
Worker(
Rota,
"worker3",
"group1",
1,
name="worker3", site="group1", grade=1,
not_available_to_work=(
[
(i, "NOT AROUND")
{"date":i, "reason":"NOT AROUND"}
for i in list(date_generator(datetime.date(2022, 3, 7), 23))
]
),
pref_not_to_work=[datetime.date(2022, 5, 15)],
pref_not_to_work=[{"date":datetime.date(2022, 5, 15)}],
),
)
@@ -309,25 +291,19 @@ class TestWorkerRequests:
Rota.add_workers(
[Worker(
Rota,
"worker3",
"group1",
1,
name="worker3", site="group1", grade=1,
not_available_to_work=(
[
(i, "NOT AROUND")
{"date": i, "reason": "NOT AROUND"}
for i in list(date_generator(datetime.date(2022, 3, 7), 25))
]
),
),
Worker(
Rota,
"worker4",
"group1",
1,
name="worker4", site="group1", grade=1,
not_available_to_work=(
[
(i, "NOT AROUND")
{"date": i, "reason": "NOT AROUND"}
for i in list(date_generator(datetime.date(2022, 3, 7), 25))
]
),
@@ -359,25 +335,19 @@ class TestWorkerRequests:
Rota.add_workers(
[Worker(
Rota,
"worker3",
"group1",
1,
name="worker3", site="group1", grade=1,
work_requests=(
[
(i, "a")
{"date": i, "shift": "a"}
for i in list(date_generator(datetime.date(2022, 3, 7), 35))
]
),
),
Worker(
Rota,
"worker4",
"group1",
1,
name="worker4", site="group1", grade=1,
work_requests=(
[
(i, "a")
{"date": i, "shift": "a"}
for i in list(date_generator(datetime.date(2022, 3, 7), 35))
]
),
@@ -419,13 +389,10 @@ class TestWorkerRequests:
Rota.add_worker(
Worker(
Rota,
"worker3",
"group1",
1,
name="worker3", site="group1", grade=1,
work_requests=(
[
(i, "b")
{"date": i, "shift": "b"}
for i in list(date_generator(datetime.date(2022, 3, 7), 35))
]
),
@@ -482,13 +449,10 @@ class TestWorkerRequests:
Rota.add_worker(
Worker(
Rota,
"worker4",
"group1",
1,
name="worker4", site="group1", grade=1,
work_requests=(
[
(i, "d")
{"date": i, "shift": "d"}
for i in list(date_generator(datetime.date(2022, 3, 7), 35))
]
),