d686bf52a6
* Faster dask tests * Reuse AllReducer objects in tests. * Faster boost from prediction tests. * Use rmm dask fixture. * Speed up dask demo. * mypy * Format with black. * mypy * Clang-tidy Co-authored-by: Hyunsu Philip Cho <chohyu01@cs.washington.edu>
36 lines
956 B
Python
36 lines
956 B
Python
import os
|
|
import subprocess
|
|
import sys
|
|
import pytest
|
|
sys.path.append("tests/python")
|
|
import testing as tm
|
|
import test_demos as td # noqa
|
|
|
|
|
|
@pytest.mark.skipif(**tm.no_cupy())
|
|
def test_data_iterator():
|
|
script = os.path.join(td.PYTHON_DEMO_DIR, 'quantile_data_iterator.py')
|
|
cmd = ['python', script]
|
|
subprocess.check_call(cmd)
|
|
|
|
|
|
def test_update_process_demo():
|
|
script = os.path.join(td.PYTHON_DEMO_DIR, 'update_process.py')
|
|
cmd = ['python', script]
|
|
subprocess.check_call(cmd)
|
|
|
|
|
|
def test_categorical_demo():
|
|
script = os.path.join(td.PYTHON_DEMO_DIR, 'categorical.py')
|
|
cmd = ['python', script]
|
|
subprocess.check_call(cmd)
|
|
|
|
|
|
@pytest.mark.skipif(**tm.no_dask())
|
|
@pytest.mark.skipif(**tm.no_dask_cuda())
|
|
@pytest.mark.skipif(**tm.no_cupy())
|
|
@pytest.mark.mgpu
|
|
def test_dask_training():
|
|
script = os.path.join(tm.PROJECT_ROOT, 'demo', 'dask', 'gpu_training.py')
|
|
cmd = ['python', script]
|
|
subprocess.check_call(cmd) |