def test_pdf_to_svg(self):
if not exists(get_config().get('utilities', 'pdftocairo')):
self.skipTest('pdftocairo not available')
page = pdf_to_svg(example_pdf, page=1)
self.assertEqual(determine_figure_type(page), FigureType.SVG)
def test_pdf_to_png_ghostscript(self):
if not exists(get_config().get('utilities', 'ghostscript')):
self.skipTest('Ghostscript not available')
# Render using "ghostscript".
pages = pdf_to_png(example_pdf, renderer='ghostscript')
self.assertEqual(len(pages), 1)
self.assertEqual(determine_figure_type(pages[0]), FigureType.PNG)
def test_pdf_to_png_cairo(self):
if not exists(get_config().get('utilities', 'pdftocairo')):
self.skipTest('pdftocairo not available')
# Repeat using "pdftocairo".
pages = pdf_to_png(example_pdf, renderer='pdftocairo')
self.assertEqual(len(pages), 1)
self.assertEqual(determine_figure_type(pages[0]), FigureType.PNG)
def test_ps_to_png(self):
if not exists(get_config().get('utilities', 'ghostscript')):
self.skipTest('Ghostscript not available')
pages = ps_to_png(example_eps)
self.assertEqual(len(pages), 1)
self.assertEqual(determine_figure_type(pages[0]), FigureType.PNG)
def test_graphviz_to_png(self):
example_dot = 'graph G {x -- y}'
invalid_dot = 'graph G {x -> y}'
out = graphviz_to_png(example_dot)
self.assertEqual(determine_figure_type(out), FigureType.PNG)
with self.assertRaises(ConversionError):
graphviz_to_png(invalid_dot)
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_figure_type(self):
self.assertEqual(determine_figure_type(example_png), FigureType.PNG)
self.assertEqual(determine_figure_type(example_jpeg), FigureType.JPEG)
self.assertEqual(determine_figure_type(example_pdf), FigureType.PDF)
self.assertEqual(determine_figure_type(example_eps), FigureType.PS)
with self.assertRaisesRegexp(UserError, 'text/plain'):
determine_figure_type(b'PLAIN TEXT')
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)
