1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
"""Tests for making sure experimental imports work as expected."""
import textwrap
import pytest
from sklearn.utils import _IS_WASM
from sklearn.utils._testing import assert_run_python_script_without_output
@pytest.mark.xfail(_IS_WASM, reason="cannot start subprocess")
def test_import_raises_warning():
code = """
import pytest
with pytest.warns(UserWarning, match="it is not needed to import"):
from sklearn.experimental import enable_hist_gradient_boosting # noqa
"""
pattern = "it is not needed to import enable_hist_gradient_boosting anymore"
assert_run_python_script_without_output(textwrap.dedent(code), pattern=pattern)
|
