From 10146f9a184d57933233c17570c17dfebbfd554b Mon Sep 17 00:00:00 2001 From: Luis Aleixo Date: Tue, 21 May 2024 11:03:59 +0200 Subject: [PATCH] added baseline model with sr --- caimira/tests/apps/calculator/conftest.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/caimira/tests/apps/calculator/conftest.py b/caimira/tests/apps/calculator/conftest.py index 38c19e97..59ce26db 100644 --- a/caimira/tests/apps/calculator/conftest.py +++ b/caimira/tests/apps/calculator/conftest.py @@ -11,3 +11,12 @@ def baseline_form_data(): @pytest.fixture def baseline_form(baseline_form_data, data_registry): return model_generator.VirusFormData.from_dict(baseline_form_data, data_registry) + + +@pytest.fixture +def baseline_form_with_sr(baseline_form_data, data_registry): + form_data_sr = baseline_form_data + form_data_sr['short_range_option'] = 'short_range_yes' + form_data_sr['short_range_interactions'] = '[{"expiration": "Shouting", "start_time": "10:30", "duration": "30"}]' + form_data_sr['short_range_total_people'] = 5 + return model_generator.VirusFormData.from_dict(form_data_sr, data_registry) \ No newline at end of file