Simplify poisson LL tests with only one variable

HopkinsIDD · Dec 10, 2024 · 14d9bf3 · 14d9bf3
1 parent 10e5d98
commit 14d9bf3
Showing 1 changed file with 3 additions and 11 deletions.
diff --git a/flepimop/gempyor_pkg/tests/statistics/test_statistic_class.py b/flepimop/gempyor_pkg/tests/statistics/test_statistic_class.py
@@ -240,10 +240,6 @@ def simple_valid_factory_with_pois() -> MockStatisticInput:
                 list(data_coords.keys()),
                 np.random.poisson(lam=20.0, size=data_dim),
             ),
-            "incidD": (
-                list(data_coords.keys()),
-                np.random.poisson(lam=20.0, size=data_dim),
-            ),
         },
         coords=data_coords,
     )
@@ -253,10 +249,6 @@ def simple_valid_factory_with_pois() -> MockStatisticInput:
                 list(data_coords.keys()),
                 np.random.poisson(lam=20.0, size=data_dim),
             ),
-            "incidD": (
-                list(data_coords.keys()),
-                np.random.poisson(lam=20.0, size=data_dim),
-            ),
         },
         coords=data_coords,
     )
@@ -287,10 +279,10 @@ def simple_valid_factory_with_pois_with_some_zeros() -> MockStatisticInput:
         }
     ] = 0
 
-    mock_input.gt_data["incidD"].loc[
+    mock_input.gt_data["incidH"].loc[
         {
-            "date": mock_input.gt_data.coords["date"][0],
-            "subpop": mock_input.gt_data.coords["subpop"][0],
+            "date": mock_input.gt_data.coords["date"][2],
+            "subpop": mock_input.gt_data.coords["subpop"][2],
         }
     ] = 0