added baseline model with sr

This commit is contained in:
Luis Aleixo 2024-05-21 11:03:59 +02:00
parent 04b8e91639
commit 10146f9a18

View file

@ -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)