use option_context

Author: hippowm

pandas/tests/strings/test_strings.py

@@ -11,7 +11,7 @@
     Index,
     MultiIndex,
     Series,
-    set_option,
+    option_context,
 )
 import pandas._testing as tm
 from pandas.core.strings.accessor import StringMethods
@@ -783,8 +783,8 @@ def test_series_str_decode():
 
 def test_decode_with_dtype_none():
     # Ensure that future.infer_string is enabled
-    set_option("future.infer_string", True)
-    ser = Series([b"a", b"b", b"c"])  # Use byte strings
-    result = ser.str.decode("utf-8", dtype=None)
-    expected = Series(["a", "b", "c"], dtype="str")
-    tm.assert_series_equal(result, expected)
+    with option_context("future.infer_string", True):
+        ser = Series([b"a", b"b", b"c"])  # Use byte strings
+        result = ser.str.decode("utf-8", dtype=None)
+        expected = Series(["a", "b", "c"], dtype="str")
+        tm.assert_series_equal(result, expected)