def test_operation(self): instance = UMLObject() operation = "+operation()" instance.add_operation(operation) result = instance.operations()[0] self.assertEquals(result, operation)
def test_operations(self): ops = ["+b()", "-a()"] o = UMLObject() [o.add_operation(op) for op in reversed(ops)] instance = Drawable(o) self.assertEqual(ops, instance.operations())
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)