def test_attribute(self): instance = UMLObject() attribute = "+attribute" instance.add_attribute(attribute) result = instance.attributes()[0] self.assertEquals(attribute, result)
def test_attributes(self): attrs = ["+b", "-a"] o = UMLObject() [o.add_attribute(attr) for attr in reversed(attrs)] instance = Drawable(o) self.assertEqual(attrs, instance.attributes())
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)