improving test cases

Author: sanjanam1998

pandas/tests/base/test_value_counts.py

@@ -14,6 +14,7 @@
     Series,
     Timedelta,
     TimedeltaIndex,
+    Timestamp,
     array,
 )
 import pandas._testing as tm
@@ -344,52 +345,74 @@ def test_value_counts_object_inference_deprecated():
 @pytest.mark.parametrize(
     ("index", "expected_index"),
     [
-        pytest.param(
+        [
             pd.date_range("2016-01-01", periods=5, freq="D"),
             pd.date_range("2016-01-01", periods=5, freq="D"),
-        ),
-        pytest.param(
+        ],
+        [
             pd.timedelta_range(Timedelta(0), periods=5, freq="h"),
             pd.timedelta_range(Timedelta(0), periods=5, freq="h"),
-        ),
-        pytest.param(
-            pd.date_range("2016-01-01", periods=5, freq="D").insert(
-                1, pd.date_range("2016-01-01", periods=5, freq="D")[1]
+        ],
+        [
+            DatetimeIndex(
+                [Timestamp("2016-01-01") + Timedelta(days=i) for i in range(1)]
+                + [Timestamp("2016-01-02")]
+                + [Timestamp("2016-01-01") + Timedelta(days=i) for i in range(1, 5)]
             ),
             DatetimeIndex(pd.date_range("2016-01-01", periods=5, freq="D")),
-        ),
-        pytest.param(
-            pd.timedelta_range(Timedelta(0), periods=5, freq="h").insert(
-                1, pd.timedelta_range(Timedelta(0), periods=5, freq="h")[1]
+        ],
+        [
+            TimedeltaIndex(
+                [Timedelta(hours=i) for i in range(1)]
+                + [Timedelta(hours=1)]
+                + [Timedelta(hours=i) for i in range(1, 5)],
             ),
             TimedeltaIndex(pd.timedelta_range(Timedelta(0), periods=5, freq="h")),
-        ),
-        pytest.param(
-            pd.date_range("2016-01-01", periods=5, freq="D").delete(2),
-            DatetimeIndex(pd.date_range("2016-01-01", periods=5, freq="D").delete(2)),
-        ),
-        pytest.param(
-            pd.timedelta_range(Timedelta(0), periods=5, freq="h").delete(2),
+        ],
+        [
+            DatetimeIndex(
+                [Timestamp("2016-01-01") + Timedelta(days=i) for i in range(2)]
+                + [Timestamp("2016-01-01") + Timedelta(days=i) for i in range(3, 5)],
+            ),
+            DatetimeIndex(
+                [Timestamp("2016-01-01") + Timedelta(days=i) for i in range(2)]
+                + [Timestamp("2016-01-01") + Timedelta(days=i) for i in range(3, 5)],
+            ),
+        ],
+        [
+            TimedeltaIndex(
+                [Timedelta(hours=i) for i in range(2)]
+                + [Timedelta(hours=i) for i in range(3, 5)],
+            ),
             TimedeltaIndex(
-                pd.timedelta_range(Timedelta(0), periods=5, freq="h").delete(2)
+                [Timedelta(hours=i) for i in range(2)]
+                + [Timedelta(hours=i) for i in range(3, 5)],
             ),
-        ),
-        pytest.param(
-            pd.date_range("2016-01-01", periods=5, freq="D").insert(1, pd.NaT),
+        ],
+        [
             DatetimeIndex(
-                list(pd.date_range("2016-01-01", periods=5, freq="D")[:1])
+                [Timestamp("2016-01-01")]
+                + [pd.NaT]
+                + [Timestamp("2016-01-01") + Timedelta(days=i) for i in range(1, 5)],
+            ),
+            DatetimeIndex(
+                [Timestamp("2016-01-01")]
+                + [pd.NaT]
+                + [Timestamp("2016-01-01") + Timedelta(days=i) for i in range(1, 5)],
+            ),
+        ],
+        [
+            TimedeltaIndex(
+                [Timedelta(hours=0)]
                 + [pd.NaT]
-                + list(pd.date_range("2016-01-01", periods=5, freq="D")[1:])
+                + [Timedelta(hours=i) for i in range(1, 5)],
             ),
-        ),
-        pytest.param(
-            pd.timedelta_range(Timedelta(0), periods=5, freq="h").insert(1, pd.NaT),
             TimedeltaIndex(
-                list(pd.timedelta_range(Timedelta(0), periods=5, freq="h")[:1])
+                [Timedelta(hours=0)]
                 + [pd.NaT]
-                + list(pd.timedelta_range(Timedelta(0), periods=5, freq="h")[1:])
+                + [Timedelta(hours=i) for i in range(1, 5)],
             ),
-        ),
+        ],
     ],
 )
 def test_value_counts_index_datetimelike(index, expected_index):