fix npi test

HopkinsIDD · Nov 13, 2024 · 1bf2f16 · 1bf2f16
1 parent dd9c00f
commit 1bf2f16
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/flepimop/gempyor_pkg/tests/npi/test_SinglePeriodModifier.py b/flepimop/gempyor_pkg/tests/npi/test_SinglePeriodModifier.py
@@ -30,15 +30,16 @@ def test_SinglePeriodModifier_success(self):
 
         test = NPI.SinglePeriodModifier(
             npi_config=s.npi_config_seir,
-            modinf=s,
+            modinf_ti=s.ti,
+            modinf_tf=s.tf,
             modifiers_library="",
             subpops=s.subpop_struct.subpop_names,
             loaded_df=None,
         )
         """
         test2 = NPI.SinglePeriodModifier(
             npi_config=s.npi_config_seir,
-            modinf=s,
+            modinf_ti=s.ti, modinf_tf=s.tf,
             modifiers_library="",
             subpops=s.subpop_struct.subpop_names,
             loaded_df=test.parameters,
@@ -64,7 +65,8 @@ def test_SinglePeriodModifier_start_date_fail(self):
 
             test = NPI.SinglePeriodModifier(
                 npi_config=s.npi_config_seir,
-                modinf=s,
+                modinf_ti=s.ti,
+                modinf_tf=s.tf,
                 modifiers_library="",
                 subpops=s.subpop_struct.subpop_names,
                 loaded_df=None,
@@ -89,7 +91,8 @@ def test_SinglePeriodModifier_end_date_fail(self):
 
             test = NPI.SinglePeriodModifier(
                 npi_config=s.npi_config_seir,
-                modinf=s,
+                modinf_ti=s.ti,
+                modinf_tf=s.tf,
                 modifiers_library="",
                 subpops=s.subpop_struct.subpop_names,
                 loaded_df=None,
@@ -110,7 +113,8 @@ def test_SinglePeriodModifier_checkerrors(self):
 
         test = NPI.SinglePeriodModifier(
             npi_config=s.npi_config_seir,
-            modinf=s,
+            modinf_ti=s.ti,
+            modinf_tf=s.tf,
             modifiers_library="",
             subpops=s.subpop_struct.subpop_names,
             loaded_df=None,