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