2016-05-01 17:27:43 +09:00

23 lines
415 B
Python

# coding: utf-8
import nose
from xgboost.compat import SKLEARN_INSTALLED, PANDAS_INSTALLED
def _skip_if_no_sklearn():
if not SKLEARN_INSTALLED:
raise nose.SkipTest()
def _skip_if_no_pandas():
if not PANDAS_INSTALLED:
raise nose.SkipTest()
def _skip_if_no_matplotlib():
try:
import matplotlib.pyplot as plt # noqa
except ImportError:
raise nose.SkipTest()