CLN: Refactor pandas/tests/base - part3

pandas/conftest.py

@@ -881,3 +881,72 @@ def index_or_series(request):
     See GH#29725
     """
     return request.param
+
+
+_indexes = {
+    "bool-index": tm.makeBoolIndex(10, name="a"),
+    "int-index": tm.makeIntIndex(10, name="a"),
+    "float-index": tm.makeFloatIndex(10, name="a"),
+    "date-index": tm.makeDateIndex(10, name="a"),
+    "localized-date-index": tm.makeDateIndex(10, name="a", tz="US/Eastern"),
+    "period-index": tm.makePeriodIndex(10, name="a"),
+    "string-index": tm.makeStringIndex(10, name="a"),
+    "unicode-index": tm.makeUnicodeIndex(10, name="a"),
+}
+
+
+@pytest.fixture(params=_indexes.keys())
+def index(request):
+    """
+    Fixture for tests on indexes
+
+    copy to avoid mutation, e.g. setting .name
+    """
+    return _indexes[request.param].copy()
+
+
+_arr = np.random.randn(10)
+
+
+@pytest.fixture
+def int_series():
+    """ Fixture for Series with random numbers and integer index """
+    index = _indexes["int-index"].copy()
+    return pd.Series(_arr, index=index, name=index.name)
+
+
+_series = {
+    f"series-with-{i_id}": pd.Series(_arr, index=i, name=i.name)
+    for i_id, i in _indexes.items()
+}
+
+_arr_int = np.random.choice(10, size=10, replace=False)
+
+
+def _create_narrow_series(data, data_dtype):
+    """ Helper for the _narrow_series dict """
+    index = _indexes["int-index"].copy()
+    return pd.Series(data.astype(data_dtype), index=index, name="a")
+
+
+_narrow_series = {
+    "float32-series": _create_narrow_series(_arr, np.float32),
+    "int8-series": _create_narrow_series(_arr_int, np.int8),
+    "int16-series": _create_narrow_series(_arr_int, np.int16),
+    "int32-series": _create_narrow_series(_arr_int, np.int32),
+    "uint8-series": _create_narrow_series(_arr_int, np.uint8),
+    "uint16-series": _create_narrow_series(_arr_int, np.uint16),
+    "uint32-series": _create_narrow_series(_arr_int, np.uint32),
+}
+
+_all_objs = {**_indexes, **_series, **_narrow_series}
+
+
+@pytest.fixture(params=_all_objs.keys())
+def index_or_series_obj(request):
+    """
+    Fixture for tests on indexes, series and series with a narrow dtype
+
+    copy to avoid mutation, e.g. setting .name
+    """
+    return _all_objs[request.param].copy(deep=True)

pandas/tests/base/test_ops.py

@@ -130,27 +130,28 @@ def check_ops_properties(self, props, filter=None, ignore_failures=False):
                     with pytest.raises(err):
                         getattr(o, op)
 
-    @pytest.mark.parametrize("klass", [Series, DataFrame])
-    def test_binary_ops_docs(self, klass):
-        op_map = {
-            "add": "+",
-            "sub": "-",
-            "mul": "*",
-            "mod": "%",
-            "pow": "**",
-            "truediv": "/",
-            "floordiv": "//",
-        }
-        for op_name in op_map:
-            operand1 = klass.__name__.lower()
-            operand2 = "other"
-            op = op_map[op_name]
-            expected_str = " ".join([operand1, op, operand2])
-            assert expected_str in getattr(klass, op_name).__doc__
-
-            # reverse version of the binary ops
-            expected_str = " ".join([operand2, op, operand1])
-            assert expected_str in getattr(klass, "r" + op_name).__doc__
+
+@pytest.mark.parametrize("klass", [Series, DataFrame])
+def test_binary_ops_docs(klass):
+    op_map = {
+        "add": "+",
+        "sub": "-",
+        "mul": "*",
+        "mod": "%",
+        "pow": "**",
+        "truediv": "/",
+        "floordiv": "//",
+    }
+    for op_name in op_map:
+        operand1 = klass.__name__.lower()
+        operand2 = "other"
+        op = op_map[op_name]
+        expected_str = " ".join([operand1, op, operand2])
+        assert expected_str in getattr(klass, op_name).__doc__
+
+        # reverse version of the binary ops
+        expected_str = " ".join([operand2, op, operand1])
+        assert expected_str in getattr(klass, "r" + op_name).__doc__
 
 
 class TestTranspose(Ops):

pandas/tests/base/utils.py

@@ -0,0 +1,59 @@
+from typing import Any, Callable, List, Type
+
+import numpy as np
+import pytest
+
+from pandas import Index, Series
+from pandas.core.indexes.datetimelike import DatetimeIndexOpsMixin
+import pandas.util.testing as tm
+
+
+def allow_na_ops(obj: Any) -> bool:
+    """Whether to skip test cases including NaN"""
+    is_bool_index = isinstance(obj, Index) and obj.is_boolean()
+    return not is_bool_index and obj._can_hold_na
+
+
+def check_ops_properties_valid(obj: Any, props: List[str], filter: Callable) -> None:
+    """ Validates that certain properties are available """
+    for op in props:
+        # if a filter, skip if it doesn't match
+        index = obj.index if isinstance(obj, Series) else obj
+        if not filter(index):
+            continue
+
+        try:
+            if isinstance(obj, Series):
+                expected = Series(getattr(obj.index, op), index=obj.index, name="a")
+            else:
+                expected = getattr(obj, op)
+        except AttributeError:
+            continue
+
+        result = getattr(obj, op)
+
+        # these could be series, arrays or scalars
+        if isinstance(result, Series) and isinstance(expected, Series):
+            tm.assert_series_equal(result, expected)
+        elif isinstance(result, Index) and isinstance(expected, Index):
+            tm.assert_index_equal(result, expected)
+        elif isinstance(result, np.ndarray) and isinstance(expected, np.ndarray):
+            tm.assert_numpy_array_equal(result, expected)
+        else:
+            assert result == expected
+
+
+def check_ops_properties_invalid(obj: Any, props: List[str]) -> None:
+    """ Validates that certain properties are not available """
+    for op in props:
+        # freq raises AttributeError on an Int64Index because its not
+        # defined we mostly care about Series here anyhow
+
+        # an object that is datetimelike will raise a TypeError,
+        # otherwise an AttributeError
+        err: Type[Exception] = AttributeError
+        if issubclass(type(obj), DatetimeIndexOpsMixin):
+            err = TypeError
+
+        with pytest.raises(err):
+            getattr(obj, op)