如何在生成器对象中使用unittest的self.assertRaise异常？

Question

我收到了一个我想单元测试的生成器对象。它经历一个循环，并且在循环结束时某个变量仍然为0，我引发一个异常。我想单元测试，但我不知道如何。 拿这个案例生成：

class Example(): 
    def generatorExample(self): 
     count = 0 
     for int in range(1,100): 
      count += 1 
      yield count 
     if count > 0: 
      raise RuntimeError, 'an example error that will always happen' 

我想这样做是

class testExample(unittest.TestCase): 
    def test_generatorExample(self): 
     self.assertRaises(RuntimeError, Example.generatorExample) 

然而，一台发电机的对象不是calable，这给

TypeError: 'generator' object is not callable 

那么，你如何测试一个生成器函数中是否引发异常？

Answer 1

assertRaises是因为Python 2.7上下文管理器，所以你可以做这样的：

class testExample(unittest.TestCase): 

    def test_generatorExample(self): 
     with self.assertRaises(RuntimeError): 
      list(Example().generatorExample()) 

如果你有Python的< 2.7，那么你可以使用lambda用尽发电机：

self.assertRaises(RuntimeError, lambda: list(Example().generatorExample()))