1313 categories ,
1414 floats ,
1515 integers ,
16+ map_category ,
1617 relevant_var ,
1718 rename_mapping ,
1819 security_levels_mapping ,
@@ -77,7 +78,7 @@ def test_rename_enusc(enusc_raw):
7778def test_value_presence (enusc_raw_short , column , expected_value ):
7879 enusc_filtered = _filter_enusc (enusc_raw_short , relevant_var )
7980 enusc_renamed = _rename_enusc (enusc_filtered , rename_mapping )
80- enusc_category = _map_categories (enusc_renamed )
81+ enusc_category = _map_categories (enusc_renamed , map_category )
8182
8283 assert (
8384 expected_value in enusc_category [column ].values
@@ -91,7 +92,7 @@ def test_value_presence(enusc_raw_short, column, expected_value):
9192def test_column_is_categorical (enusc_raw_short , column ):
9293 enusc_filtered = _filter_enusc (enusc_raw_short , relevant_var )
9394 enusc_renamed = _rename_enusc (enusc_filtered , rename_mapping )
94- enusc_category = _map_categories (enusc_renamed )
95+ enusc_category = _map_categories (enusc_renamed , map_category )
9596
9697 assert isinstance (
9798 enusc_category [column ].dtype , pd .CategoricalDtype
@@ -109,7 +110,7 @@ def test_value_presence_continued(enusc_raw_short, column, values_dict):
109110 errors = []
110111 enusc_filtered = _filter_enusc (enusc_raw_short , relevant_var )
111112 enusc_renamed = _rename_enusc (enusc_filtered , rename_mapping )
112- enusc_category = _map_categories (enusc_renamed )
113+ enusc_category = _map_categories (enusc_renamed , map_category )
113114
114115 for value in enusc_category [column ]:
115116 if pd .isna (value ): # Skip missing values
@@ -125,14 +126,14 @@ def test_value_presence_continued(enusc_raw_short, column, values_dict):
125126def test_fail_age_not_categorical (enusc_raw_short ):
126127 enusc_filtered = _filter_enusc (enusc_raw_short , relevant_var )
127128 enusc_renamed = _rename_enusc (enusc_filtered , rename_mapping )
128- enusc_category = _map_categories (enusc_renamed )
129+ enusc_category = _map_categories (enusc_renamed , map_category )
129130 assert isinstance (enusc_category ["age" ].dtype , pd .CategoricalDtype )
130131
131132
132133def test_amount_of_missing (enusc_raw_short ):
133134 enusc_filtered = _filter_enusc (enusc_raw_short , relevant_var )
134135 enusc_renamed = _rename_enusc (enusc_filtered , rename_mapping )
135- enusc_category = _map_categories (enusc_renamed )
136+ enusc_category = _map_categories (enusc_renamed , map_category )
136137 enusc_filled = _fill_missing (enusc_category )
137138
138139 errors = []
@@ -168,7 +169,7 @@ def test_set_data_types_not_mapped_var(enusc_raw_short, column, expected_dtype):
168169 errors = []
169170 enusc_filtered = _filter_enusc (enusc_raw_short , relevant_var )
170171 enusc_renamed = _rename_enusc (enusc_filtered , rename_mapping )
171- enusc_category = _map_categories (enusc_renamed )
172+ enusc_category = _map_categories (enusc_renamed , map_category )
172173 enusc_filled = _fill_missing (enusc_category )
173174 enusc_dtypes = _set_data_types_not_mapped_var (enusc_filled )
174175
@@ -202,7 +203,7 @@ def test_fail_set_data_types_not_mapped_var(
202203 errors = []
203204 enusc_filtered = _filter_enusc (enusc_raw_short , relevant_var )
204205 enusc_renamed = _rename_enusc (enusc_filtered , rename_mapping )
205- enusc_category = _map_categories (enusc_renamed )
206+ enusc_category = _map_categories (enusc_renamed , map_category )
206207 enusc_filled = _fill_missing (enusc_category )
207208 enusc_dtypes = _set_data_types_not_mapped_var (enusc_filled )
208209
0 commit comments