Skip to content

Commit 4294c16

Browse files
committed
add more tests
1 parent c49d785 commit 4294c16

File tree

1 file changed

+22
-58
lines changed

1 file changed

+22
-58
lines changed

pandas/tests/window/test_rolling.py

+22-58
Original file line numberDiff line numberDiff line change
@@ -1781,82 +1781,46 @@ def test_step_not_positive_raises():
17811781
["values", "window", "min_periods", "expected"],
17821782
[
17831783
[
1784-
np.array([20, 10, 10, np.inf, 1, 1, 2, 3]),
1784+
[20, 10, 10, np.inf, 1, 1, 2, 3],
17851785
3,
17861786
1,
1787-
np.array(
1788-
[
1789-
np.nan,
1790-
50.0,
1791-
33.33333333333333,
1792-
0.0,
1793-
40.5,
1794-
0.0,
1795-
0.3333333333333333,
1796-
1.0,
1797-
]
1798-
),
1787+
[np.nan, 50, 100 / 3, 0, 40.5, 0, 1 / 3, 1],
17991788
],
18001789
[
1801-
np.array([20, 10, 10, np.nan, 10, 1, 2, 3]),
1790+
[20, 10, 10, np.nan, 10, 1, 2, 3],
18021791
3,
18031792
1,
1804-
np.array(
1805-
[
1806-
np.nan,
1807-
50.0,
1808-
33.33333333333333,
1809-
0.0,
1810-
0.0,
1811-
40.5,
1812-
24.333333333333332,
1813-
1.0,
1814-
]
1815-
),
1793+
[np.nan, 50, 100 / 3, 0, 0, 40.5, 73 / 3, 1],
18161794
],
18171795
[
1818-
np.array([np.nan, 5, 6, 7, 5, 5, 5]),
1796+
[np.nan, 5, 6, 7, 5, 5, 5],
18191797
3,
18201798
3,
1821-
np.array([np.nan, np.nan, np.nan, 1.0, 1.0, 1.3333333333333335, 0.0]),
1799+
[np.nan] * 3 + [1, 1, 4 / 3, 0],
18221800
],
18231801
[
1824-
np.array([5, 7, 7, 7, np.nan, np.inf, 4, 3, 3, 3]),
1802+
[5, 7, 7, 7, np.nan, np.inf, 4, 3, 3, 3],
18251803
3,
18261804
3,
1827-
np.array(
1828-
[
1829-
np.nan,
1830-
np.nan,
1831-
1.3333333333333335,
1832-
0.0,
1833-
np.nan,
1834-
np.nan,
1835-
np.nan,
1836-
np.nan,
1837-
0.33333333333333337,
1838-
0.0,
1839-
]
1840-
),
1805+
[np.nan] * 2 + [4 / 3, 0] + [np.nan] * 4 + [1 / 3, 0],
18411806
],
18421807
[
1843-
np.array([5, 7, 7, 7, np.nan, np.inf, 7, 3, 3, 3]),
1808+
[5, 7, 7, 7, np.nan, np.inf, 7, 3, 3, 3],
18441809
3,
18451810
3,
1846-
np.array(
1847-
[
1848-
np.nan,
1849-
np.nan,
1850-
1.3333333333333335,
1851-
0.0,
1852-
np.nan,
1853-
np.nan,
1854-
np.nan,
1855-
np.nan,
1856-
5.333333333333333,
1857-
0.0,
1858-
]
1859-
),
1811+
[np.nan] * 2 + [4 / 3, 0] + [np.nan] * 4 + [16 / 3, 0],
1812+
],
1813+
[
1814+
[5, 7] * 4,
1815+
3,
1816+
3,
1817+
[np.nan] * 2 + [4 / 3] * 6,
1818+
],
1819+
[
1820+
[5, 7, 5, np.nan, 7, 5, 7],
1821+
3,
1822+
2,
1823+
[np.nan, 2, 4 / 3] + [2] * 3 + [4 / 3],
18601824
],
18611825
],
18621826
)

0 commit comments

Comments
 (0)