try fix type annotations

Author: jorisvandenbossche

pandas/tests/extension/base/ops.py

@@ -24,7 +24,7 @@ class BaseOpsUtil:
 
     def _get_expected_exception(
         self, op_name: str, obj, other
-    ) -> type[Exception] | None:
+    ) -> type[Exception] | tuple[type[Exception]] | None:
         # Find the Exception, if any we expect to raise calling
         #  obj.__op_name__(other)
 

pandas/tests/extension/decimal/test_decimal.py

@@ -68,7 +68,7 @@ def data_for_grouping():
 class TestDecimalArray(base.ExtensionTests):
     def _get_expected_exception(
         self, op_name: str, obj, other
-    ) -> type[Exception] | None:
+    ) -> type[Exception] | tuple[type[Exception]] | None:
         return None
 
     def _supports_reduction(self, ser: pd.Series, op_name: str) -> bool:

pandas/tests/extension/test_arrow.py

@@ -807,8 +807,6 @@ def test_value_counts_returns_pyarrow_int64(self, data):
 
     _combine_le_expected_dtype = "bool[pyarrow]"
 
-    divmod_exc = NotImplementedError
-
     def get_op_from_name(self, op_name):
         short_opname = op_name.strip("_")
         if short_opname == "rtruediv":
@@ -944,7 +942,7 @@ def _get_expected_exception(
         self, op_name: str, obj, other
     ) -> type[Exception] | tuple[type[Exception], ...] | None:
         if op_name in ("__divmod__", "__rdivmod__"):
-            return (self.divmod_exc, TypeError)
+            return (NotImplementedError, TypeError)
 
         dtype = tm.get_dtype(obj)
         # error: Item "dtype[Any]" of "dtype[Any] | ExtensionDtype" has no
@@ -968,10 +966,7 @@ def _get_expected_exception(
             or pa.types.is_integer(pa_dtype)
             or pa.types.is_decimal(pa_dtype)
         ):
-            # TODO: in many of these cases, e.g. non-duration temporal,
-            #  these will *never* be allowed. Would it make more sense to
-            #  re-raise as TypeError, more consistent with non-pyarrow cases?
-            exc = (pa.ArrowNotImplementedError, TypeError)
+            exc = TypeError
         else:
             exc = None
         return exc

pandas/tests/extension/test_string.py

@@ -165,7 +165,7 @@ def test_fillna_no_op_returns_copy(self, data):
 
     def _get_expected_exception(
         self, op_name: str, obj, other
-    ) -> type[Exception] | None:
+    ) -> type[Exception] | tuple[type[Exception]] | None:
         if op_name in [
             "__mod__",
             "__rmod__",