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)
