def test_operation(self): instance = UMLObject() operation = Mock() instance.add_operation(operation) result = instance.operations()[0] self.assertEquals(result, operation)
def test_not_eq_operation(self): instance = UMLObject() other = UMLObject() (o1, o2) = (Mock(), Mock()) instance.add_operation(o1) instance.add_operation(o2) self.assertNotEqual(instance, other)
def test_eq_operation(self): instance = UMLObject() other = UMLObject() o1 = MagicMock() o1.__eq__.return_value = True o2 = MagicMock() o2.__eq__.return_value = True instance.add_operation(o1) other.add_operation(o2) self.assertEqual(instance, other)