diff --git a/src/conftest.py b/src/conftest.py index a26eb47..4d7d02a 100644 --- a/src/conftest.py +++ b/src/conftest.py @@ -7,7 +7,7 @@ def _collectstatic(): call_command("collectstatic", interactive=False, verbosity=0) -@pytest.fixture() +@pytest.fixture def live_server(settings, live_server): settings.STORAGES = { "default": { @@ -20,13 +20,13 @@ def live_server(settings, live_server): return live_server -@pytest.fixture() +@pytest.fixture def firefox_options(firefox_options): firefox_options.add_argument("-headless") return firefox_options -@pytest.fixture() +@pytest.fixture def selenium(selenium): selenium.implicitly_wait(3) selenium.set_window_size(3860, 2140) diff --git a/src/purchase/management/commands/generate_dummy_baskets.py b/src/purchase/management/commands/generate_dummy_baskets.py index fa0643d..ce3a9d9 100644 --- a/src/purchase/management/commands/generate_dummy_baskets.py +++ b/src/purchase/management/commands/generate_dummy_baskets.py @@ -40,10 +40,8 @@ class Command(BaseCommand): method = random.choices(payment_methods, weights=methods_weights)[0] basket = Basket.objects.create(payment_method=method) items_in_basket = int(random.normalvariate(3, 2)) - if items_in_basket > len(products): - items_in_basket = len(products) - if items_in_basket < 1: - items_in_basket = 1 + items_in_basket = min(items_in_basket, len(products)) + items_in_basket = max(items_in_basket, 1) rng = np.random.default_rng() selected_products = rng.choice( products,