parametrize test

Author: pedroreys

Diff for: pandas/tests/io/test_pickle.py

@@ -382,14 +382,23 @@ def test_read(self, protocol, get_random_path):
             tm.assert_frame_equal(df, df2)
 
 
-def test_unicode_decode_error(datapath):
+@pytest.mark.parametrize(
+    ["pickle_file", "excols"],
+    [
+        ("test_py27.pkl", pd.Index(["a", "b", "c"])),
+        (
+            "test_mi_py27.pkl",
+            pd.MultiIndex.from_arrays([["a", "b", "c"], ["A", "B", "C"]]),
+        ),
+    ],
+)
+def test_unicode_decode_error(datapath, pickle_file, excols):
     # pickle file written with py27, should be readable without raising
-    #  UnicodeDecodeError, see GH#28645
-    path = datapath("io", "data", "pickle", "test_py27.pkl")
+    #  UnicodeDecodeError, see GH#28645 and GH#31988
+    path = datapath("io", "data", "pickle", pickle_file)
     df = pd.read_pickle(path)
 
     # just test the columns are correct since the values are random
-    excols = pd.Index(["a", "b", "c"])
     tm.assert_index_equal(df.columns, excols)
 
 
@@ -501,15 +510,3 @@ def test_read_pickle_with_subclass():
 
     tm.assert_series_equal(result[0], expected[0])
     assert isinstance(result[1], MyTz)
-
-
-def test_read_py27_pickle_with_MultiIndex_column(datapath):
-    # pickle file with MultiIndex column written with py27
-    # should be readable without raising UnicodeDecodeError
-    # see GH#31988
-    path = datapath("io", "data", "pickle", "test_mi_py27.pkl")
-    df = pd.read_pickle(path)
-
-    # just test the columns are correct since the values are random
-    expected = pd.MultiIndex.from_arrays([["a", "b", "c"], ["A", "B", "C"]])
-    tm.assert_index_equal(df.columns, expected)