From 52500c582f66701771be01187ff0ad7dd77d093a Mon Sep 17 00:00:00 2001 From: afmagee42 Date: Wed, 4 Dec 2024 08:18:38 -0800 Subject: [PATCH] bonus easy test, vaccination should reduce R, and we know the right R --- tests/test_ngm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_ngm.py b/tests/test_ngm.py index 687406e..e4e5fe5 100644 --- a/tests/test_ngm.py +++ b/tests/test_ngm.py @@ -28,8 +28,8 @@ def test_vax_beta(): current = ngm.get_R(beta=beta, n=n, n_vax=n_vax, ve=ve) expected = np.array( [ - [10.0 * .2 * .5, 0.1 * .8], - [0.1 * .2 * .5, 1.0 * .8], + [10.0 * .2 * .5, 0.1 * .2 * .5], + [0.1 * .8, 1.0 * .8], ] )