test coefficient estimation

Author: jmafoster1

tests/testing_tests/test_causal_test_engine.py

@@ -178,6 +178,22 @@ def test_execute_test_observational_linear_regression_estimator_direct_effect(se
         causal_test_result = causal_test_engine.execute_test(estimation_model, causal_test_case)
         self.assertAlmostEqual(causal_test_result.test_value.value, 4, delta=1e-10)
 
+    def test_execute_test_observational_linear_regression_estimator_coefficient(self):
+        """Check that executing the causal test case returns the correct results for dummy data using a linear
+        regression estimator."""
+        estimation_model = LinearRegressionEstimator(
+            "D",
+            self.treatment_value,
+            self.control_value,
+            self.minimal_adjustment_set,
+            "A",
+            self.causal_test_engine.scenario_execution_data_df,
+        )
+        causal_test_result = self.causal_test_engine.execute_test(
+            estimation_model, self.causal_test_case, estimate_type="coefficient"
+        )
+        self.assertEqual(int(causal_test_result.test_value.value), 0)
+
     def test_execute_test_observational_linear_regression_estimator_risk_ratio(self):
         """Check that executing the causal test case returns the correct results for dummy data using a linear
         regression estimator."""