update message

jbrockmendel · Aug 22, 2024 · 7f10e72 · 7f10e72
1 parent c498d20
commit 7f10e72
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/pandas/tests/groupby/test_raises.py b/pandas/tests/groupby/test_raises.py
@@ -219,7 +219,13 @@ def test_groupby_raises_string(
         elif groupby_func in ["cummin", "cummax"]:
             msg = msg.replace("object", "str")
         elif groupby_func == "corrwith":
-            msg = "'.*NumpySemantics' with dtype str does not support operation 'mean'"
+            if df["d"].dtype.storage == "pyarrow":
+                msg = (
+                    "ArrowStringArrayNumpySemantics' with dtype str does not "
+                    "support operation 'mean'"
+                )
+            else:
+                msg = "Cannot perform reduction 'mean' with string dtype"
 
     if groupby_func == "fillna":
         kind = "Series" if groupby_series else "DataFrame"