Function test_setitem

Wrapping/Python/tests/sitkImageIndexingTest.py:124–179

testing __setitem__ with pasting to roi

()

    assert img["test"] == "value"

def test_setitem():
    """testing __setitem__ with pasting to roi"""

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

    img = sitk.GetImageFromArray(nda)
    img["test"] = "value"

    img[1:3, 2:4, 0:2] = sitk.Image([2, 2, 2], sitk.sitkFloat64) + 1
    assert all([px == 1 for px in img[1:3, 2:4, 0:2]])
    img[0:1, 0:1, 0:1] = sitk.Image([1, 1, 1], sitk.sitkFloat64) + 2
    assert 2 == img.GetPixel(0, 0, 0)
    img[0:1, 0:2, 0:3] = sitk.Image([2, 3], sitk.sitkFloat64) + 3
    assert all([px == 3 for px in img[0:1, 0:2, 0:3]])
    img[0:2, 0:1, 0:3] = sitk.Image([2, 3], sitk.sitkFloat64) + 4
    assert all([px == 4 for px in img[0:2, 0:1, 0:3]])
    img[0:2, 0:3, 0:1] = sitk.Image([2, 3], sitk.sitkFloat64) + 5
    assert all([px == 5 for px in img[0:2, 0:3, 0:1]])
    assert img["test"] == "value"

    img[-1, 0:1, 0:3] = sitk.Image([1, 3], sitk.sitkFloat64) + 6
    assert all([px == 6 for px in img[-1, 0:1, 0:3]])
    img[1:2, 1, 0:3] = sitk.Image([1, 3], sitk.sitkFloat64) + 7
    assert all([px == 7 for px in img[1:2, 1, 0:3]])
    img[2:3, 0:3, 1] = sitk.Image([1, 3], sitk.sitkFloat64) + 8
    assert all([px == 8 for px in img[2:3, 0:3, 1]])
    assert img["test"] == "value"

    img[..., 2:4, 2:3] = sitk.Image([5, 2, 1], sitk.sitkFloat64) + 9
    assert all([px == 9 for px in img[:, 2:4, 2:3]])
    img[-2, ...] = sitk.Image([4, 3], sitk.sitkFloat64) + 10
    assert all([px == 10 for px in img[-2, :, :]])
    img[...] = sitk.Image([5, 4, 3], sitk.sitkFloat64) + 11
    assert all([px == 11 for px in img])
    img[2, 3:4, 1:2, ...] = sitk.Image([1, 1], sitk.sitkFloat64) + 12
    assert all([px == 12 for px in img[2, 3:4, 1:2]])
    assert img["test"] == "value"

    with pytest.raises(IndexError):
        img[1:3, 2:4] = sitk.Image([2, 2, 2], sitk.sitkFloat64)
    assert img["test"] == "value"

    with pytest.raises(IndexError):
        do_image_assign(img, (slice(0, 2),) * 3, sitk.Image([2, 2], sitk.sitkFloat64))
    assert img["test"] == "value"

    with pytest.raises(IndexError):
        do_image_assign(img, (slice(0, 2),) * 3, sitk.Image([2, 2, 1], sitk.sitkFloat64))

    with pytest.raises(IndexError):
        do_image_assign(img, (slice(0, 2),) * 3, sitk.Image([2, 1, 2], sitk.sitkFloat64))

    with pytest.raises(IndexError):
        do_image_assign(img, (slice(0, 2),) * 3, sitk.Image([1, 2, 2], sitk.sitkFloat64))

    assert img["test"] == "value"

def test_5d_setitem():