13 lines
227 B
Python
13 lines
227 B
Python
import pytest
|
|
|
|
import pandas as pd
|
|
|
|
|
|
@pytest.fixture
|
|
def df_from_dict():
|
|
def maker(dct, is_categorical=False):
|
|
df = pd.DataFrame(dct)
|
|
return df.astype("category") if is_categorical else df
|
|
|
|
return maker
|