Python unittest条件断言

Question

我需要测试assertEqual(var, 'a')或assertEqual(var2, 'a')是否为真。

我不能像以下那样写它们：

assertEqual(var, 'a')
assertEqual(var2, 'a')

因为这不是测试的范围。无论var = 'a'还是var2 = 'a'，测试都应该成功，但在这种情况下，如果例如var2 = 'b'则会失败。

那么我怎么能写这个测试呢？因为如果我使用if assertEqual(var, 'a') or assertEqual(var2, 'a'):，我应该在里面输入什么？

Answer 1

您可以使用assertTrue：

sed

Answer 2

我遇到了一个稍微不同的问题：如果 var 是 'a' 或 'b'

我使用了 assertIn(var, {'a', 'b'})