Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Import data lazily during tests. (#9176)

Browse Source
This commit is contained in:
Jiaming Yuan
2023-05-23 03:58:31 +08:00
committed by GitHub
parent 6274fba0a5
commit 3913ff470f
8 changed files with 66 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/python-gpu/test_gpu_linear.py
View File

@@ -28,8 +28,7 @@ def train_result(param, dmat, num_rounds):
class TestGPULinear:
@given(parameter_strategy, strategies.integers(10, 50),
tm.dataset_strategy)
@given(parameter_strategy, strategies.integers(10, 50), tm.make_dataset_strategy())
@settings(deadline=None, max_examples=20, print_blob=True)
def test_gpu_coordinate(self, param, num_rounds, dataset):
assume(len(dataset.y) > 0)
@@ -45,7 +44,7 @@ class TestGPULinear:
@given(
parameter_strategy,
strategies.integers(10, 50),
tm.dataset_strategy,
tm.make_dataset_strategy(),
strategies.floats(1e-5, 0.8),
strategies.floats(1e-5, 0.8)
)

14
tests/python-gpu/test_gpu_prediction.py
View File

@@ -265,8 +265,9 @@ class TestGPUPredict:
base_margin = cudf.Series(rng.randn(rows))
self.run_inplace_base_margin(booster, dtrain, X, base_margin)
@given(strategies.integers(1, 10),
tm.dataset_strategy, shap_parameter_strategy)
@given(
strategies.integers(1, 10), tm.make_dataset_strategy(), shap_parameter_strategy
)
@settings(deadline=None, max_examples=20, print_blob=True)
def test_shap(self, num_rounds, dataset, param):
if dataset.name.endswith("-l1"): # not supported by the exact tree method
@@ -281,8 +282,9 @@ class TestGPUPredict:
assume(len(dataset.y) > 0)
assert np.allclose(np.sum(shap, axis=len(shap.shape) - 1), margin, 1e-3, 1e-3)
@given(strategies.integers(1, 10),
tm.dataset_strategy, shap_parameter_strategy)
@given(
strategies.integers(1, 10), tm.make_dataset_strategy(), shap_parameter_strategy
)
@settings(deadline=None, max_examples=10, print_blob=True)
def test_shap_interactions(self, num_rounds, dataset, param):
if dataset.name.endswith("-l1"): # not supported by the exact tree method
@@ -335,7 +337,7 @@ class TestGPUPredict:
np.testing.assert_equal(cpu_leaf, gpu_leaf)
@given(predict_parameter_strategy, tm.dataset_strategy)
@given(predict_parameter_strategy, tm.make_dataset_strategy())
@settings(deadline=None, max_examples=20, print_blob=True)
def test_predict_leaf_gbtree(self, param, dataset):
# Unsupported for random forest
@@ -346,7 +348,7 @@ class TestGPUPredict:
param['tree_method'] = 'gpu_hist'
self.run_predict_leaf_booster(param, 10, dataset)
@given(predict_parameter_strategy, tm.dataset_strategy)
@given(predict_parameter_strategy, tm.make_dataset_strategy())
@settings(deadline=None, max_examples=20, print_blob=True)
def test_predict_leaf_dart(self, param: dict, dataset: tm.TestDataset) -> None:
# Unsupported for random forest

10
tests/python-gpu/test_gpu_updaters.py
View File

@@ -48,7 +48,9 @@ class TestGPUUpdatersMulti:
class TestGPUUpdaters:
cputest = test_up.TestTreeMethod()
@given(hist_parameter_strategy, strategies.integers(1, 20), tm.dataset_strategy)
@given(
hist_parameter_strategy, strategies.integers(1, 20), tm.make_dataset_strategy()
)
@settings(deadline=None, max_examples=50, print_blob=True)
def test_gpu_hist(self, param, num_rounds, dataset):
param["tree_method"] = "gpu_hist"
@@ -150,7 +152,7 @@ class TestGPUUpdaters:
@given(
hist_parameter_strategy,
strategies.integers(1, 20),
tm.dataset_strategy
tm.make_dataset_strategy(),
)
@settings(deadline=None, max_examples=20, print_blob=True)
def test_gpu_hist_device_dmatrix(
@@ -171,7 +173,7 @@ class TestGPUUpdaters:
@given(
hist_parameter_strategy,
strategies.integers(1, 3),
tm.dataset_strategy
tm.make_dataset_strategy(),
)
@settings(deadline=None, max_examples=10, print_blob=True)
def test_external_memory(self, param, num_rounds, dataset):
@@ -213,7 +215,7 @@ class TestGPUUpdaters:
np.testing.assert_allclose(predictions, 0.0, 1e-6)
@pytest.mark.mgpu
@given(tm.dataset_strategy, strategies.integers(0, 10))
@given(tm.make_dataset_strategy(), strategies.integers(0, 10))
@settings(deadline=None, max_examples=10, print_blob=True)
def test_specified_gpu_id_gpu_update(self, dataset, gpu_id):
param = {'tree_method': 'gpu_hist', 'gpu_id': gpu_id}