pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improved PIL detection for tests

Browse Source
This commit is contained in:
Ross Lawley 2013-06-03 13:38:58 +00:00
parent 7e6b035ca2
commit ceece5a7e2
1 changed files with 18 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
tests/fields/file_tests.py
View File

@ -14,6 +14,12 @@ from mongoengine import *
from mongoengine.connection import get_db from mongoengine.connection import get_db
from mongoengine.python_support import PY3, b, StringIO from mongoengine.python_support import PY3, b, StringIO
try:
from PIL import Image
HAS_PIL = True
except ImportError:
HAS_PIL = False
TEST_IMAGE_PATH = os.path.join(os.path.dirname(__file__), 'mongoengine.png') TEST_IMAGE_PATH = os.path.join(os.path.dirname(__file__), 'mongoengine.png')
TEST_IMAGE2_PATH = os.path.join(os.path.dirname(__file__), 'mongodb_leaf.png') TEST_IMAGE2_PATH = os.path.join(os.path.dirname(__file__), 'mongodb_leaf.png')
@ -255,8 +261,8 @@ class FileTest(unittest.TestCase):
self.assertFalse(test_file.the_file in [{"test": 1}]) self.assertFalse(test_file.the_file in [{"test": 1}])
def test_image_field(self): def test_image_field(self):
if PY3: if not HAS_PIL:
raise SkipTest('PIL does not have Python 3 support') raise SkipTest('PIL not installed')
class TestImage(Document): class TestImage(Document):
image = ImageField() image = ImageField()
@ -278,8 +284,8 @@ class FileTest(unittest.TestCase):
t.image.delete() t.image.delete()
def test_image_field_reassigning(self): def test_image_field_reassigning(self):
if PY3: if not HAS_PIL:
raise SkipTest('PIL does not have Python 3 support') raise SkipTest('PIL not installed')
class TestFile(Document): class TestFile(Document):
the_file = ImageField() the_file = ImageField()
@ -294,8 +300,8 @@ class FileTest(unittest.TestCase):
self.assertEqual(test_file.the_file.size, (45, 101)) self.assertEqual(test_file.the_file.size, (45, 101))
def test_image_field_resize(self): def test_image_field_resize(self):
if PY3: if not HAS_PIL:
raise SkipTest('PIL does not have Python 3 support') raise SkipTest('PIL not installed')
class TestImage(Document): class TestImage(Document):
image = ImageField(size=(185, 37)) image = ImageField(size=(185, 37))
@ -317,8 +323,8 @@ class FileTest(unittest.TestCase):
t.image.delete() t.image.delete()
def test_image_field_resize_force(self): def test_image_field_resize_force(self):
if PY3: if not HAS_PIL:
raise SkipTest('PIL does not have Python 3 support') raise SkipTest('PIL not installed')
class TestImage(Document): class TestImage(Document):
image = ImageField(size=(185, 37, True)) image = ImageField(size=(185, 37, True))
@ -340,8 +346,8 @@ class FileTest(unittest.TestCase):
t.image.delete() t.image.delete()
def test_image_field_thumbnail(self): def test_image_field_thumbnail(self):
if PY3: if not HAS_PIL:
raise SkipTest('PIL does not have Python 3 support') raise SkipTest('PIL not installed')
class TestImage(Document): class TestImage(Document):
image = ImageField(thumbnail_size=(92, 18)) image = ImageField(thumbnail_size=(92, 18))
@ -409,8 +415,8 @@ class FileTest(unittest.TestCase):
def test_get_image_by_grid_id(self): def test_get_image_by_grid_id(self):
if PY3: if not HAS_PIL:
raise SkipTest('PIL does not have Python 3 support') raise SkipTest('PIL not installed')
class TestImage(Document): class TestImage(Document):