Codecov

Author: jmafoster1

tests/estimation_tests/test_logistic_regression_estimator.py

@@ -21,3 +21,11 @@ def test_odds_ratio(self):
         )
         effect_estimate = logistic_regression_estimator.estimate_unit_odds_ratio()
         self.assertEqual(round(effect_estimate.value.iloc[0], 4), 0.8948)
+
+    def test_odds_ratio_data(self):
+        df = self.scarf_df.copy()
+        logistic_regression_estimator = LogisticRegressionEstimator(
+            BaseTestCase(Input("length_in", float), Output("completed", bool)), 65, 55, set()
+        )
+        effect_estimate = logistic_regression_estimator.estimate_unit_odds_ratio(df)
+        self.assertEqual(round(effect_estimate.value.iloc[0], 4), 0.8948)