在Python中测试 - 如何在使用unittest测试中使用assertRaises？

Question

我想在Python中使用unittest做一个简单的测试，看看如果一个类如果得到一个不适合的构造函数输入的话就会抛出一个异常。这个类看起来是这样的：

class SummaryFormula: 
    def __init__(self, summaryFormula): 
     self.atoms = {} 
     for atom in re.finditer("([A-Z][a-z]{0,2})(\d*)", summaryFormula): 
      symbol = atom.group(1) 
      count = atom.group(2) 

      if pocet != "": 
       self.atoms[ symbol] = int(count) 
      else: 
       self.atoms[ symbol] = 1 

我的测试如下：

class ConstructorTestCase(unittest.TestCase): 
     def testEmptyString(self): 
     self.assertRaises(TypeError, ukol1.SummaryFormula(), "testtest") 

    if __name__ == '__main__': 
     unittest.main() 

所有我想要的是测试失败，这意味着不合适输入为构造函数中的异常没有被处理。

取而代之，我得到一个错误：__init__() takes exactly 2 arguments (1 given)。

我错过了什么？我应该指定什么第二个参数？

此外，我应该使用什么类型的错误来处理异常，即我的正则表达式无法匹配的输入传递给构造函数？

谢谢，托马斯

Answer 1

那是因为你的类需要一个参数，而实例化对象

，而你是通过

ukol1.SummaryFormula() 

你应该已经传递参数summaryFormula它。

ukol1.SummaryFormula(someSummaryFormula) 

而且混乱是因为你的类名是SummaryFormula和你传递给__init__参数也SummaryFormula

或这应该是

self.assertRaises(TypeError, ukol1.SummaryFormula, "testtest") 

Answer 2

assertRaises是有点混乱，因为你需要给它可调用的，而不是表达这个调用的表达式。

你的代码更改为：

self.assertRaises(TypeError, ukol1.SummaryFormula, "testtest") 

在你的代码，你自己调用构造函数，它会引发你没有足够的论据例外。相反，您需要给assertRaises可调用（ukol1.SummaryFormula）以及用（“testtest”）调用它的参数。然后它可以调用它，捕获并检查异常。

Answer 3

一个更通用的替代格式是

args=['testtest'] 
kwargs = {} 
self.assertRaises(TypeError, ukol1.SummaryFormula, *args, **kwargs) 

这是有用的，如果你的构造函数是多态的，你要循环的误写的参数不同的方式列表，如：

arg_lists = [ 
    ['testtest'], 
    ['anothertest'], 
    ['YAT'], 
] 
for args in arg_lists: 
    self.assertRaises(TypeError, ukol1.SummaryFormula, *args) 

Answer 4

更为pythonic的方式是使用with命令（在Python 2中添加。7）：

with self.assertRaises(SomeException): 
    do_something() 

文档：https://docs.python.org/2/library/unittest.html#unittest.TestCase.assertRaises

Answer 5

由于没有其他的答案指向你如何使用它封装引起异常的代码，这里是你如何能做到这一点的情况下。

with self.assertRaises(ValueError) as ctx: 
    <some code that throws an exception> 

expected_msg = 'foo_bar_baz' 
self.assertEquals(ctx.exception.message, expected_msg) 

感兴趣的属性在此unittest.case._AssertRaisesContext，分别是：

• 例外
• 预计
• expected_regexp
• failureException