Skip to content

Commit acebea2

Browse files
committed
TST: fix inconsistent tests on Float64Index with missing keys
1 parent 8371fb0 commit acebea2

File tree

1 file changed

+19
-5
lines changed

1 file changed

+19
-5
lines changed

pandas/tests/series/indexing/test_numeric.py

Lines changed: 19 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -57,14 +57,28 @@ def test_get_nan():
5757
assert s.get(np.nan) is None
5858
assert s.get(np.nan, default='Missing') == 'Missing'
5959

60-
# ensure that fixing the above hasn't broken get
60+
def test_get_nan_multiple():
61+
# GH 8569
62+
# ensure that fixing "test_get_nan" above hasn't broken get
6163
# with multiple elements
64+
s = pd.Float64Index(range(10)).to_series()
65+
66+
idx = [2, 30]
67+
with tm.assert_produces_warning(FutureWarning, check_stacklevel=False):
68+
assert_series_equal(s.get(idx),
69+
Series([2, np.nan], index=idx))
70+
71+
idx = [2, np.nan]
72+
with tm.assert_produces_warning(FutureWarning, check_stacklevel=False):
73+
assert_series_equal(s.get(idx),
74+
Series([2, np.nan], index=idx))
75+
76+
# GH 17295 - all missing keys
6277
idx = [20, 30]
63-
assert_series_equal(s.get(idx),
64-
Series([np.nan] * 2, index=idx))
78+
assert(s.get(idx) is None)
79+
6580
idx = [np.nan, np.nan]
66-
assert_series_equal(s.get(idx),
67-
Series([np.nan] * 2, index=idx))
81+
assert(s.get(idx) is None)
6882

6983

7084
def test_delitem():

0 commit comments

Comments
 (0)