40 lines
1.4 KiB
Python
40 lines
1.4 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_iterative_imputer
|
||
|
|
from sklearn.impute import IterativeImputer
|
||
|
|
"""
|
||
|
|
assert_run_python_script(textwrap.dedent(good_import))
|
||
|
|
|
||
|
|
good_import_with_ensemble_first = """
|
||
|
|
import sklearn.ensemble
|
||
|
|
from sklearn.experimental import enable_iterative_imputer
|
||
|
|
from sklearn.impute import IterativeImputer
|
||
|
|
"""
|
||
|
|
assert_run_python_script(textwrap.dedent(good_import_with_ensemble_first))
|
||
|
|
|
||
|
|
bad_imports = """
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
with pytest.raises(ImportError, match='IterativeImputer is experimental'):
|
||
|
|
from sklearn.impute import IterativeImputer
|
||
|
|
|
||
|
|
import sklearn.experimental
|
||
|
|
with pytest.raises(ImportError, match='IterativeImputer is experimental'):
|
||
|
|
from sklearn.impute import IterativeImputer
|
||
|
|
"""
|
||
|
|
assert_run_python_script(textwrap.dedent(bad_imports))
|