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
|