Function test_3d

Wrapping/Python/tests/sitkImageIndexingTest.py:308–347

testing __getitem__ for 3D sub-region

()

    assert_image_ndarray_equals(img[2, ..., 2:0:-1], nda[2:0:-1, ..., 2])

def test_3d():
    """testing __getitem__ for 3D sub-region"""

    nda = np.linspace(0, 59, 60).reshape(3, 4, 5)

    img = sitk.GetImageFromArray(nda)

    # check some exceptions
    with pytest.raises(IndexError):
        img[0, 3, 0, 0]
    with pytest.raises(IndexError):
        img[0, 4, 0]
    with pytest.raises(IndexError):
        img[0, 0, 3]
    with pytest.raises(IndexError):
        img[-6, 0, 0]
    with pytest.raises(IndexError):
        img[0, 0, -4]

    # check basic indexing
    assert img[0, 0, 0] == 0.0
    assert img[(1, 0, 0)] == 1.0
    assert img[[2, 0, 0]] == 2.0
    assert img[3, 2, 1] == 33.0
    assert img[-3, -2, -1] == 52.0

    assert_image_ndarray_equals(img[1:-1, 1:-1, 1:-1], nda[1:-1, 1:-1, 1:-1])
    assert_image_ndarray_equals(img[:, :, 1:2], nda[1:2, :, :])
    assert_image_ndarray_equals(img[1:5, 2:4, 0:2], nda[0:2, 2:4, 1:5])
    assert_image_ndarray_equals(img[1:, 2:, :2], nda[:2, 2:, 1:])

    # some negative cases
    assert_image_ndarray_equals(img[::-1, ::-1, ::-1], nda[::-1, ::-1, ::-1])
    assert_image_ndarray_equals(img[:, ::-2, ::-1], nda[::-1, ::-2, :])
    assert_image_ndarray_equals(img[-1:-4:-1, :, :], nda[:, :, -1:-4:-1])

    assert_image_ndarray_equals(img[1, ...], nda[..., 1])
    assert_image_ndarray_equals(img[2, 1:3, ...], nda[..., 1:3, 2])
    assert_image_ndarray_equals(img[...], nda[...])
    assert_image_ndarray_equals(img[..., -2], nda[-2, ...])

def test_4d_extract():
    """testing __getitem__ for extracting slices from 4D image"""