def test_attribute(self): instance = UMLObject() attribute = "+attribute" instance.add_attribute(attribute) result = instance.attributes()[0] self.assertEquals(attribute, result)
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)