Skip to content

Commit 7401a36

Browse files
committed
test: rewrite test to allow better parametrization
1 parent 9991295 commit 7401a36

File tree

1 file changed

+14
-6
lines changed

1 file changed

+14
-6
lines changed

pandas/tests/window/test_rolling.py

Lines changed: 14 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1438,17 +1438,25 @@ def test_rolling_skew_kurt_numerical_stability(method):
14381438

14391439

14401440
@pytest.mark.parametrize(
1441-
("method", "values"),
1441+
"method, data, values",
14421442
[
1443-
("skew", [2.0, 0.854563, 0.0, 1.999984]),
1444-
("kurt", [4.0, -1.289256, -1.2, 3.999946]),
1443+
(
1444+
"skew",
1445+
[3000000, 1, 1, 2, 3, 4, 999],
1446+
[np.nan] * 3 + [2.0, 0.854563, 0.0, 1.999984],
1447+
),
1448+
(
1449+
"kurt",
1450+
[3000000, 1, 1, 2, 3, 4, 999],
1451+
[np.nan] * 3 + [4.0, -1.289256, -1.2, 3.999946],
1452+
),
14451453
],
14461454
)
1447-
def test_rolling_skew_kurt_large_value_range(method, values):
1455+
def test_rolling_skew_kurt_large_value_range(method, data, values):
14481456
# GH: 37557
1449-
s = Series([3000000, 1, 1, 2, 3, 4, 999])
1457+
s = Series(data)
14501458
result = getattr(s.rolling(4), method)()
1451-
expected = Series([np.nan] * 3 + values)
1459+
expected = Series(values)
14521460
tm.assert_series_equal(result, expected)
14531461

14541462

0 commit comments

Comments
 (0)