File tree Expand file tree Collapse file tree 1 file changed +10
-0
lines changed Expand file tree Collapse file tree 1 file changed +10
-0
lines changed Original file line number Diff line number Diff line change
1
+ # -*- coding: utf8 -*-
2
+
1
3
import pytest
2
4
from pkg_resources import parse_version
3
5
@@ -128,6 +130,14 @@ def test_transformed_names_binarizer(complex_dataframe):
128
130
assert mapper .transformed_names_ == ['target_a' , 'target_b' , 'target_c' ]
129
131
130
132
133
+ def test_transformed_names_binarizer_unicode ():
134
+ df = pd .DataFrame ({'target' : [u'ñ' , u'á' , u'é' ]})
135
+ mapper = DataFrameMapper ([('target' , LabelBinarizer ())])
136
+ mapper .fit_transform (df )
137
+ expected_names = {u'target_ñ' , u'target_á' , u'target_é' }
138
+ assert set (mapper .transformed_names_ ) == expected_names
139
+
140
+
131
141
def test_transformed_names_transformers_list (complex_dataframe ):
132
142
"""
133
143
When using a list of transformers, use them in inverse order to get the
You can’t perform that action at this time.
0 commit comments