测试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是链接到文档。