fix test_convert_dtypes

jorisvandenbossche · jorisvandenbossche · commit c33e14af9fc7 · 2024-07-29T18:15:14.000+02:00
diff --git a/pandas/tests/frame/methods/test_convert_dtypes.py b/pandas/tests/frame/methods/test_convert_dtypes.py
@@ -19,16 +19,13 @@ def test_convert_dtypes(
         if using_infer_string:
             string_storage = "pyarrow_numpy"
 
-        string_storage_option = string_storage
-        if string_storage == "pyarrow_numpy":
-            string_storage_option = "pyarrow"
         df = pd.DataFrame(
             {
                 "a": pd.Series([1, 2, 3], dtype=np.dtype("int32")),
                 "b": pd.Series(["x", "y", "z"], dtype=np.dtype("O")),
             }
         )
-        with pd.option_context("string_storage", string_storage_option):
+        with pd.option_context("string_storage", string_storage):
             result = df.convert_dtypes(True, True, convert_integer, False)
         expected = pd.DataFrame(
             {

Original file line number	Diff line number	Diff line change
`@@ -19,16 +19,13 @@ def test_convert_dtypes(`
`19`	`19`	`if using_infer_string:`
`20`	`20`	`string_storage = "pyarrow_numpy"`
`21`	`21`
`22`		`- string_storage_option = string_storage`
`23`		`- if string_storage == "pyarrow_numpy":`
`24`		`- string_storage_option = "pyarrow"`
`25`	`22`	`df = pd.DataFrame(`
`26`	`23`	`{`
`27`	`24`	`"a": pd.Series([1, 2, 3], dtype=np.dtype("int32")),`
`28`	`25`	`"b": pd.Series(["x", "y", "z"], dtype=np.dtype("O")),`
`29`	`26`	`}`
`30`	`27`	`)`
`31`		`- with pd.option_context("string_storage", string_storage_option):`
	`28`	`+ with pd.option_context("string_storage", string_storage):`
`32`	`29`	`result = df.convert_dtypes(True, True, convert_integer, False)`
`33`	`30`	`expected = pd.DataFrame(`
`34`	`31`	`{`