update test

Author: ganevgv

pandas/tests/resample/test_base.py

@@ -273,66 +273,44 @@ def test_resample_quantile(series):
     "_index_factory,_series_name,_index_start,_index_end", [DATE_RANGE, PERIOD_RANGE]
 )
 @pytest.mark.parametrize(
-    "freq, result_dict",
+    "freq, result_data, result_index, result_freq",
     [
         (
             "D",
-            {
-                "dti": {
-                    "data": [1.0] * 5 + [np.nan] * 5,
-                    "index": DatetimeIndex(
-                        ["2005-01-{}".format(i) for i in range(1, 11)], freq="D"
-                    ),
-                },
-                "pi": {
-                    "data": [1.0] * 5 + [np.nan] * 5,
-                    "index": PeriodIndex(
-                        ["2005-01-{}".format(i) for i in range(1, 11)], freq="D"
-                    ),
-                },
-            },
+            [1.0] * 5 + [np.nan] * 5,
+            ["2005-01-{}".format(i) for i in range(1, 11)],
+            "D",
         ),
         (
-            "W",
-            {
-                "dti": {
-                    "data": [2.0, 3.0, np.nan],
-                    "index": DatetimeIndex(
-                        ["2005-01-02", "2005-01-09", "2005-01-16"], freq="W-SUN"
-                    ),
-                },
-                "pi": {
-                    "data": [2.0, 3.0, np.nan],
-                    "index": PeriodIndex(
-                        [
-                            "2004-12-27/2005-01-02",
-                            "2005-01-03/2005-01-09",
-                            "2005-01-10/2005-01-16",
-                        ],
-                        freq="W-SUN",
-                    ),
-                },
-            },
+            "D",
+            [1.0] * 5 + [np.nan] * 5,
+            ["2005-01-{}".format(i) for i in range(1, 11)],
+            "D",
         ),
+        ("W", [2.0, 3.0, np.nan], ["2005-01-02", "2005-01-09", "2005-01-16"], "W-SUN"),
         (
-            "M",
-            {
-                "dti": {
-                    "data": [5.0],
-                    "index": DatetimeIndex(["2005-01-31"], freq="M"),
-                },
-                "pi": {"data": [5.0], "index": PeriodIndex(["2005-01"], freq="M")},
-            },
+            "W",
+            [2.0, 3.0, np.nan],
+            ["2004-12-27/2005-01-02", "2005-01-03/2005-01-09", "2005-01-10/2005-01-16"],
+            "W-SUN",
         ),
+        ("M", [5.0], ["2005-01-31"], "M"),
+        ("M", [5.0], ["2005-01"], "M"),
     ],
 )
-def test_resample_sum(series, freq, result_dict):
+def test_resample_sum(series, freq, result_data, result_index, result_freq):
     # GH 19974
     series[:5] = 1
     series[5:] = np.nan
     result = series.resample(freq).sum(min_count=1)
 
-    key = result.name
-    index = result_dict[key]["index"]
-    expected = Series(result_dict[key]["data"], index, name=key)
+    result_name = result.name
+    if isinstance(series.index, DatetimeIndex) and result_name == "dti":
+        index = DatetimeIndex(result_index, freq=result_freq)
+    elif isinstance(series.index, PeriodIndex) and result_name == "pi":
+        index = PeriodIndex(result_index, freq=result_freq)
+    else:
+        pytest.skip("unsupported configuration")
+
+    expected = Series(result_data, index, name=result_name)
     tm.assert_series_equal(result, expected)