diff --git a/caimira/tests/test_conditional_probability.py b/caimira/tests/test_conditional_probability.py index 0a76899c..ec5fcbdd 100644 --- a/caimira/tests/test_conditional_probability.py +++ b/caimira/tests/test_conditional_probability.py @@ -43,7 +43,7 @@ def baseline_exposure_model(): ) -@retry(tries=2) +@retry(tries=3) def test_conditional_prob_inf_given_vl_dist(baseline_exposure_model): viral_loads = np.array([3., 5., 7., 9.,]) @@ -72,6 +72,6 @@ def test_conditional_prob_inf_given_vl_dist(baseline_exposure_model): report_generator.conditional_prob_inf_given_vl_dist(infection_probability, viral_loads, specific_vl, step) ) - assert np.allclose(actual_pi_means, expected_pi_means, atol=0.001) - assert np.allclose(actual_lower_percentiles, expected_lower_percentiles, atol=0.001) - assert np.allclose(actual_upper_percentiles, expected_upper_percentiles, atol=0.001) + assert np.allclose(actual_pi_means, expected_pi_means, atol=0.002) + assert np.allclose(actual_lower_percentiles, expected_lower_percentiles, atol=0.002) + assert np.allclose(actual_upper_percentiles, expected_upper_percentiles, atol=0.002)