xgboost/tests/python-gpu/test_gpu_plotting.py
Jiaming Yuan d6ea5cc1ed
Cover approx tree method for categorical data tests. (#7569)
* Add tree to df tests.
* Add plotting tests.
* Add histogram tests.
2022-01-16 11:31:40 +08:00

18 lines
374 B
Python

import sys
import pytest
sys.path.append("tests/python")
import testing as tm
import test_plotting as tp
pytestmark = pytest.mark.skipif(**tm.no_multiple(tm.no_matplotlib(), tm.no_graphviz()))
class TestPlotting:
cputest = tp.TestPlotting()
@pytest.mark.skipif(**tm.no_pandas())
def test_categorical(self):
self.cputest.run_categorical("gpu_hist")