def test_resize_broken_file(self):
# broken images are not resized
broken = os.path.join(self.tempdir, 'broken.jpg')
open(broken, 'w').write('blah blah')
self.assertEqual(
resize(broken, 64), None)
return
def create_resized_image(self, response, resolution):
"""Create a resized version of current content image.
"""
orig_file = tempfile.TemporaryFile()
for item in response.copy().app_iter:
orig_file.write(item)
orig_file.seek(0)
return resize(orig_file, resolution)
def test_resize(self):
# we can resize images
im = Image.open(sample_jpg)
self.assertEqual(im.size, (128, 128))
im_type, result = resize(sample_jpg, 64)
new_size = Image.open(result).size
self.assertEqual(new_size, (64, 64))
self.assertEqual(im_type, 'JPEG')
return
def test_resize_invalid_img_path(self):
# invalid paths result in `None
self.assertEqual(
resize('not-a-path', 64), None)
return
def test_resize_too_small(self):
# if resolution >= image width, we get no new image
self.assertEqual(resize(sample_jpg, 128), None)
self.assertEqual(resize(sample_jpg, 256), None)
return
def test_resize_no_resolution(self):
# w/o resolution, no resizing
self.assertEqual(resize(sample_jpg, None), None)
return
