Unverified Commit 195c26fc authored by Nick Miller's avatar Nick Miller Committed by GitHub
Browse files

[python-package] Add specific error messages to `test_dask` tests (#6955)

parent e7c6c437
...@@ -1131,7 +1131,10 @@ def test_model_and_local_version_are_picklable_whether_or_not_client_set_explici ...@@ -1131,7 +1131,10 @@ def test_model_and_local_version_are_picklable_whether_or_not_client_set_explici
local_model = dask_model.to_local() local_model = dask_model.to_local()
assert "client" not in local_model.get_params() assert "client" not in local_model.get_params()
with pytest.raises(AttributeError): no_client_attr_msg = re.compile(
f"{repr(type(local_model).__name__)} object has no attribute '(client|client_)'"
)
with pytest.raises(AttributeError, match=no_client_attr_msg):
local_model.client local_model.client
local_model.client_ local_model.client_
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment