U
    }±Ëht  ã                   @   s$   d Z ddlZddlmZ dd„ ZdS )z<Tests for making sure experimental imports work as expected.é    N)Úassert_run_python_scriptc                  C   s:   d} t t | ¡ƒ d}t t |¡ƒ d}t t |¡ƒ d S )Nzs
    from sklearn.experimental import enable_iterative_imputer
    from sklearn.impute import IterativeImputer
    z
    import sklearn.ensemble
    from sklearn.experimental import enable_iterative_imputer
    from sklearn.impute import IterativeImputer
    a?  
    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
    )r   ÚtextwrapÚdedent)Zgood_importZgood_import_with_ensemble_firstZbad_imports© r   ú\/tmp/pip-unpacked-wheel-ig1s1lm8/sklearn/experimental/tests/test_enable_iterative_imputer.pyÚtest_imports_strategies   s    
r   )Ú__doc__r   Zsklearn.utils._testingr   r   r   r   r   r   Ú<module>   s   