add messages to tests

Author: Raisa Dzhamtyrova

pandas/tests/extension/decimal/test_decimal.py

@@ -146,9 +146,10 @@ class TestMissing(BaseDecimal, base.BaseMissingTests):
 
 class Reduce:
     def check_reduce(self, s, op_name, skipna):
-
         if op_name in ["median", "skew", "kurt"]:
-            with pytest.raises(NotImplementedError):
+            # breakpoint()
+            msg = r"decimal does not support the .* operation"
+            with pytest.raises(NotImplementedError, match=msg):
                 getattr(s, op_name)(skipna=skipna)
 
         else:

pandas/tests/extension/json/test_json.py

@@ -136,10 +136,11 @@ def test_custom_asserts(self):
         self.assert_frame_equal(a.to_frame(), a.to_frame())
 
         b = pd.Series(data.take([0, 0, 1]))
-        with pytest.raises(AssertionError):
+        msg = r"ExtensionArray are different.*"
+        with pytest.raises(AssertionError, match=msg):
             self.assert_series_equal(a, b)
 
-        with pytest.raises(AssertionError):
+        with pytest.raises(AssertionError, match=msg):
             self.assert_frame_equal(a.to_frame(), b.to_frame())
 
 

pandas/tests/extension/test_boolean.py

@@ -112,7 +112,8 @@ def _check_op(self, s, op, other, op_name, exc=NotImplementedError):
                 # subtraction for bools raises TypeError (but not yet in 1.13)
                 if _np_version_under1p14:
                     pytest.skip("__sub__ does not yet raise in numpy 1.13")
-                with pytest.raises(TypeError):
+                msg = r"numpy boolean subtract, the \`-\` operator, is not supported.*"
+                with pytest.raises(TypeError, match=msg):
                     op(s, other)
 
                 return
@@ -139,7 +140,7 @@ def _check_op(self, s, op, other, op_name, exc=NotImplementedError):
                 expected[result.isna()] = np.nan
             self.assert_series_equal(result, expected)
         else:
-            with pytest.raises(exc):
+            with pytest.raises(exc, match=msg):
                 op(s, other)
 
     def _check_divmod_op(self, s, op, other, exc=None):

pandas/tests/extension/test_categorical.py

@@ -278,7 +278,8 @@ def _compare_other(self, s, data, op_name, other):
             assert (result == expected).all()
 
         else:
-            with pytest.raises(TypeError):
+            msg = "Unordered Categoricals can only compare equality or not"
+            with pytest.raises(TypeError, match=msg):
                 op(data, other)
 
 

pandas/tests/frame/indexing/test_categorical.py

@@ -115,7 +115,11 @@ def test_assigning_ops(self):
         tm.assert_frame_equal(df, exp_single_cats_value)
 
         #   - assign a single value not in the current categories set
-        with pytest.raises(ValueError):
+        msg = (
+            r"(:?Cannot setitem on a Categorical with a new category.*)"
+            r"|(:?Cannot set a Categorical with another, without identical categories)"
+        )
+        with pytest.raises(ValueError, match=msg):
             df = orig.copy()
             df.iloc[2, 0] = "c"
 
@@ -125,7 +129,7 @@ def test_assigning_ops(self):
         tm.assert_frame_equal(df, exp_single_row)
 
         #   - assign a complete row (mixed values) not in categories set
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             df = orig.copy()
             df.iloc[2, :] = ["c", 2]
 
@@ -134,7 +138,7 @@ def test_assigning_ops(self):
         df.iloc[2:4, :] = [["b", 2], ["b", 2]]
         tm.assert_frame_equal(df, exp_multi_row)
 
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             df = orig.copy()
             df.iloc[2:4, :] = [["c", 2], ["c", 2]]
 
@@ -144,12 +148,12 @@ def test_assigning_ops(self):
         df.iloc[2:4, 0] = Categorical(["b", "b"], categories=["a", "b"])
         tm.assert_frame_equal(df, exp_parts_cats_col)
 
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             # different categories -> not sure if this should fail or pass
             df = orig.copy()
             df.iloc[2:4, 0] = Categorical(list("bb"), categories=list("abc"))
 
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             # different values
             df = orig.copy()
             df.iloc[2:4, 0] = Categorical(list("cc"), categories=list("abc"))
@@ -160,7 +164,7 @@ def test_assigning_ops(self):
         df.iloc[2:4, 0] = ["b", "b"]
         tm.assert_frame_equal(df, exp_parts_cats_col)
 
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             df.iloc[2:4, 0] = ["c", "c"]
 
         #  loc
@@ -175,7 +179,7 @@ def test_assigning_ops(self):
         tm.assert_frame_equal(df, exp_single_cats_value)
 
         #   - assign a single value not in the current categories set
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             df = orig.copy()
             df.loc["j", "cats"] = "c"
 
@@ -185,7 +189,7 @@ def test_assigning_ops(self):
         tm.assert_frame_equal(df, exp_single_row)
 
         #   - assign a complete row (mixed values) not in categories set
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             df = orig.copy()
             df.loc["j", :] = ["c", 2]
 
@@ -194,7 +198,7 @@ def test_assigning_ops(self):
         df.loc["j":"k", :] = [["b", 2], ["b", 2]]
         tm.assert_frame_equal(df, exp_multi_row)
 
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             df = orig.copy()
             df.loc["j":"k", :] = [["c", 2], ["c", 2]]
 
@@ -204,14 +208,14 @@ def test_assigning_ops(self):
         df.loc["j":"k", "cats"] = Categorical(["b", "b"], categories=["a", "b"])
         tm.assert_frame_equal(df, exp_parts_cats_col)
 
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             # different categories -> not sure if this should fail or pass
             df = orig.copy()
             df.loc["j":"k", "cats"] = Categorical(
                 ["b", "b"], categories=["a", "b", "c"]
             )
 
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             # different values
             df = orig.copy()
             df.loc["j":"k", "cats"] = Categorical(
@@ -224,7 +228,7 @@ def test_assigning_ops(self):
         df.loc["j":"k", "cats"] = ["b", "b"]
         tm.assert_frame_equal(df, exp_parts_cats_col)
 
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             df.loc["j":"k", "cats"] = ["c", "c"]
 
         #  loc
@@ -239,7 +243,7 @@ def test_assigning_ops(self):
         tm.assert_frame_equal(df, exp_single_cats_value)
 
         #   - assign a single value not in the current categories set
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             df = orig.copy()
             df.loc["j", df.columns[0]] = "c"
 
@@ -249,7 +253,7 @@ def test_assigning_ops(self):
         tm.assert_frame_equal(df, exp_single_row)
 
         #   - assign a complete row (mixed values) not in categories set
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             df = orig.copy()
             df.loc["j", :] = ["c", 2]
 
@@ -258,7 +262,7 @@ def test_assigning_ops(self):
         df.loc["j":"k", :] = [["b", 2], ["b", 2]]
         tm.assert_frame_equal(df, exp_multi_row)
 
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             df = orig.copy()
             df.loc["j":"k", :] = [["c", 2], ["c", 2]]
 
@@ -268,14 +272,14 @@ def test_assigning_ops(self):
         df.loc["j":"k", df.columns[0]] = Categorical(["b", "b"], categories=["a", "b"])
         tm.assert_frame_equal(df, exp_parts_cats_col)
 
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             # different categories -> not sure if this should fail or pass
             df = orig.copy()
             df.loc["j":"k", df.columns[0]] = Categorical(
                 ["b", "b"], categories=["a", "b", "c"]
             )
 
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             # different values
             df = orig.copy()
             df.loc["j":"k", df.columns[0]] = Categorical(
@@ -288,7 +292,7 @@ def test_assigning_ops(self):
         df.loc["j":"k", df.columns[0]] = ["b", "b"]
         tm.assert_frame_equal(df, exp_parts_cats_col)
 
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             df.loc["j":"k", df.columns[0]] = ["c", "c"]
 
         # iat
@@ -297,7 +301,7 @@ def test_assigning_ops(self):
         tm.assert_frame_equal(df, exp_single_cats_value)
 
         #   - assign a single value not in the current categories set
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             df = orig.copy()
             df.iat[2, 0] = "c"
 
@@ -308,7 +312,7 @@ def test_assigning_ops(self):
         tm.assert_frame_equal(df, exp_single_cats_value)
 
         #   - assign a single value not in the current categories set
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             df = orig.copy()
             df.at["j", "cats"] = "c"
 
@@ -332,7 +336,7 @@ def test_assigning_ops(self):
         df.at["j", "cats"] = "b"
         tm.assert_frame_equal(df, exp_single_cats_value)
 
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match=msg):
             df = orig.copy()
             df.at["j", "cats"] = "c"
 

pandas/tests/frame/indexing/test_indexing.py

@@ -481,7 +481,8 @@ def test_setitem(self, float_frame):
         # so raise/warn
         smaller = float_frame[:2]
 
-        with pytest.raises(com.SettingWithCopyError):
+        msg = r"\nA value is trying to be set on a copy of a slice from a DataFrame.*"
+        with pytest.raises(com.SettingWithCopyError, match=msg):
             smaller["col10"] = ["1", "2"]
 
         assert smaller["col10"].dtype == np.object_
@@ -865,7 +866,8 @@ def test_fancy_getitem_slice_mixed(self, float_frame, float_string_frame):
         # setting it triggers setting with copy
         sliced = float_frame.iloc[:, -3:]
 
-        with pytest.raises(com.SettingWithCopyError):
+        msg = r"\nA value is trying to be set on a copy of a slice from a DataFrame.*"
+        with pytest.raises(com.SettingWithCopyError, match=msg):
             sliced["C"] = 4.0
 
         assert (float_frame["C"] == 4).all()
@@ -992,7 +994,7 @@ def test_getitem_setitem_fancy_exceptions(self, float_frame):
         with pytest.raises(IndexingError, match="Too many indexers"):
             ix[:, :, :]
 
-        with pytest.raises(IndexingError):
+        with pytest.raises(IndexingError, match="Too many indexers"):
             ix[:, :, :] = 1
 
     def test_getitem_setitem_boolean_misaligned(self, float_frame):
@@ -1071,10 +1073,10 @@ def test_getitem_setitem_float_labels(self):
 
         cp = df.copy()
 
-        with pytest.raises(TypeError):
+        with pytest.raises(TypeError, match=msg):
             cp.iloc[1.0:5] = 0
 
-        with pytest.raises(TypeError):
+        with pytest.raises(TypeError, match=msg):
             result = cp.iloc[1.0:5] == 0  # noqa
 
         assert result.values.all()
@@ -1470,7 +1472,8 @@ def test_iloc_row(self):
 
         # verify slice is view
         # setting it makes it raise/warn
-        with pytest.raises(com.SettingWithCopyError):
+        msg = r"\nA value is trying to be set on a copy of a slice from a DataFrame.*"
+        with pytest.raises(com.SettingWithCopyError, match=msg):
             result[2] = 0.0
 
         exp_col = df[2].copy()
@@ -1501,7 +1504,8 @@ def test_iloc_col(self):
 
         # verify slice is view
         # and that we are setting a copy
-        with pytest.raises(com.SettingWithCopyError):
+        msg = r"\nA value is trying to be set on a copy of a slice from a DataFrame.*"
+        with pytest.raises(com.SettingWithCopyError, match=msg):
             result[8] = 0.0
 
         assert (df[8] == 0).all()

pandas/tests/frame/indexing/test_where.py

@@ -50,7 +50,8 @@ def _check_get(df, cond, check_dtypes=True):
         # check getting
         df = where_frame
         if df is float_string_frame:
-            with pytest.raises(TypeError):
+            msg = "'>' not supported between instances of 'str' and 'int'"
+            with pytest.raises(TypeError, match=msg):
                 df > 0
             return
         cond = df > 0
@@ -114,7 +115,8 @@ def _check_align(df, cond, other, check_dtypes=True):
 
         df = where_frame
         if df is float_string_frame:
-            with pytest.raises(TypeError):
+            msg = "'>' not supported between instances of 'str' and 'int'"
+            with pytest.raises(TypeError, match=msg):
                 df > 0
             return
 
@@ -172,7 +174,8 @@ def _check_set(df, cond, check_dtypes=True):
 
         df = where_frame
         if df is float_string_frame:
-            with pytest.raises(TypeError):
+            msg = "'>' not supported between instances of 'str' and 'int'"
+            with pytest.raises(TypeError, match=msg):
                 df > 0
             return
 
@@ -358,7 +361,8 @@ def test_where_datetime(self):
         )
 
         stamp = datetime(2013, 1, 3)
-        with pytest.raises(TypeError):
+        msg = "'>' not supported between instances of 'float' and 'datetime.datetime'"
+        with pytest.raises(TypeError, match=msg):
             df > stamp
 
         result = df[df.iloc[:, :-1] > stamp]

pandas/tests/frame/methods/test_diff.py

@@ -74,7 +74,7 @@ def test_diff_datetime_axis1(self, tz):
             )
             tm.assert_frame_equal(result, expected)
         else:
-            with pytest.raises(NotImplementedError):
+            with pytest.raises(NotImplementedError, match=""):
                 result = df.diff(axis=1)
 
     def test_diff_timedelta(self):

pandas/tests/frame/methods/test_explode.py

@@ -9,11 +9,12 @@ def test_error():
     df = pd.DataFrame(
         {"A": pd.Series([[0, 1, 2], np.nan, [], (3, 4)], index=list("abcd")), "B": 1}
     )
-    with pytest.raises(ValueError):
+    msg = r"(:?columns must be unique)|(:?column must be a scalar)"
+    with pytest.raises(ValueError, match=msg):
         df.explode(list("AA"))
 
     df.columns = list("AA")
-    with pytest.raises(ValueError):
+    with pytest.raises(ValueError, match=msg):
         df.explode("A")