You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
1.8 KiB
57 lines
1.8 KiB
1 year ago
|
import numpy as np
|
||
|
import pytest
|
||
|
|
||
|
from pandas import IntervalIndex
|
||
|
import pandas._testing as tm
|
||
|
|
||
|
|
||
|
class TestInterval:
|
||
|
"""
|
||
|
Tests specific to the shared common index tests; unrelated tests should be placed
|
||
|
in test_interval.py or the specific test file (e.g. test_astype.py)
|
||
|
"""
|
||
|
|
||
|
@pytest.fixture
|
||
|
def simple_index(self) -> IntervalIndex:
|
||
|
return IntervalIndex.from_breaks(range(11), closed="right")
|
||
|
|
||
|
@pytest.fixture
|
||
|
def index(self):
|
||
|
return tm.makeIntervalIndex(10)
|
||
|
|
||
|
def test_take(self, closed):
|
||
|
index = IntervalIndex.from_breaks(range(11), closed=closed)
|
||
|
|
||
|
result = index.take(range(10))
|
||
|
tm.assert_index_equal(result, index)
|
||
|
|
||
|
result = index.take([0, 0, 1])
|
||
|
expected = IntervalIndex.from_arrays([0, 0, 1], [1, 1, 2], closed=closed)
|
||
|
tm.assert_index_equal(result, expected)
|
||
|
|
||
|
def test_where(self, simple_index, listlike_box):
|
||
|
klass = listlike_box
|
||
|
|
||
|
idx = simple_index
|
||
|
cond = [True] * len(idx)
|
||
|
expected = idx
|
||
|
result = expected.where(klass(cond))
|
||
|
tm.assert_index_equal(result, expected)
|
||
|
|
||
|
cond = [False] + [True] * len(idx[1:])
|
||
|
expected = IntervalIndex([np.nan] + idx[1:].tolist())
|
||
|
result = idx.where(klass(cond))
|
||
|
tm.assert_index_equal(result, expected)
|
||
|
|
||
|
def test_getitem_2d_deprecated(self, simple_index):
|
||
|
# GH#30588 multi-dim indexing is deprecated, but raising is also acceptable
|
||
|
idx = simple_index
|
||
|
with pytest.raises(ValueError, match="multi-dimensional indexing not allowed"):
|
||
|
idx[:, None]
|
||
|
with pytest.raises(ValueError, match="multi-dimensional indexing not allowed"):
|
||
|
# GH#44051
|
||
|
idx[True]
|
||
|
with pytest.raises(ValueError, match="multi-dimensional indexing not allowed"):
|
||
|
# GH#44051
|
||
|
idx[False]
|