Added extra test

Author: jmafoster1

tests/estimation_tests/test_multinomial_regression_estimator.py

@@ -23,6 +23,15 @@ def test_odds_ratio(self):
         effect_estimate = multinomial_regression_estimator.estimate_unit_odds_ratio()
         self.assertEqual(round(effect_estimate.value.iloc[0], 4), 0.8948)
 
+    def test_odds_ratio_category(self):
+        df = self.scarf_df.copy()
+        multinomial_regression_estimator = MultinomialRegressionEstimator(
+            BaseTestCase(Input("length_in", float), Output("color", bool)), 65, 55, set(), df
+        )
+        effect_estimate = multinomial_regression_estimator.estimate_unit_odds_ratio()
+        print(effect_estimate.value)
+        self.assertTrue(effect_estimate.value.round(4).equals, pd.Series({"grey": 1.0072, "orange": 0.9668}))
+
     def test_odds_ratio_data(self):
         df = self.scarf_df.copy()
         multinomial_regression_estimator = MultinomialRegressionEstimator(