Adding a test of fromarray that exercises the if strides is not None block.

This commit is contained in:
Matt Davis 2013-05-21 21:55:58 -04:00
parent 1082173030
commit c3d6b05d10

View File

@ -54,3 +54,11 @@ def test_numpy_to_image():
assert_image(to_image(numpy.uint8, 3), "RGB", (10, 10))
assert_image(to_image(numpy.uint8, 4), "RGBA", (10, 10))
# based on an erring example at http://is.gd/6F0esS
def test_3d_array():
a = numpy.ones((10, 10, 10), dtype=numpy.uint8)
assert_image(Image.fromarray(a[1, :, :]), "L", (10, 10))
assert_image(Image.fromarray(a[:, 1, :]), "L", (10, 10))
assert_image(Image.fromarray(a[:, :, 1]), "L", (10, 10))