Тестирование self.assertEqual один или другой

Question

В тестах на Python, я знаю, как проверить ожидаемое значение переменной:

def test_smth(self): 
    var = 4 
    self.assertEqual(var, 4) 

Но можно проверить с той или иной переменной без затопления код с if?

Что-то вроде:

def test_smth(self): 
    var = 5 
    self.assertEqual(var, 4 or 5) # this doesn't work 

Answer 1

Благодаря jonrsharpe.

self.assertIn(var, (4, 5)) 

Here является ссылкой на документы.