172 lines
4.9 KiB
Python
172 lines
4.9 KiB
Python
"""
|
|
Tests that can be parametrized over _any_ Index object.
|
|
"""
|
|
import re
|
|
|
|
import numpy as np
|
|
import pytest
|
|
|
|
from pandas.errors import InvalidIndexError
|
|
|
|
import pandas._testing as tm
|
|
|
|
|
|
def test_boolean_context_compat(index):
|
|
# GH#7897
|
|
with pytest.raises(ValueError, match="The truth value of a"):
|
|
if index:
|
|
pass
|
|
|
|
with pytest.raises(ValueError, match="The truth value of a"):
|
|
bool(index)
|
|
|
|
|
|
def test_sort(index):
|
|
msg = "cannot sort an Index object in-place, use sort_values instead"
|
|
with pytest.raises(TypeError, match=msg):
|
|
index.sort()
|
|
|
|
|
|
def test_hash_error(index):
|
|
with pytest.raises(TypeError, match=f"unhashable type: '{type(index).__name__}'"):
|
|
hash(index)
|
|
|
|
|
|
def test_mutability(index):
|
|
if not len(index):
|
|
return
|
|
msg = "Index does not support mutable operations"
|
|
with pytest.raises(TypeError, match=msg):
|
|
index[0] = index[0]
|
|
|
|
|
|
def test_map_identity_mapping(index, request):
|
|
# GH#12766
|
|
|
|
result = index.map(lambda x: x)
|
|
if index.dtype == object and result.dtype == bool:
|
|
assert (index == result).all()
|
|
# TODO: could work that into the 'exact="equiv"'?
|
|
return # FIXME: doesn't belong in this file anymore!
|
|
tm.assert_index_equal(result, index, exact="equiv")
|
|
|
|
|
|
def test_wrong_number_names(index):
|
|
names = index.nlevels * ["apple", "banana", "carrot"]
|
|
with pytest.raises(ValueError, match="^Length"):
|
|
index.names = names
|
|
|
|
|
|
def test_view_preserves_name(index):
|
|
assert index.view().name == index.name
|
|
|
|
|
|
def test_ravel(index):
|
|
# GH#19956 ravel returning ndarray is deprecated, in 2.0 returns a view on self
|
|
res = index.ravel()
|
|
tm.assert_index_equal(res, index)
|
|
|
|
|
|
class TestConversion:
|
|
def test_to_series(self, index):
|
|
# assert that we are creating a copy of the index
|
|
|
|
ser = index.to_series()
|
|
assert ser.values is not index.values
|
|
assert ser.index is not index
|
|
assert ser.name == index.name
|
|
|
|
def test_to_series_with_arguments(self, index):
|
|
# GH#18699
|
|
|
|
# index kwarg
|
|
ser = index.to_series(index=index)
|
|
|
|
assert ser.values is not index.values
|
|
assert ser.index is index
|
|
assert ser.name == index.name
|
|
|
|
# name kwarg
|
|
ser = index.to_series(name="__test")
|
|
|
|
assert ser.values is not index.values
|
|
assert ser.index is not index
|
|
assert ser.name != index.name
|
|
|
|
def test_tolist_matches_list(self, index):
|
|
assert index.tolist() == list(index)
|
|
|
|
|
|
class TestRoundTrips:
|
|
def test_pickle_roundtrip(self, index):
|
|
result = tm.round_trip_pickle(index)
|
|
tm.assert_index_equal(result, index, exact=True)
|
|
if result.nlevels > 1:
|
|
# GH#8367 round-trip with timezone
|
|
assert index.equal_levels(result)
|
|
|
|
def test_pickle_preserves_name(self, index):
|
|
original_name, index.name = index.name, "foo"
|
|
unpickled = tm.round_trip_pickle(index)
|
|
assert index.equals(unpickled)
|
|
index.name = original_name
|
|
|
|
|
|
class TestIndexing:
|
|
def test_get_loc_listlike_raises_invalid_index_error(self, index):
|
|
# and never TypeError
|
|
key = np.array([0, 1], dtype=np.intp)
|
|
|
|
with pytest.raises(InvalidIndexError, match=r"\[0 1\]"):
|
|
index.get_loc(key)
|
|
|
|
with pytest.raises(InvalidIndexError, match=r"\[False True\]"):
|
|
index.get_loc(key.astype(bool))
|
|
|
|
def test_getitem_ellipsis(self, index):
|
|
# GH#21282
|
|
result = index[...]
|
|
assert result.equals(index)
|
|
assert result is not index
|
|
|
|
def test_slice_keeps_name(self, index):
|
|
assert index.name == index[1:].name
|
|
|
|
@pytest.mark.parametrize("item", [101, "no_int", 2.5])
|
|
def test_getitem_error(self, index, item):
|
|
msg = "|".join(
|
|
[
|
|
r"index 101 is out of bounds for axis 0 with size [\d]+",
|
|
re.escape(
|
|
"only integers, slices (`:`), ellipsis (`...`), "
|
|
"numpy.newaxis (`None`) and integer or boolean arrays "
|
|
"are valid indices"
|
|
),
|
|
"index out of bounds", # string[pyarrow]
|
|
]
|
|
)
|
|
with pytest.raises(IndexError, match=msg):
|
|
index[item]
|
|
|
|
|
|
class TestRendering:
|
|
def test_str(self, index):
|
|
# test the string repr
|
|
index.name = "foo"
|
|
assert "'foo'" in str(index)
|
|
assert type(index).__name__ in str(index)
|
|
|
|
|
|
class TestReductions:
|
|
def test_argmax_axis_invalid(self, index):
|
|
# GH#23081
|
|
msg = r"`axis` must be fewer than the number of dimensions \(1\)"
|
|
with pytest.raises(ValueError, match=msg):
|
|
index.argmax(axis=1)
|
|
with pytest.raises(ValueError, match=msg):
|
|
index.argmin(axis=2)
|
|
with pytest.raises(ValueError, match=msg):
|
|
index.min(axis=-2)
|
|
with pytest.raises(ValueError, match=msg):
|
|
index.max(axis=-3)
|