在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()