Skip to content

Commit 59e8d86

Browse files
committed
TST: add NaN tests for all data types (pandas-dev#12535)
1 parent 269d368 commit 59e8d86

File tree

1 file changed

+15
-1
lines changed

1 file changed

+15
-1
lines changed

pandas/tests/window/test_dtypes.py

+15-1
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,18 @@ def get_expects(self):
5353
"var": Series([np.nan, 2, 2, 2, 2], dtype="float64"),
5454
"median": Series([np.nan, 9, 7, 5, 3], dtype="float64"),
5555
},
56+
"sr3": {
57+
"count": Series([1, 2, 2, 1, 1], dtype="float64"),
58+
"max": Series([np.nan, 1, 2, np.nan, np.nan], dtype="float64"),
59+
"min": Series([np.nan, 0, 1, np.nan, np.nan], dtype="float64"),
60+
"sum": Series([np.nan, 1, 3, np.nan, np.nan], dtype="float64"),
61+
"mean": Series([np.nan, 0.5, 1.5, np.nan, np.nan], dtype="float64"),
62+
"std": Series(
63+
[np.nan] + [np.sqrt(0.5)] * 2 + [np.nan] * 2, dtype="float64"
64+
),
65+
"var": Series([np.nan, 0.5, 0.5, np.nan, np.nan], dtype="float64"),
66+
"median": Series([np.nan, 0.5, 1.5, np.nan, np.nan], dtype="float64"),
67+
},
5668
"df": {
5769
"count": DataFrame(
5870
{0: Series([1, 2, 2, 2, 2]), 1: Series([1, 2, 2, 2, 2])},
@@ -99,9 +111,11 @@ def get_expects(self):
99111
def _create_dtype_data(self, dtype):
100112
sr1 = Series(np.arange(5), dtype=dtype)
101113
sr2 = Series(np.arange(10, 0, -2), dtype=dtype)
114+
sr3 = sr1.copy()
115+
sr3[3] = np.NaN
102116
df = DataFrame(np.arange(10).reshape((5, 2)), dtype=dtype)
103117

104-
data = {"sr1": sr1, "sr2": sr2, "df": df}
118+
data = {"sr1": sr1, "sr2": sr2, "sr3": sr3, "df": df}
105119

106120
return data
107121

0 commit comments

Comments
 (0)