traceback from a warning

Question

I have a code which, at some point shows a warning, I think that it is having a problem calculating a mean()

I would like to know if there is any way to

Answer 1

You can turn warnings into exceptions:

import warnings

warnings.simplefilter("error")

Now instead of printing a warning, an exception will be raised, giving you a traceback.

You can get the same effect with the -W command line switch:

$ python -W error somescript.py

or by setting the PYTHONWARNINGS environment variable:

$ export PYTHONWARNINGS=error

You can play with the other warnings.simplefilter() arguments to be more specific about what warning should raise an exception. You could filter on warnings.RuntimeWarning and a line number, for example.