File: test_gpu_plotting.py

package info (click to toggle)
xgboost 3.0.0-1
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 13,796 kB
  • sloc: cpp: 67,502; python: 35,503; java: 4,676; ansic: 1,426; sh: 1,320; xml: 1,197; makefile: 204; javascript: 19
file content (18 lines) | stat: -rw-r--r-- 388 bytes parent folder | download | duplicates (2) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
 
import sys

import pytest

from xgboost import testing as tm

sys.path.append("tests/python")
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")