9ac163d0bb
* Allow import via python datatable. * Write unit tests * Refactor dt API functions * Refactor python code * Lint fixes * Address review comments
28 lines
506 B
Python
28 lines
506 B
Python
# coding: utf-8
|
|
|
|
import nose
|
|
|
|
from xgboost.compat import SKLEARN_INSTALLED, PANDAS_INSTALLED, DT_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_dt():
|
|
if not DT_INSTALLED:
|
|
raise nose.SkipTest()
|
|
|
|
|
|
def _skip_if_no_matplotlib():
|
|
try:
|
|
import matplotlib.pyplot as _ # noqa
|
|
except ImportError:
|
|
raise nose.SkipTest()
|