Skip to content

Commit 535847c

Browse files
committed
TST: IndexEngine overrides
1 parent df6b665 commit 535847c

File tree

1 file changed

+38
-1
lines changed

1 file changed

+38
-1
lines changed

pandas/tests/indexes/test_indexing.py

+38-1
Original file line numberDiff line numberDiff line change
@@ -264,7 +264,44 @@ def test_getitem_deprecated_float(idx):
264264

265265

266266
@pytest.mark.parametrize(
267-
"idx, target, expected",
267+
"idx,targets",
268+
[(Float64Index([]), np.array([np.nan])), (DatetimeIndex([]), np.array([NaT]))],
269+
)
270+
def test_get_stargets(idx, targets):
271+
# GH 35392
272+
assert -1 in idx._engine.get_stargets(targets)
273+
274+
275+
@pytest.mark.parametrize(
276+
"idx,val,expected",
277+
[
278+
(Int64Index([]), 1, 1),
279+
(Float64Index([]), np.nan, -1),
280+
(DatetimeIndex([]), NaT, -1),
281+
],
282+
)
283+
def test_convert_val_if_nan(idx, val, expected):
284+
# GH 35392
285+
assert idx._engine.convert_val_if_nan(val) == expected
286+
287+
288+
@pytest.mark.parametrize(
289+
"idx,target,val,expected",
290+
[
291+
(Int64Index([]), 1, 1, True),
292+
(Float64Index([]), np.nan, np.nan, True),
293+
(Float64Index([]), 1, np.nan, False),
294+
(DatetimeIndex([]), NaT, NaT, True),
295+
(DatetimeIndex([]), 1, NaT, False),
296+
],
297+
)
298+
def test_should_update_d(idx, target, val, expected):
299+
# GH 35392
300+
assert idx._engine.should_update_d(target, val) == expected
301+
302+
303+
@pytest.mark.parametrize(
304+
"idx,target,expected",
268305
[
269306
([np.nan, "var1", np.nan], [np.nan], np.array([0, 2], dtype=np.intp)),
270307
(

0 commit comments

Comments
 (0)