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

This commit is contained in:
Jiaming Yuan 2022-08-15 12:23:12 +08:00 committed by GitHub
parent 9d816d9988
commit 97d89c3ca1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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