05a37e0353e7055319cc7ea59ff4a32104e42775,dit/inference/tests/test_segmentaxis.py,,test_sa5,#,64

Before Change

    
    a = np.arange(5)
    sa = segment_axis(a, 5, 0, end="wrap")
    sa_correct = np.array([[0, 1, 2, 3, 4]])
    assert np.all(sa == sa_correct)


def test_sa6():

After Change

    Test bad wrapping.
    
    a = np.arange(5)
    with pytest.raises(ValueError):
        sa = segment_axis(a, 10, 0, end="cut")


def test_sa6():
    
    Test exceptions.
    

Instances