Example #1
0
 def test_export_image(self):
     graph = objects_reachable_from([[1, 2, 3], [4, [5, 6]]])
     tempdir = tempfile.mkdtemp()
     try:
         filename = os.path.join(tempdir, "output.png")
         graph.export_image(filename)
         self.assertTrue(os.path.exists(filename))
     finally:
         shutil.rmtree(tempdir)
Example #2
0
 def test_export_image_explicit_format(self):
     graph = objects_reachable_from([[1, 2, 3], [4, [5, 6]]])
     tempdir = tempfile.mkdtemp()
     try:
         # Deliberately using a misleading extension...
         filename = os.path.join(tempdir, "output.png")
         graph.export_image(filename, format="svg")
         self.assertTrue(os.path.exists(filename))
         self.assertTrue(is_svg(filename))
     finally:
         shutil.rmtree(tempdir)