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)
