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