42 lines
1.6 KiB
Python
42 lines
1.6 KiB
Python
|
"""Tests for making sure experimental imports work as expected."""
|
||
|
|
|
||
|
|
import textwrap
|
||
|
|
|
||
|
|
from sklearn.utils._testing import assert_run_python_script
|
||
|
|
|
||
|
|
|
||
|
|
def test_imports_strategies():
|
||
|
|
# Make sure different import strategies work or fail as expected.
|
||
|
|
|
||
|
|
# Since Python caches the imported modules, we need to run a child process
|
||
|
|
# for every test case. Else, the tests would not be independent
|
||
|
|
# (manually removing the imports from the cache (sys.modules) is not
|
||
|
|
# recommended and can lead to many complications).
|
||
|
|
|
||
|
|
good_import = """
|
||
|
|
from sklearn.experimental import enable_halving_search_cv
|
||
|
|
from sklearn.model_selection import HalvingGridSearchCV
|
||
|
|
from sklearn.model_selection import HalvingRandomSearchCV
|
||
|
|
"""
|
||
|
|
assert_run_python_script(textwrap.dedent(good_import))
|
||
|
|
|
||
|
|
good_import_with_model_selection_first = """
|
||
|
|
import sklearn.model_selection
|
||
|
|
from sklearn.experimental import enable_halving_search_cv
|
||
|
|
from sklearn.model_selection import HalvingGridSearchCV
|
||
|
|
from sklearn.model_selection import HalvingRandomSearchCV
|
||
|
|
"""
|
||
|
|
assert_run_python_script(textwrap.dedent(good_import_with_model_selection_first))
|
||
|
|
|
||
|
|
bad_imports = """
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
with pytest.raises(ImportError, match='HalvingGridSearchCV is experimental'):
|
||
|
|
from sklearn.model_selection import HalvingGridSearchCV
|
||
|
|
|
||
|
|
import sklearn.experimental
|
||
|
|
with pytest.raises(ImportError, match='HalvingRandomSearchCV is experimental'):
|
||
|
|
from sklearn.model_selection import HalvingRandomSearchCV
|
||
|
|
"""
|
||
|
|
assert_run_python_script(textwrap.dedent(bad_imports))
|