tests pass again

Author: jmafoster1

tests/estimation_tests/test_ipcw_estimator.py

@@ -15,8 +15,8 @@ class TestIPCWEstimator(unittest.TestCase):
 
     def test_estimate_hazard_ratio(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)]
+        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")
@@ -37,8 +37,8 @@ def test_estimate_hazard_ratio(self):
 
     def test_invalid_treatment_strategies(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)]
+        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")
@@ -59,8 +59,8 @@ def test_invalid_treatment_strategies(self):
 
     def test_invalid_fault_t_do(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)]
+        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")

tests/testing_tests/test_causal_test_adequacy.py

@@ -111,8 +111,8 @@ def test_data_adequacy_cateogorical(self):
 
     def test_data_adequacy_group_by(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)]
+        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")