Function test_4d

Wrapping/Python/tests/sitkImageIndexingTest.py:402–448

testing __getitem__ for getting 4D sub-regions

()

    assert_image_ndarray_equals(img[..., 0], nda[0, ...])

def test_4d():
    """testing __getitem__ for getting 4D sub-regions"""

    # Check if we have 4D Image support
    try:
        sitk.Image([2, 2, 2, 2], sitk.sitkFloat32)
    except RuntimeError:
        pytest.skip("4D Image not supported")

    nda = np.linspace(0, 119, 120).reshape(2, 3, 4, 5)

    img = sitk.GetImageFromArray(nda, isVector=False)

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

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

    assert_image_ndarray_equals(img[1:-1, 1:-1, 1:-1], nda[1:-1, 1:-1, 1:-1])
    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[3:6, 1:5, 2:4, 0:2], nda[0:2, 2:4, 1:5, 3:6])
    assert_image_ndarray_equals(img[1:, 2:, :2, :], nda[:, :2, 2:, 1:])

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

def test_5d():
    """testing __getitem__ for 5D image"""