Skip to content

Commit 5f71404

Browse files
committed
TST: get_dummies UnicodeEncodeError tests
1 parent c272c52 commit 5f71404

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

pandas/tests/reshape/test_reshape.py

+20
Original file line numberDiff line numberDiff line change
@@ -302,6 +302,26 @@ def test_dataframe_dummies_with_categorical(self, df, sparse, dtype):
302302
expected.sort_index(axis=1)
303303
assert_frame_equal(result, expected)
304304

305+
def test_dataframe_dummies_unicode(self):
306+
df = pd.DataFrame(({u'ä': ['a']}))
307+
result = get_dummies(df)
308+
expected = pd.DataFrame({u'ä_a': [1]}, dtype=np.uint8)
309+
assert_frame_equal(result, expected)
310+
311+
df = pd.DataFrame({'x': [u'ä']})
312+
result = pd.get_dummies(df)
313+
expected = pd.DataFrame({u'x_ä': [1]}, dtype=np.uint8)
314+
assert_frame_equal(result, expected)
315+
316+
df = pd.DataFrame({'x': ['a']})
317+
result = pd.get_dummies(df, prefix=u'ä')
318+
expected = pd.DataFrame({u'ä_a': [1]}, dtype=np.uint8)
319+
assert_frame_equal(result, expected)
320+
321+
result = pd.get_dummies(df, prefix_sep=u'ä')
322+
expected = pd.DataFrame({u'xäa': [1]}, dtype=np.uint8)
323+
assert_frame_equal(result, expected)
324+
305325
def test_basic_drop_first(self, sparse):
306326
# GH12402 Add a new parameter `drop_first` to avoid collinearity
307327
# Basic case

0 commit comments

Comments
 (0)