Skip to content

Commit c492adf

Browse files
committed
TST: fix inconsistent tests on Float64Index with missing keys
1 parent a3a1ccc commit c492adf

File tree

1 file changed

+14
-4
lines changed

1 file changed

+14
-4
lines changed

pandas/tests/series/indexing/test_numeric.py

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -59,12 +59,22 @@ def test_get_nan():
5959

6060
# ensure that fixing the above hasn't broken get
6161
# with multiple elements
62+
idx = [2, 30]
63+
with tm.assert_produces_warning(FutureWarning, check_stacklevel=False):
64+
assert_series_equal(s.get(idx),
65+
Series([2, np.nan], index=idx))
66+
67+
idx = [2, np.nan]
68+
with tm.assert_produces_warning(FutureWarning, check_stacklevel=False):
69+
assert_series_equal(s.get(idx),
70+
Series([2, np.nan], index=idx))
71+
72+
# GH 17295 - all missing keys
6273
idx = [20, 30]
63-
assert_series_equal(s.get(idx),
64-
Series([np.nan] * 2, index=idx))
74+
assert(s.get(idx) is None)
75+
6576
idx = [np.nan, np.nan]
66-
assert_series_equal(s.get(idx),
67-
Series([np.nan] * 2, index=idx))
77+
assert(s.get(idx) is None)
6878

6979

7080
def test_delitem():

0 commit comments

Comments
 (0)