def do_diagram(self, inp):
image_name = input('Image name/type:')
MyCreator(image_name, inp).create_diagram()
def do_deletediagram(self, inp):
MyCreator(inp, 'pass').delete_diagram()
def test_validate_file(self):
dc = MyCreator('ClassDiagram.png', 'TestClass.py')
self.assertTrue(dc.validate_file_name())