Fixed the tests

Author: jmafoster1

tests/estimation_tests/test_ipcw_estimator.py

@@ -33,7 +33,7 @@ def test_estimate_hazard_ratio(self):
             eligibility=None,
         )
         estimate, intervals = estimation_model.estimate_hazard_ratio()
-        self.assertEqual(estimate["trtrand"], 1.0)
+        self.assertEqual(round(estimate["trtrand"], 3), 1.936)
 
     def test_invalid_treatment_strategies(self):
         timesteps_per_intervention = 1

tests/testing_tests/test_causal_test_adequacy.py

@@ -144,11 +144,12 @@ def test_data_adequacy_group_by(self):
         causal_test_result = causal_test_case.execute_test(estimation_model, None)
         adequacy_metric = DataAdequacy(causal_test_case, estimation_model, group_by="id")
         adequacy_metric.measure_adequacy()
-        causal_test_result.adequacy = adequacy_metric
-        print(causal_test_result.adequacy.to_dict())
+        adequacy_dict = adequacy_metric.to_dict()
+        self.assertEqual(round(adequacy_dict["kurtosis"]["trtrand"], 3), -0.336)
+        adequacy_dict.pop("kurtosis")
         self.assertEqual(
-            causal_test_result.adequacy.to_dict(),
-            {"kurtosis": {"trtrand": 0.0}, "bootstrap_size": 100, "passing": 0, "successful": 95},
+            adequacy_dict,
+            {"bootstrap_size": 100, "passing": 28, "successful": 95},
         )
 
     def test_dag_adequacy_dependent(self):