def test_sa7():
"""
Test exceptions.
"""
a = np.arange(5)
with pytest.raises(ValueError):
segment_axis(a, 2, -1, end='wrap')
def test_sa8():
"""
Test exceptions.
"""
a = np.arange(5)
with pytest.raises(ValueError):
segment_axis(a, 7, 0, end='pants')
def test_sa5():
"""
Test bad wrapping.
"""
a = np.arange(5)
with pytest.raises(ValueError):
segment_axis(a, 10, 0, end='cut')
def test_sa4():
"""
Test wrapping.
"""
a = np.arange(5)
sa = segment_axis(a, 2, 0, end='wrap')
sa_correct = np.array([[0, 1], [2, 3], [4, 0]])
assert np.all(sa == sa_correct)
def test_sa3():
"""
Test cutting.
"""
a = np.arange(5)
sa = segment_axis(a, 2, 0, end='cut')
sa_correct = np.array([[0, 1], [2, 3]])
assert np.all(sa == sa_correct)
def test_sa2():
"""
Test padding.
"""
a = np.arange(5)
sa = segment_axis(a, 2, 0, axis=0, end='pad', endvalue=7)
sa_correct = np.array([[0, 1], [2, 3], [4, 7]])
assert np.all(sa == sa_correct)
def test_sa1():
"""
Test 2d.
"""
a = np.arange(9).reshape(3, 3)
sa = segment_axis(a, 2, 1)
sa_correct = np.array([[0, 1], [1, 2], [2, 3], [3, 4], [4, 5], [5, 6],
[6, 7], [7, 8]])
assert np.all(sa == sa_correct)