def test_replace_ellipsis():
# 1D, single item
assert (0,) == replace_ellipsis(0, (100,))
# 1D
assert (slice(None),) == replace_ellipsis(Ellipsis, (100,))
assert (slice(None),) == replace_ellipsis(slice(None), (100,))
assert (slice(None, 100),) == replace_ellipsis(slice(None, 100), (100,))
assert (slice(0, None),) == replace_ellipsis(slice(0, None), (100,))
assert (slice(None),) == replace_ellipsis((slice(None), Ellipsis), (100,))
assert (slice(None),) == replace_ellipsis((Ellipsis, slice(None)), (100,))
# 2D, single item
assert (0, 0) == replace_ellipsis((0, 0), (100, 100))
assert (-1, 1) == replace_ellipsis((-1, 1), (100, 100))
# 2D, single col/row
assert (0, slice(None)) == replace_ellipsis((0, slice(None)), (100, 100))
assert (0, slice(None)) == replace_ellipsis((0,), (100, 100))
assert (slice(None), 0) == replace_ellipsis((slice(None), 0), (100, 100))
# 2D slice
assert ((slice(None), slice(None)) ==
replace_ellipsis(Ellipsis, (100, 100)))
assert ((slice(None), slice(None)) ==
replace_ellipsis(slice(None), (100, 100)))
assert ((slice(None), slice(None)) ==
replace_ellipsis((slice(None), slice(None)), (100, 100)))
assert ((slice(None), slice(None)) ==
replace_ellipsis((Ellipsis, slice(None)), (100, 100)))
assert ((slice(None), slice(None)) ==
replace_ellipsis((slice(None), Ellipsis), (100, 100)))
assert ((slice(None), slice(None)) ==
replace_ellipsis((slice(None), Ellipsis, slice(None)), (100, 100)))
assert ((slice(None), slice(None)) ==
replace_ellipsis((Ellipsis, slice(None), slice(None)), (100, 100)))
assert ((slice(None), slice(None)) ==
replace_ellipsis((slice(None), slice(None), Ellipsis), (100, 100)))
def test_replace_ellipsis():
# 1D, single item
eq((0,), replace_ellipsis(0, (100,)))
# 1D
eq((slice(None),), replace_ellipsis(Ellipsis, (100,)))
eq((slice(None),), replace_ellipsis(slice(None), (100,)))
eq((slice(None, 100),), replace_ellipsis(slice(None, 100), (100,)))
eq((slice(0, None),), replace_ellipsis(slice(0, None), (100,)))
eq((slice(None),), replace_ellipsis((slice(None), Ellipsis), (100,)))
eq((slice(None),), replace_ellipsis((Ellipsis, slice(None)), (100,)))
# 2D, single item
eq((0, 0), replace_ellipsis((0, 0), (100, 100)))
eq((-1, 1), replace_ellipsis((-1, 1), (100, 100)))
# 2D, single col/row
eq((0, slice(None)), replace_ellipsis((0, slice(None)), (100, 100)))
eq((0, slice(None)), replace_ellipsis((0,), (100, 100)))
eq((slice(None), 0), replace_ellipsis((slice(None), 0), (100, 100)))
# 2D slice
eq((slice(None), slice(None)),
replace_ellipsis(Ellipsis, (100, 100)))
eq((slice(None), slice(None)),
replace_ellipsis(slice(None), (100, 100)))
eq((slice(None), slice(None)),
replace_ellipsis((slice(None), slice(None)), (100, 100)))
eq((slice(None), slice(None)),
replace_ellipsis((Ellipsis, slice(None)), (100, 100)))
eq((slice(None), slice(None)),
replace_ellipsis((slice(None), Ellipsis), (100, 100)))
eq((slice(None), slice(None)),
replace_ellipsis((slice(None), Ellipsis, slice(None)), (100, 100)))
eq((slice(None), slice(None)),
replace_ellipsis((Ellipsis, slice(None), slice(None)), (100, 100)))
eq((slice(None), slice(None)),
replace_ellipsis((slice(None), slice(None), Ellipsis), (100, 100)))