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

[dask] Use an invalid port for test. (#8064)

Browse Source
This commit is contained in:
Jiaming Yuan
2022-07-13 11:59:02 +08:00
committed by GitHub
parent 0bdaca25ca
commit 8959622836
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/python/test_with_dask.py
View File

@@ -1,6 +1,7 @@
"""Copyright 2019-2022 XGBoost contributors""" """Copyright 2019-2022 XGBoost contributors"""
from pathlib import Path from pathlib import Path
import pickle import pickle
import socket
import testing as tm import testing as tm
import pytest import pytest
import xgboost as xgb import xgboost as xgb
@@ -1243,11 +1244,11 @@ class TestWithDask:
os.remove(after_fname) os.remove(after_fname)
with dask.config.set({'xgboost.foo': "bar"}): with dask.config.set({'xgboost.foo': "bar"}):
with pytest.raises(ValueError): with pytest.raises(ValueError, match=r"Unknown configuration.*"):
xgb.dask.train(client, {}, dtrain, num_boost_round=4) xgb.dask.train(client, {}, dtrain, num_boost_round=4)
with dask.config.set({'xgboost.scheduler_address': "127.0.0.1:22"}): with dask.config.set({'xgboost.scheduler_address': "127.0.0.1:foo"}):
with pytest.raises(PermissionError): with pytest.raises(socket.gaierror, match=r".*not known.*"):
xgb.dask.train(client, {}, dtrain, num_boost_round=1) xgb.dask.train(client, {}, dtrain, num_boost_round=1)
def run_updater_test( def run_updater_test(