Skip to content

Commit 0fe52e1

Browse files
committed
add test for dtype argument in str.decode
1 parent 1d153bb commit 0fe52e1

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

pandas/tests/strings/test_strings.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
Index,
1212
MultiIndex,
1313
Series,
14+
set_option
1415
)
1516
import pandas._testing as tm
1617
from pandas.core.strings.accessor import StringMethods
@@ -778,3 +779,11 @@ def test_series_str_decode():
778779
result = Series([b"x", b"y"]).str.decode(encoding="UTF-8", errors="strict")
779780
expected = Series(["x", "y"], dtype="str")
780781
tm.assert_series_equal(result, expected)
782+
783+
def test_decode_with_dtype_none():
784+
# Ensure that future.infer_string is enabled
785+
set_option('future.infer_string', True)
786+
ser = Series([b'a', b'b', b'c']) # Use byte strings
787+
result = ser.str.decode('utf-8', dtype=None)
788+
expected = Series(['a', 'b', 'c'], dtype='str')
789+
tm.assert_series_equal(result, expected)

0 commit comments

Comments
 (0)