Skip to content

Commit

Permalink
Update tests
Browse files Browse the repository at this point in the history
  • Loading branch information
s3alfisc committed Oct 16, 2024
1 parent 1cffa60 commit d4aa0b8
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions tests/test_vs_fixest.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down Expand Up @@ -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))
Expand Down Expand Up @@ -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))
Expand Down

0 comments on commit d4aa0b8

Please sign in to comment.