python - 在不中断程序的情况下在 Python 中发出警告

Question

我正在尝试在 Python 中发出警告而不使程序崩溃/停止/中断。

我使用以下简单函数来检查用户是否向其传递了一个非零数字。如果是这样，程序应该警告他们，但照常继续。它应该像下面的代码一样工作，但应该使用类 Warning()、Error() 或 Exception() 而不是打印警告手动。

def is_zero(i):
   if i != 0:
     print "OK"
   else:
     print "WARNING: the input is 0!"
   return i

如果我使用下面的代码并将 0 传递给函数，程序会崩溃并且永远不会返回该值。相反，我希望程序能够正常运行，并且只通知用户他将 0 传递给了函数。

def is_zero(i):
   if i != 0:
     print "OK"
   else:
     raise Warning("the input is 0!")
   return i

我希望能够测试单元测试是否已抛出警告。如果我只是将消息打印出来，我将无法在 unittest 中使用 assertRaises 对其进行测试。

Answer 1

import warnings
warnings.warn("Warning...........Message")

查看 python 文档:here