在Python`unittest`中，如何在尝试中发生异常/警告后返回值 - 除了？

Question

这是我的代码。

import unittest 
import warnings 

def function_that_raises_CustomWarning(): 
    warnings.warn("warning") 
    return True 

class test(unittest.TestCase): 

    def test(self): 
     is_this_True = False 
     is_CustomWarning_raised = False 

     try: 
      is_this_True = function_that_raises_CustomWarning() 
     except Warning: 
      is_CustomWarning_raised = True 

     self.assertTrue(is_this_True) 
     self.assertTrue(is_CustomWarning_raised) 

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

在self.assertTrue(is_this_True)is_this_True是False，因此未能通过测试。

我想要的是is_this_True在self.assertTrue(is_this_True)是True。但是，由于返回值在之后function_that_raises_CustomWarning()中发出警告，因此返回值未被“俘获”。

如何返回function_that_raises_CustomWarning()中的值，并且还“捕获”了except中的警告？

Answer 1

当我在Windows上运行3.6代码时，失败是self.assertTrue(is_CustomWarning_raised)。默认情况下，警告不是例外，不能被except:捕获。解决方案是使用assertWarns或assertWarnsRegex。我使用后者来展示如何使用它来添加额外的测试。

import unittest 
import warnings 

def function_that_raises_CustomWarning(): 
    warnings.warn("my warning") 
    return True 

class test(unittest.TestCase): 

    def test(self): 
     is_this_True = False 

     with self.assertWarnsRegex(Warning, 'my warning'): 
      is_this_True = function_that_raises_CustomWarning() 
     self.assertTrue(is_this_True) 


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