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

Turn warning messages into Python warnings. (#9387)

Browse Source
This commit is contained in:
Jiaming Yuan
2023-07-15 07:46:43 +08:00
committed by GitHub
parent 04aff3af8e
commit 9da5050643
3 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/python/test_with_sklearn.py
View File

@@ -3,6 +3,7 @@ import os
import pickle
import random
import tempfile
import warnings
from typing import Callable, Optional
import numpy as np
@@ -1091,25 +1092,22 @@ def test_constraint_parameters():
)
@pytest.mark.filterwarnings("error")
def test_parameter_validation():
reg = xgb.XGBRegressor(foo='bar', verbosity=1)
reg = xgb.XGBRegressor(foo="bar", verbosity=1)
X = np.random.randn(10, 10)
y = np.random.randn(10)
with tm.captured_output() as (out, err):
with pytest.warns(Warning, match="foo"):
reg.fit(X, y)
output = out.getvalue().strip()
assert output.find('foo') != -1
reg = xgb.XGBRegressor(n_estimators=2, missing=3,
importance_type='gain', verbosity=1)
reg = xgb.XGBRegressor(
n_estimators=2, missing=3, importance_type="gain", verbosity=1
)
X = np.random.randn(10, 10)
y = np.random.randn(10)
with tm.captured_output() as (out, err):
reg.fit(X, y)
output = out.getvalue().strip()
assert len(output) == 0
with warnings.catch_warnings():
reg.fit(X, y)
def test_deprecate_position_arg():