Move tests

Author: dsaxton

pandas/tests/frame/methods/test_truncate.py

@@ -87,15 +87,3 @@ def test_truncate_nonsortedindex(self):
         msg = "truncate requires a sorted index"
         with pytest.raises(ValueError, match=msg):
             df.truncate(before=2, after=20, axis=1)
-
-    @pytest.mark.parametrize(
-        "before, after, indices",
-        [(1, 2, [2, 1]), (None, 2, [2, 1, 0]), (1, None, [3, 2, 1])],
-    )
-    def test_truncate_decreasing_index(self, before, after, indices):
-        # https://github.com/pandas-dev/pandas/issues/33756
-        idx = pd.Index([3, 2, 1, 0])
-        values = pd.DataFrame(range(len(idx)), index=idx)
-        result = values.truncate(before=before, after=after)
-        expected = values.loc[indices]
-        tm.assert_frame_equal(result, expected)

pandas/tests/generic/methods/test_truncate.py

@@ -0,0 +1,21 @@
+import pytest
+
+from pandas import DataFrame, Index, Series
+import pandas._testing as tm
+
+
+@pytest.mark.parametrize(
+    "before, after, indices",
+    [(1, 2, [2, 1]), (None, 2, [2, 1, 0]), (1, None, [3, 2, 1])],
+)
+@pytest.mark.parametrize("klass", [Series, DataFrame])
+def test_truncate_decreasing_index(before, after, indices, klass):
+    # https://github.com/pandas-dev/pandas/issues/33756
+    idx = Index([3, 2, 1, 0])
+    values = klass(range(len(idx)), index=idx)
+    result = values.truncate(before=before, after=after)
+    expected = values.loc[indices]
+    if klass is Series:
+        tm.assert_series_equal(result, expected)
+    else:
+        tm.assert_frame_equal(result, expected)

pandas/tests/series/methods/test_truncate.py

@@ -80,18 +80,6 @@ def test_truncate_nonsortedindex(self):
         with pytest.raises(ValueError, match=msg):
             ts.sort_values(ascending=False).truncate(before="2011-11", after="2011-12")
 
-    @pytest.mark.parametrize(
-        "before, after, indices",
-        [(1, 2, [2, 1]), (None, 2, [2, 1, 0]), (1, None, [3, 2, 1])],
-    )
-    def test_truncate_decreasing_index(self, before, after, indices):
-        # https://github.com/pandas-dev/pandas/issues/33756
-        idx = pd.Index([3, 2, 1, 0])
-        values = pd.Series(range(len(idx)), index=idx)
-        result = values.truncate(before=before, after=after)
-        expected = values.loc[indices]
-        tm.assert_series_equal(result, expected)
-
     def test_truncate_datetimeindex_tz(self):
         # GH 9243
         idx = date_range("4/1/2005", "4/30/2005", freq="D", tz="US/Pacific")