9c9070aea2
* Do not derive from unittest.TestCase (not needed for pytest) * assertRaises -> pytest.raises * Simplify test_empty_dmatrix with test parametrization * setUpClass -> setup_class, tearDownClass -> teardown_class * Don't import unittest; import pytest * Use plain assert * Use parametrized tests in more places * Fix test_gpu_with_sklearn.py * Put back run_empty_dmatrix_reg / run_empty_dmatrix_cls * Fix test_eta_decay_gpu_hist * Add parametrized tests for monotone constraints * Fix test names * Remove test parametrization * Revise test_slice to be not flaky
17 lines
523 B
Python
17 lines
523 B
Python
import numpy as np
|
|
import sys
|
|
sys.path.append("tests/python")
|
|
# Don't import the test class, otherwise they will run twice.
|
|
import test_interaction_constraints as test_ic # noqa
|
|
rng = np.random.RandomState(1994)
|
|
|
|
|
|
class TestGPUInteractionConstraints:
|
|
cputest = test_ic.TestInteractionConstraints()
|
|
|
|
def test_interaction_constraints(self):
|
|
self.cputest.run_interaction_constraints(tree_method='gpu_hist')
|
|
|
|
def test_training_accuracy(self):
|
|
self.cputest.training_accuracy(tree_method='gpu_hist')
|