bonus easy test, vaccination should reduce R, and we know the right R

CDCgov · Dec 4, 2024 · 52500c5 · 52500c5
1 parent 8e6d602
commit 52500c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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],
         ]
     )