codecov

Author: jmafoster1

tests/estimation_tests/test_ipcw_estimator.py

@@ -79,3 +79,47 @@ def test_invalid_fault_t_do(self):
         estimation_model.df["fault_t_do"] = 0
         with self.assertRaises(ValueError):
             estimate, intervals = estimation_model.estimate_hazard_ratio()
+
+    def test_no_individual_began_control_strategy(self):
+        timesteps_per_intervention = 1
+        control_strategy = [[t, "t", 0] for t in range(1, 4, timesteps_per_intervention)]
+        treatment_strategy = [[t, "t", 1] for t in range(1, 4, timesteps_per_intervention)]
+        outcome = "outcome"
+        fit_bl_switch_formula = "xo_t_do ~ time"
+        df = pd.read_csv("tests/resources/data/temporal_data.csv")
+        df["t"] = 1
+        df["ok"] = df["outcome"] == 1
+        with self.assertRaises(ValueError):
+            estimation_model = IPCWEstimator(
+                df,
+                timesteps_per_intervention,
+                control_strategy,
+                treatment_strategy,
+                outcome,
+                "ok",
+                fit_bl_switch_formula=fit_bl_switch_formula,
+                fit_bltd_switch_formula=fit_bl_switch_formula,
+                eligibility=None,
+            )
+
+    def test_no_individual_began_treatment_strategy(self):
+        timesteps_per_intervention = 1
+        control_strategy = [[t, "t", 0] for t in range(1, 4, timesteps_per_intervention)]
+        treatment_strategy = [[t, "t", 1] for t in range(1, 4, timesteps_per_intervention)]
+        outcome = "outcome"
+        fit_bl_switch_formula = "xo_t_do ~ time"
+        df = pd.read_csv("tests/resources/data/temporal_data.csv")
+        df["t"] = 0
+        df["ok"] = df["outcome"] == 1
+        with self.assertRaises(ValueError):
+            estimation_model = IPCWEstimator(
+                df,
+                timesteps_per_intervention,
+                control_strategy,
+                treatment_strategy,
+                outcome,
+                "ok",
+                fit_bl_switch_formula=fit_bl_switch_formula,
+                fit_bltd_switch_formula=fit_bl_switch_formula,
+                eligibility=None,
+            )