HK3-Lab-Team / pytrousse

PyTrousse collects into one toolbox a set of data wrangling procedures tailored for composing reproducible analytics pipelines.
Apache License 2.0
0 stars 1 forks source link

Test column list by type #8

Closed lorenz-gorini closed 4 years ago

lorenz-gorini commented 4 years ago

Added 2 tests for methods in DataFrameWithInfo class: _get_categorical_cols -> Testing how categorical columns are identified column_list_by_type -> Testing if this property returns appropriate output and result

To this purpose, created 1 mock df_categorical_cols and modified df_multi_type (to accept "sample_size" attribute since categorical columns could not be distinguished in a DataFrame with few samples).