Skip to content

Commit

Permalink
Merge pull request #349 from wiredfool/numpy-tests
Browse files Browse the repository at this point in the history
Numpy tests
  • Loading branch information
aclark4life committed Sep 30, 2013
2 parents b1177c8 + 753a326 commit 6954e5c
Showing 1 changed file with 11 additions and 1 deletion.
12 changes: 11 additions & 1 deletion Tests/test_numpy.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,9 +56,19 @@ def to_image(dtype, bands=1, bool=0):
assert_image(to_image(numpy.uint8, 4), "RGBA", (10, 10))


# based on an erring example at http://is.gd/6F0esS
# based on an erring example at http://is.gd/6F0esS (which resolves to)
# http://stackoverflow.com/questions/10854903/what-is-causing-dimension-dependent-attributeerror-in-pil-fromarray-function
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))


def test_16bit():
img = Image.open('Tests/images/12bit.cropped.tif')
px = img.load()
np_img = numpy.array(img)
assert_equal(np_img.shape, (64,64))
assert_equal(px[1,1],np_img[1,1])

0 comments on commit 6954e5c

Please sign in to comment.