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)
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)