.

test/rota_constraints/test_avoid_grades_by_week.py

@@ -1,8 +1,8 @@
 import datetime
 import pytest
-from rota.shifts import InvalidShift, NoWorkers, RotaBuilder, SingleShift, days
+from rota_generator.shifts import InvalidShift, NoWorkers, RotaBuilder, SingleShift, days
 
-from rota.workers import Worker
+from rota_generator.workers import Worker
 
 import itertools
 

test/rota_constraints/test_avoid_workers_by_week.py

@@ -1,8 +1,8 @@
 import datetime
 import pytest
-from rota.shifts import InvalidShift, NoWorkers, RotaBuilder, SingleShift, days
+from rota_generator.shifts import InvalidShift, NoWorkers, RotaBuilder, SingleShift, days
 
-from rota.workers import Worker
+from rota_generator.workers import Worker
 
 import itertools
 

test/rota_constraints/test_summed_shift_grade_constraints.py

@@ -1,8 +1,8 @@
 import datetime
 import pytest
-from rota.shifts import InvalidShift, NoWorkers, RotaBuilder, SingleShift, days, MinSummedGradeByShiftsPerDayConstraint
+from rota_generator.shifts import InvalidShift, NoWorkers, RotaBuilder, SingleShift, days, MinSummedGradeByShiftsPerDayConstraint
 
-from rota.workers import Worker
+from rota_generator.workers import Worker
 
 import itertools
 

test/test_balance_shift_sites.py

@@ -1,7 +1,7 @@
 import pytest
 import datetime
-from rota.shifts import BalanceAcrossGroupsConstraint, RotaBuilder, SingleShift, days
-from rota.workers import Worker
+from rota_generator.shifts import BalanceAcrossGroupsConstraint, RotaBuilder, SingleShift, days
+from rota_generator.workers import Worker
 
 @pytest.fixture
 def demo_rota_balance_sites():

test/test_general_balancing.py

@@ -1,8 +1,8 @@
 import datetime
 import pytest
 from pytest import approx
-from rota.shifts import PostShiftConstraint, PreShiftConstraint, RotaBuilder, SingleShift, days
-from rota.workers import Worker
+from rota_generator.shifts import PostShiftConstraint, PreShiftConstraint, RotaBuilder, SingleShift, days
+from rota_generator.workers import Worker
 
 def generate_basic_rota(weeks_to_rota=10):
     start_date = datetime.date(2022, 3, 7)

test/test_limit_constraints.py

@@ -1,7 +1,7 @@
 import pytest
 import datetime
-from rota.shifts import MinimumGradeNumberConstraint, RequireRemoteSitePresenceConstraint, RotaBuilder, SingleShift, days, LimitGradeNumberConstraint
-from rota.workers import Worker
+from rota_generator.shifts import MinimumGradeNumberConstraint, RequireRemoteSitePresenceConstraint, RotaBuilder, SingleShift, days, LimitGradeNumberConstraint
+from rota_generator.workers import Worker
 
 @pytest.fixture
 def limit_constraint_rota():

test/test_multiple_shifts_per_day.py

@@ -1,8 +1,8 @@
 
 import pytest
 import datetime
-from rota.shifts import RotaBuilder, SingleShift, days
-from rota.workers import Worker
+from rota_generator.shifts import RotaBuilder, SingleShift, days
+from rota_generator.workers import Worker
 
 @pytest.fixture
 def basic_rota():

test/test_night_unavailable.py

@@ -1,7 +1,7 @@
 import pytest
 import datetime
-from rota.shifts import NightConstraint, RotaBuilder, SingleShift, days
-from rota.workers import Worker
+from rota_generator.shifts import NightConstraint, RotaBuilder, SingleShift, days
+from rota_generator.workers import Worker
 
 @pytest.fixture
 def demo_rota_night_unavailable():

test/test_nights2.py

@@ -1,7 +1,7 @@
 import datetime
 import pytest
-from rota.shifts import InvalidShift, NightConstraint, NoWorkers, PostShiftConstraint, PreShiftConstraint, RotaBuilder, SingleShift, days
-from rota.workers import Worker
+from rota_generator.shifts import InvalidShift, NightConstraint, NoWorkers, PostShiftConstraint, PreShiftConstraint, RotaBuilder, SingleShift, days
+from rota_generator.workers import Worker
 
 def generate_basic_rota(weeks_to_rota=10):
     start_date = datetime.date(2022, 3, 7)

test/test_nwds.py

@@ -1,7 +1,7 @@
 import pytest
-from rota.shifts import NoWorkers, RotaBuilder, SingleShift, days
+from rota_generator.shifts import NoWorkers, RotaBuilder, SingleShift, days
 import datetime
-from rota.workers import Worker
+from rota_generator.workers import Worker
 
 def weeks_from_list(lst):
     """Yield successive n-sized chunks from a lst."""

test/test_remote.py

@@ -1,9 +1,9 @@
 
 import pytest
-from rota.shifts import NoWorkers, RequireRemoteSitePresenceConstraint, RotaBuilder, SingleShift, days
+from rota_generator.shifts import NoWorkers, RequireRemoteSitePresenceConstraint, RotaBuilder, SingleShift, days
 
 import datetime
-from rota.workers import Worker
+from rota_generator.workers import Worker
 
 def setup_basic_rota():
     weeks_to_rota = 10

test/test_rota.py

@@ -3,11 +3,11 @@ from re import S
 from black import main
 import pytest
 from pytest import approx
-from rota.shifts import NoWorkers, RotaBuilder, SingleShift, WarningTermination, days
+from rota_generator.shifts import NoWorkers, RotaBuilder, SingleShift, WarningTermination, days
 
 import datetime
 
-from rota.workers import Worker
+from rota_generator.workers import Worker
 
 
 @pytest.fixture

test/test_rota_clear.py

@@ -1,6 +1,6 @@
 import pytest
-from rota.shifts import PostShiftConstraint, PreShiftConstraint, RotaBuilder, SingleShift, days
-from rota.workers import Worker
+from rota_generator.shifts import PostShiftConstraint, PreShiftConstraint, RotaBuilder, SingleShift, days
+from rota_generator.workers import Worker
 import datetime
 
 

test/test_rota_nights.py

@@ -1,8 +1,8 @@
 import datetime
 import pytest
 from pytest import approx
-from rota.shifts import PostShiftConstraint, RotaBuilder, SingleShift, days
-from rota.workers import Worker
+from rota_generator.shifts import PostShiftConstraint, RotaBuilder, SingleShift, days
+from rota_generator.workers import Worker
 from web.rota.shifts import PreShiftConstraint
 
 @pytest.fixture

test/test_shift_constraints.py

@@ -1,8 +1,8 @@
 import pytest
 import datetime
 from copy import deepcopy
-from rota.shifts import RotaBuilder, SingleShift, days
-from rota.workers import Worker
+from rota_generator.shifts import RotaBuilder, SingleShift, days
+from rota_generator.workers import Worker
 
 @pytest.fixture
 def rota_2workers():

test/test_shifts.py

@@ -1,11 +1,11 @@
 import datetime
 import pytest
-from rota.shifts import (
+from rota_generator.shifts import (
     InvalidShift, MaxShiftsPerWeekBlockConstraint, MaxShiftsPerWeekConstraint,
     NoWorkers, PostShiftConstraint, PreShiftConstraint, RotaBuilder, SingleShift,
     WarningTermination, days, WorkerRequirement
 )
-from rota.workers import NotAvailableToWork, WorkRequests, Worker, generate_not_available_to_works
+from rota_generator.workers import NotAvailableToWork, WorkRequests, Worker, generate_not_available_to_works
 
 def generate_basic_rota(weeks_to_rota=10, workers=2, start_date=datetime.date(2022, 3, 7)):
     Rota = RotaBuilder(

test/test_weekend_padding.py

@@ -1,9 +1,9 @@
 import pytest
-from rota.shifts import NoWorkers, RotaBuilder, SingleShift, days
+from rota_generator.shifts import NoWorkers, RotaBuilder, SingleShift, days
 
 import datetime
 
-from rota.workers import Worker
+from rota_generator.workers import Worker
 
 def weeks_from_list(lst):
     """Yield successive n-sized chunks from a lst."""

test/test_worker_requests.py

@@ -1,9 +1,9 @@
 import datetime
 import pytest
 from pytest import approx
-from rota.shifts import InvalidShift, NoWorkers, RotaBuilder, SingleShift, days
+from rota_generator.shifts import InvalidShift, NoWorkers, RotaBuilder, SingleShift, days
 
-from rota.workers import Worker
+from rota_generator.workers import Worker
 
 import itertools
 

test/test_workers.py

@@ -1,9 +1,9 @@
 import pytest
-from rota.shifts import InvalidShift, MaxShiftsPerWeekConstraint, NoWorkers, PostShiftConstraint, RotaBuilder, SingleShift, WarningTermination, WorkerRequirement, days
+from rota_generator.shifts import InvalidShift, MaxShiftsPerWeekConstraint, NoWorkers, PostShiftConstraint, RotaBuilder, SingleShift, WarningTermination, WorkerRequirement, days
 
 import datetime
-from rota.workers import NonWorkingDays, NotAvailableToWork, Worker, generate_not_available_to_works
-from rota.workers import WorkRequests
+from rota_generator.workers import NonWorkingDays, NotAvailableToWork, Worker, generate_not_available_to_works
+from rota_generator.workers import WorkRequests
 from web.rota.shifts import PreShiftConstraint
 
 def generate_basic_rota(