started to fixturize pandas/tests/base

Author: SaturnFromTitan

pandas/tests/base/conftest.py

@@ -0,0 +1,52 @@
+import numpy as np
+import pytest
+
+import pandas as pd
+from pandas.tests.indexes.conftest import indices_dict
+
+
+def _create_series(index):
+    """ Helper for the _series dict """
+    data = np.random.randn(len(index))
+    return pd.Series(data, index=index, name=index.name)
+
+
+_series = {
+    f"series-with-{i_id}-index": _create_series(i) for i_id, i in indices_dict.items()
+}
+
+
+def _create_narrow_series(data_dtype):
+    """ Helper for the _narrow_series dict """
+    index = indices_dict["int"].copy()
+    size = len(index)
+    if np.issubdtype(data_dtype, np.float):
+        data = np.random.choice(size, size=size, replace=False)
+    elif np.issubdtype(data_dtype, np.integer):
+        data = np.random.randn(size)
+    else:
+        raise ValueError(f"Received an unexpected data_dtype: {data_dtype}")
+
+    return pd.Series(data.astype(data_dtype), index=index, name="a")
+
+
+_narrow_series = {
+    "float32-series": _create_narrow_series(np.float32),
+    "int8-series": _create_narrow_series(np.int8),
+    "int16-series": _create_narrow_series(np.int16),
+    "int32-series": _create_narrow_series(np.int32),
+    "uint8-series": _create_narrow_series(np.uint8),
+    "uint16-series": _create_narrow_series(np.uint16),
+    "uint32-series": _create_narrow_series(np.uint32),
+}
+
+_all_objs = {**indices_dict, **_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

@@ -109,26 +109,26 @@ def test_binary_ops(klass, op_name, op):
     assert expected_str in getattr(klass, "r" + op_name).__doc__
 
 
-class TestTranspose(Ops):
+class TestTranspose:
     errmsg = "the 'axes' parameter is not supported"
 
-    def test_transpose(self):
-        for obj in self.objs:
-            tm.assert_equal(obj.transpose(), obj)
+    def test_transpose(self, index_or_series_obj):
+        obj = index_or_series_obj
+        tm.assert_equal(obj.transpose(), obj)
 
-    def test_transpose_non_default_axes(self):
-        for obj in self.objs:
-            with pytest.raises(ValueError, match=self.errmsg):
-                obj.transpose(1)
-            with pytest.raises(ValueError, match=self.errmsg):
-                obj.transpose(axes=1)
+    def test_transpose_non_default_axes(self, index_or_series_obj):
+        obj = index_or_series_obj
+        with pytest.raises(ValueError, match=self.errmsg):
+            obj.transpose(1)
+        with pytest.raises(ValueError, match=self.errmsg):
+            obj.transpose(axes=1)
 
-    def test_numpy_transpose(self):
-        for obj in self.objs:
-            tm.assert_equal(np.transpose(obj), obj)
+    def test_numpy_transpose(self, index_or_series_obj):
+        obj = index_or_series_obj
+        tm.assert_equal(np.transpose(obj), obj)
 
-            with pytest.raises(ValueError, match=self.errmsg):
-                np.transpose(obj, axes=1)
+        with pytest.raises(ValueError, match=self.errmsg):
+            np.transpose(obj, axes=1)
 
 
 class TestIndexOps(Ops):