def test_operation(self): instance = UMLObject() operation = "+operation()" instance.add_operation(operation) result = instance.operations()[0] self.assertEquals(result, operation)
def test_sorting(self): instance = UMLObject() data = ["+a", "-z", "#z", "#c", "-b", "+c"] [instance.add_attribute(attr) for attr in data] [instance.add_operation(op+"()") for op in data] attrs = instance.attributes() ops = instance.operations() self.assertEquals(["+a", "+c", "#c","#z","-b","-z"], attrs) self.assertEquals(["+a()", "+c()", "#c()","#z()","-b()","-z()"], ops)