151 lines
4.3 KiB
Python
151 lines
4.3 KiB
Python
|
"""
|
||
|
|
test_indexing tests the following Index methods:
|
||
|
|
__getitem__
|
||
|
|
get_loc
|
||
|
|
get_value
|
||
|
|
__contains__
|
||
|
|
take
|
||
|
|
where
|
||
|
|
get_indexer
|
||
|
|
slice_locs
|
||
|
|
asof_locs
|
||
|
|
|
||
|
|
The corresponding tests.indexes.[index_type].test_indexing files
|
||
|
|
contain tests for the corresponding methods specific to those Index subclasses.
|
||
|
|
"""
|
||
|
|
import numpy as np
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from pandas import (
|
||
|
|
DatetimeIndex,
|
||
|
|
Float64Index,
|
||
|
|
Index,
|
||
|
|
Int64Index,
|
||
|
|
PeriodIndex,
|
||
|
|
TimedeltaIndex,
|
||
|
|
UInt64Index,
|
||
|
|
)
|
||
|
|
import pandas._testing as tm
|
||
|
|
|
||
|
|
|
||
|
|
class TestTake:
|
||
|
|
def test_take_invalid_kwargs(self, index):
|
||
|
|
indices = [1, 2]
|
||
|
|
|
||
|
|
msg = r"take\(\) got an unexpected keyword argument 'foo'"
|
||
|
|
with pytest.raises(TypeError, match=msg):
|
||
|
|
index.take(indices, foo=2)
|
||
|
|
|
||
|
|
msg = "the 'out' parameter is not supported"
|
||
|
|
with pytest.raises(ValueError, match=msg):
|
||
|
|
index.take(indices, out=indices)
|
||
|
|
|
||
|
|
msg = "the 'mode' parameter is not supported"
|
||
|
|
with pytest.raises(ValueError, match=msg):
|
||
|
|
index.take(indices, mode="clip")
|
||
|
|
|
||
|
|
def test_take(self, index):
|
||
|
|
indexer = [4, 3, 0, 2]
|
||
|
|
if len(index) < 5:
|
||
|
|
# not enough elements; ignore
|
||
|
|
return
|
||
|
|
|
||
|
|
result = index.take(indexer)
|
||
|
|
expected = index[indexer]
|
||
|
|
assert result.equals(expected)
|
||
|
|
|
||
|
|
if not isinstance(index, (DatetimeIndex, PeriodIndex, TimedeltaIndex)):
|
||
|
|
# GH 10791
|
||
|
|
msg = r"'(.*Index)' object has no attribute 'freq'"
|
||
|
|
with pytest.raises(AttributeError, match=msg):
|
||
|
|
index.freq
|
||
|
|
|
||
|
|
def test_take_minus1_without_fill(self, index):
|
||
|
|
# -1 does not get treated as NA unless allow_fill=True is passed
|
||
|
|
if len(index) == 0:
|
||
|
|
# Test is not applicable
|
||
|
|
return
|
||
|
|
|
||
|
|
result = index.take([0, 0, -1])
|
||
|
|
|
||
|
|
expected = index.take([0, 0, len(index) - 1])
|
||
|
|
tm.assert_index_equal(result, expected)
|
||
|
|
|
||
|
|
|
||
|
|
class TestContains:
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"index,val",
|
||
|
|
[
|
||
|
|
(Index([0, 1, 2]), 2),
|
||
|
|
(Index([0, 1, "2"]), "2"),
|
||
|
|
(Index([0, 1, 2, np.inf, 4]), 4),
|
||
|
|
(Index([0, 1, 2, np.nan, 4]), 4),
|
||
|
|
(Index([0, 1, 2, np.inf]), np.inf),
|
||
|
|
(Index([0, 1, 2, np.nan]), np.nan),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_index_contains(self, index, val):
|
||
|
|
assert val in index
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"index,val",
|
||
|
|
[
|
||
|
|
(Index([0, 1, 2]), "2"),
|
||
|
|
(Index([0, 1, "2"]), 2),
|
||
|
|
(Index([0, 1, 2, np.inf]), 4),
|
||
|
|
(Index([0, 1, 2, np.nan]), 4),
|
||
|
|
(Index([0, 1, 2, np.inf]), np.nan),
|
||
|
|
(Index([0, 1, 2, np.nan]), np.inf),
|
||
|
|
# Checking if np.inf in Int64Index should not cause an OverflowError
|
||
|
|
# Related to GH 16957
|
||
|
|
(Int64Index([0, 1, 2]), np.inf),
|
||
|
|
(Int64Index([0, 1, 2]), np.nan),
|
||
|
|
(UInt64Index([0, 1, 2]), np.inf),
|
||
|
|
(UInt64Index([0, 1, 2]), np.nan),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_index_not_contains(self, index, val):
|
||
|
|
assert val not in index
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"index,val", [(Index([0, 1, "2"]), 0), (Index([0, 1, "2"]), "2")]
|
||
|
|
)
|
||
|
|
def test_mixed_index_contains(self, index, val):
|
||
|
|
# GH#19860
|
||
|
|
assert val in index
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"index,val", [(Index([0, 1, "2"]), "1"), (Index([0, 1, "2"]), 2)]
|
||
|
|
)
|
||
|
|
def test_mixed_index_not_contains(self, index, val):
|
||
|
|
# GH#19860
|
||
|
|
assert val not in index
|
||
|
|
|
||
|
|
def test_contains_with_float_index(self):
|
||
|
|
# GH#22085
|
||
|
|
integer_index = Int64Index([0, 1, 2, 3])
|
||
|
|
uinteger_index = UInt64Index([0, 1, 2, 3])
|
||
|
|
float_index = Float64Index([0.1, 1.1, 2.2, 3.3])
|
||
|
|
|
||
|
|
for index in (integer_index, uinteger_index):
|
||
|
|
assert 1.1 not in index
|
||
|
|
assert 1.0 in index
|
||
|
|
assert 1 in index
|
||
|
|
|
||
|
|
assert 1.1 in float_index
|
||
|
|
assert 1.0 not in float_index
|
||
|
|
assert 1 not in float_index
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"idx", [Index([1, 2, 3]), Index([0.1, 0.2, 0.3]), Index(["a", "b", "c"])]
|
||
|
|
)
|
||
|
|
def test_getitem_deprecated_float(idx):
|
||
|
|
# https://github.com/pandas-dev/pandas/issues/34191
|
||
|
|
|
||
|
|
with tm.assert_produces_warning(FutureWarning):
|
||
|
|
result = idx[1.0]
|
||
|
|
|
||
|
|
expected = idx[1]
|
||
|
|
assert result == expected
|