def test_add_operation_class_returned_for_operation_add_(self): patch = Patch([]) add_operation_class = patch.get_operation_class('add') self.assertEqual(add_operation_class, AddOperation)
def test_patch_exception_raised_when_operation_is_unsupported(self): patch = Patch([]) with self.assertRaises(PatchException): patch.get_operation_class('mock')