relaxing tolerances on conditional prob. test

This commit is contained in:
Nicolas Mounet 2023-09-01 16:18:10 +02:00
parent f8530d4724
commit 82f4e14905

View file

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