python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-08-16 02:04:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

[Test] fix Image.fromarray LA mode test

Browse Source 
AppVeyor: ERROR: Failure: ImportError (No module named numpy)

inspired from: Tests/test_numpy.py
This commit is contained in:
Pierrick Koch 2016-04-27 16:23:22 +02:00
parent 911bd0f7b2
commit fbb6cae785
1 changed files with 18 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
Tests/test_image_fromarray_la.py
View File

@ -2,18 +2,28 @@ from helper import unittest, PillowTestCase
from PIL import Image from PIL import Image
import numpy as np try:
import site
import numpy
except ImportError:
# Skip via setUp()
pass
class TestImageFromArrayLA(PillowTestCase): class TestImageFromArrayLA(PillowTestCase):
def test_sanity(self): def setUp(self):
ar1 = (np.random.random((40,40,2))*255).astype('uint8') try:
im1 = Image.fromarray(ar1) import site
arr = np.array(im1.getdata(), 'uint8') import numpy
ar2 = arr.reshape(im1.size[1], im1.size[0], arr.shape[1]) except ImportError:
im2 = Image.fromarray(ar2) self.skipTest("ImportError")
self.assert_image_equal(im1, im2) def test_sanity(self):
ar1 = (numpy.random.random((40,40,2))*255).astype('uint8')
im1 = Image.fromarray(ar1)
arr = numpy.array(im1.getdata(), 'uint8')
ar2 = arr.reshape(im1.size[1], im1.size[0], arr.shape[1])
self.assertTrue(numpy.array_equal(ar1, ar2))
if __name__ == '__main__': if __name__ == '__main__':