Estoy tratando de generar una advertencia en Python sin que el programa se bloquee/detenga/interrumpa.

Utilizo la siguiente función simple para verificar si el usuario le pasó un número distinto de cero. Si es así, el programa debería advertirles, pero continuar normalmente. Debería funcionar como el siguiente código, pero debería usar la clase Warning(), Error() o Exception() en lugar de imprimir la advertencia manualmente.

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

Si utilizo el siguiente código y paso 0 a la función, el programa falla y nunca se devuelve el valor. En cambio, quiero que el programa continúe normalmente y solo informar al usuario que pasó 0 a la función.

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

Quiero poder probar que se ha lanzado una advertencia probándola por unittest. Si simplemente imprimo el mensaje, no puedo probarlo con assertRaises en unittest.

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

Consulte la documentación de Python: aquí

no deberías raise la advertencia, deberías estar usando warnings módulo. Al elevarlo, está generando un error, en lugar de una advertencia.

De forma predeterminada, a diferencia de una excepción, una advertencia no interrumpe.

Después import warningses posible especificar un Advertencias class al generar una advertencia. Si no se especifica uno, es literalmente UserWarning por defecto.

>>> warnings.warn('This is a default warning.')
<string>:1: UserWarning: This is a default warning.

Para simplemente usar una clase preexistente en su lugar, por ejemplo DeprecationWarning:

>>> warnings.warn('This is a particular warning.', DeprecationWarning)
<string>:1: DeprecationWarning: This is a particular warning.

Crear una clase de advertencia personalizada es similar a crear una clase de excepción personalizada:

>>> class MyCustomWarning(UserWarning):
...     pass
... 
... warnings.warn('This is my custom warning.', MyCustomWarning)

<string>:1: MyCustomWarning: This is my custom warning.

Para la prueba, considere assertWarns o assertWarnsRegex.

Como alternativa, especialmente para aplicaciones independientes, considere la logging módulo. Puede registrar mensajes con un nivel de depurar, información, advertencia, erroretc. Registrar mensajes con un nivel de advertencia o superiores se imprimen de forma predeterminada en stderr.

Solo un pequeño fragmento de demostración

Estaba un poco desconcertado sobre cómo usar warnings.warn, por lo que proporciono aquí una breve demostración. Tenga en cuenta cómo se ejecuta / no se ejecuta la impresión (“código que se ejecuta después de la advertencia”) después warnings.warn / raise UserWarning. Además, warnings.warn imprime el mensaje de advertencia solo una vez.

>>> import warnings
>>> 
>>> warnings.warn("test warning"); print("code running after warning")
<stdin>:1: UserWarning: test warning
code running after warning
>>> 
>>> raise UserWarning('test warning'); print("code running after warning")
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
UserWarning: test warning
>>> 
>>> warnings.warn("test warning"); print("code running after warning")
code running after warning