def test_get_indexer_at_least_n_items(self):
cases = [
((20, ), (slice(10), ), (slice(-10, None), )),
((
3,
20,
), (0, slice(10)), (-1, slice(-10, None))),
((
2,
10,
), (0, slice(10)), (-1, slice(-10, None))),
((
2,
5,
), (slice(2), slice(None)), (slice(-2, None), slice(None))),
((
1,
2,
5,
), (0, slice(2), slice(None)), (-1, slice(-2, None), slice(None))),
((
2,
3,
5,
), (0, slice(2), slice(None)), (-1, slice(-2, None), slice(None))),
((
1,
10,
1,
), (0, slice(10), slice(None)), (-1, slice(-10,
None), slice(None))),
((
2,
5,
1,
), (slice(2), slice(None), slice(None)), (slice(-2,
None), slice(None),
slice(None))),
((
2,
5,
3,
), (0, slice(4), slice(None)), (-1, slice(-4, None), slice(None))),
((
2,
3,
3,
), (slice(2), slice(None), slice(None)), (slice(-2,
None), slice(None),
slice(None))),
]
for shape, start_expected, end_expected in cases:
actual = formatting._get_indexer_at_least_n_items(shape,
10,
from_end=False)
assert start_expected == actual
actual = formatting._get_indexer_at_least_n_items(shape,
10,
from_end=True)
assert end_expected == actual
def test_get_indexer_at_least_n_items(self):
cases = [
((20,), (slice(10),), (slice(-10, None),)),
((3, 20,), (0, slice(10)), (-1, slice(-10, None))),
((2, 10,), (0, slice(10)), (-1, slice(-10, None))),
((2, 5,), (slice(2), slice(None)),
(slice(-2, None), slice(None))),
((1, 2, 5,), (0, slice(2), slice(None)),
(-1, slice(-2, None), slice(None))),
((2, 3, 5,), (0, slice(2), slice(None)),
(-1, slice(-2, None), slice(None))),
((1, 10, 1,), (0, slice(10), slice(None)),
(-1, slice(-10, None), slice(None))),
((2, 5, 1,), (slice(2), slice(None), slice(None)),
(slice(-2, None), slice(None), slice(None))),
((2, 5, 3,), (0, slice(4), slice(None)),
(-1, slice(-4, None), slice(None))),
((2, 3, 3,), (slice(2), slice(None), slice(None)),
(slice(-2, None), slice(None), slice(None))),
]
for shape, start_expected, end_expected in cases:
actual = formatting._get_indexer_at_least_n_items(shape, 10,
from_end=False)
assert start_expected == actual
actual = formatting._get_indexer_at_least_n_items(shape, 10,
from_end=True)
assert end_expected == actual
def test_get_indexer_at_least_n_items(self):
cases = [
((20, ), (slice(10), )),
((
3,
20,
), (0, slice(10))),
((
2,
10,
), (0, slice(10))),
((
2,
5,
), (slice(2), slice(None))),
((
1,
2,
5,
), (0, slice(2), slice(None))),
((
2,
3,
5,
), (0, slice(2), slice(None))),
((
1,
10,
1,
), (0, slice(10), slice(None))),
((
2,
5,
1,
), (slice(2), slice(None), slice(None))),
((
2,
5,
3,
), (0, slice(4), slice(None))),
((
2,
3,
3,
), (slice(2), slice(None), slice(None))),
]
for shape, expected in cases:
actual = formatting._get_indexer_at_least_n_items(shape, 10)
self.assertEqual(expected, actual)
def test_get_indexer_at_least_n_items(self):
cases = [
((20,), (slice(10),)),
((3, 20,), (0, slice(10))),
((2, 10,), (0, slice(10))),
((2, 5,), (slice(2), slice(None))),
((1, 2, 5,), (0, slice(2), slice(None))),
((2, 3, 5,), (0, slice(2), slice(None))),
((1, 10, 1,), (0, slice(10), slice(None))),
((2, 5, 1,), (slice(2), slice(None), slice(None))),
((2, 5, 3,), (0, slice(4), slice(None))),
((2, 3, 3,), (slice(2), slice(None), slice(None))),
]
for shape, expected in cases:
actual = formatting._get_indexer_at_least_n_items(shape, 10)
self.assertEqual(expected, actual)
