scikit-learn/sklearn/externals/joblib/testing.py

20 lines
410 B
Python

"""
Helper for testing.
"""
import sys
import warnings
import os.path
def warnings_to_stdout():
""" Redirect all warnings to stdout.
"""
showwarning_orig = warnings.showwarning
def showwarning(msg, cat, fname, lno, file=None, line=0):
showwarning_orig(msg, cat, os.path.basename(fname), line, sys.stdout)
warnings.showwarning = showwarning
#warnings.simplefilter('always')