def test_get_class(self): dot = ModelSchema('My UML model') cls = dot.add_class('My class') assert dot.get_class('My class') == cls
def test_get_class(self): dot = ModelSchema('My UML model') cls = dot.add_class('My class') self.assertEqual(dot.get_class('My class'), cls)