From d4aa0b8f464de74852e8f881360332a4c0f76da7 Mon Sep 17 00:00:00 2001 From: Alexander Fischer Date: Wed, 16 Oct 2024 21:00:17 +0200 Subject: [PATCH] Update tests --- tests/test_vs_fixest.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/test_vs_fixest.py b/tests/test_vs_fixest.py index 72b755fb..b0218d8d 100644 --- a/tests/test_vs_fixest.py +++ b/tests/test_vs_fixest.py @@ -11,6 +11,7 @@ import pyfixest as pf from pyfixest.estimation.estimation import feols +from pyfixest.estimation.FixestMulti_ import FixestMulti from pyfixest.utils.set_rpy2_path import update_r_paths from pyfixest.utils.utils import get_data, ssc @@ -607,6 +608,7 @@ def test_multi_fit(N, seed, beta_type, error_type, dropna, fml_multi): try: pyfixest = feols(fml=fml_multi, data=data) + assert isinstance(pyfixest, FixestMulti) except ValueError as e: if "is not of type 'O' or 'category'" in str(e): data["f1"] = pd.Categorical(data.f1.astype(str)) @@ -680,6 +682,7 @@ def test_split_fit(N, seed, beta_type, error_type, dropna, fml_multi, split, fsp try: pyfixest = feols(fml=fml_multi, data=data, split=split, fsplit=fsplit) + assert isinstance(pyfixest, FixestMulti) except ValueError as e: if "is not of type 'O' or 'category'" in str(e): data["f1"] = pd.Categorical(data.f1.astype(str))