From ecba479c7cd19b8288b1d2c3084951b1f26a7d3b Mon Sep 17 00:00:00 2001 From: Nicolas Mounet Date: Tue, 12 Jan 2021 19:19:41 +0100 Subject: [PATCH] Adding a test without breaks for the model generator --- cara/tests/apps/calculator/test_model_generator.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/cara/tests/apps/calculator/test_model_generator.py b/cara/tests/apps/calculator/test_model_generator.py index e0e307d4..328c8f04 100644 --- a/cara/tests/apps/calculator/test_model_generator.py +++ b/cara/tests/apps/calculator/test_model_generator.py @@ -303,6 +303,20 @@ def test_valid_no_lunch(baseline_form): assert baseline_form.validate() is None +def test_no_breaks(baseline_form): + # Check that the times are correct in the absence of breaks. + baseline_form.lunch_option = False + baseline_form.coffee_breaks = 0 + baseline_form.activity_start = 9 * 60 + baseline_form.activity_finish = 17 * 60 + baseline_form.infected_start = 10 * 60 + baseline_form.infected_finish = 15 * 60 + exposed_correct = ((9, 17),) + infected_correct = ((10, 15),) + assert baseline_form.exposed_present_interval().present_times == exposed_correct + assert baseline_form.infected_present_interval().present_times == infected_correct + + def test_coffee_lunch_breaks(baseline_form): baseline_form.coffee_duration = 30 baseline_form.coffee_breaks = 4