fix smf test formula

Quantco · Jan 9, 2024 · 80316c1 · 80316c1
1 parent ab3d86f
commit 80316c1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/glm/test_glm.py b/tests/glm/test_glm.py
@@ -3121,7 +3121,8 @@ def test_formula_against_smf(get_mixed_data, formula, fit_intercept):
     else:
         beta_formula = model_formula.coef_
 
-    model_smf = smf.glm(formula, data, family=sm.families.Gaussian()).fit()
+    formula_smf = formula + "- 1" if not fit_intercept else formula
+    model_smf = smf.glm(formula_smf, data, family=sm.families.Gaussian()).fit()
 
     np.testing.assert_almost_equal(beta_formula, model_smf.params)