Сделать assertEqual не преминуло на RuntimeErrors, начиная с определенной строкой

Question

Я пишу UnitTests в Python с использованием assertEqual:

def test_SetOwner(self, owner): 
    self.objectid.SetOwner(owner) 
    self.assertEqual(self.objectid.GetOwner(), owner) 

Это работает в большинстве случаев, но в некоторых конкретных случаях это не удается с RuntimeError начиная с Could not execute... , Однако ожидается, что en не должно вызывать ошибку в пакете unittest.

Есть ли способ каким-то образом сделать assertEqual не сбой на RuntimeErrors, начиная с этой конкретной строки, все еще захватывая все остальные RuntimeErrors? Или есть лучший способ справиться с этим?

Answer 1

Вы можете сделать это, используя блок try/except, и проверить два случая отдельно.

def test_SetOwner(self, owner): 
    self.objectid.SetOwner(owner) 
    try: 
     owner_prime = self.objectid.GetOwner() 
    except RuntimeError as e: 
     self.assertTrue(e.args[0].startswith("Could not execute")) 
    else: 
     self.assertEqual(owner, owner_prime) 

Но, как упоминалось в моем комментарии выше, это не похоже на очень хороший модульный тест. Что, если, например, в какой-то момент ошибка вводится в GetOwner(), так что она всегда поднимает значение «Не удалось выполнить» RuntimeError? Это почти наверняка не будет желательным поведением, но ваш тест будет продолжать проходить.