From 14d9bf3a4d343b7c9fb1f80b51b9a0b6ec78c433 Mon Sep 17 00:00:00 2001
From: Timothy Willard <9395586+TimothyWillard@users.noreply.github.com>
Date: Tue, 10 Dec 2024 11:49:50 -0500
Subject: [PATCH] Simplify poisson LL tests with only one variable

---
 .../tests/statistics/test_statistic_class.py       | 14 +++-----------
 1 file changed, 3 insertions(+), 11 deletions(-)

diff --git a/flepimop/gempyor_pkg/tests/statistics/test_statistic_class.py b/flepimop/gempyor_pkg/tests/statistics/test_statistic_class.py
index c0397ecb2..008b5073d 100644
--- 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