def test_create_thumb_prev(self): tp = create_thumbnail_and_preview(example_png, (6, 6), (100, 100)) self.assertIsNotNone(tp.thumbnail) self.assertIsNone(tp.preview) self.assertEqual(determine_figure_type(tp.thumbnail), FigureType.PNG) with closing(StringIO(tp.thumbnail)) as f: self.assertEqual(Image.open(f).size, (3, 6)) tp = create_thumbnail_and_preview(example_png, (80, 80), (12, 12)) self.assertIsNotNone(tp.thumbnail) self.assertIsNotNone(tp.preview) self.assertEqual(determine_figure_type(tp.thumbnail), FigureType.PNG) self.assertEqual(determine_figure_type(tp.preview), FigureType.PNG) with closing(StringIO(tp.thumbnail)) as f: self.assertEqual(Image.open(f).size, (40, 80)) with closing(StringIO(tp.preview)) as f: self.assertEqual(Image.open(f).size, (6, 12))
def test_create_thumb_jpeg(self): tp = create_thumbnail_and_preview(example_jpeg, (10, 10), (20, 20)) for image in (tp.thumbnail, tp.preview): self.assertIsNotNone(image) self.assertEqual(determine_figure_type(image), FigureType.PNG)