Improve output function

This commit is contained in:
Daniel Rojas 2020-05-24 13:41:18 +02:00
parent ae959f6b92
commit e3d78f3e01

View File

@ -146,10 +146,11 @@ class Harness:
return dot return dot
def output(self, filename, format='pdf', view=True): def output(self, filename, directory='_output', view=False, cleanup=True, format='pdf'):
d = self.create_graph() d = self.create_graph()
d.format = format for f in format:
d.render(filename, view=view) d.format = f
d.render(filename=filename, directory=directory, view=view, cleanup=cleanup)
class Node: class Node: