o Z&ie@sdZddlZddlZddlmZddlmZddlZddlZddl m Z ddl m Z ee Ze D]\Z\ZZedD] Zeeeq=q2dd d ZeD] Zd eee<qOd d Z dddZejdeddZdS)zCTests for the minimum dependencies in README.rst and pyproject.tomlN) defaultdict)Path)dependent_packages) parse_versionz, zbuild-system.requireszproject.dependencies)buildinstallzproject.optional-dependencies.c Cstd}ttjjj}|d}tj|st d| d@}|D]4}| |}|s-q#| d| d}}|}|tvrWt|}tt|d}||ksWJ|dq#WddS1scwYdS)NzE\.\. \|([A-Za-z-]+)MinVersion\| replace::( [0-9]+\.[0-9]+(\.[0-9]+)?)z README.rstz%The README.rst file is not available.rr has a mismatched version)recompilersklearn__file__parentospathexistspytestskipopenmatchgrouplowerrr) patternZ readme_pathZ readme_fileflineZmatchedpackageversion min_versionrUC:\wamp64\www\opt\env\Lib\site-packages\sklearn/tests/test_min_dependencies_readme.pytest_min_dependencies_readmes*    "r!cCsPtd}|dur g}t|}ttjjj}|d}|s#td| d }| |}Wdn1s8wY| d} |} | D]} | | } qFi} | D]%} d| vr_| d\}}nd| vrk| d\}}nt | d|| |<qQt | t |ksJ| D] \}}t|}tt|d }||vrq||ksJ|d qdS) Ntomllibzpyproject.tomlz pyproject.toml is not available.rb.z>=z==zV not supported yet in this test. Only >= and == are supported for version requirementsrr )rZ importorskip-min_depencies_tag_to_packages_without_versionrr rrrrrloadsplitNotImplementedErrorsorteditemsrr)pyproject_sectionmin_dependencies_tagskip_version_check_forr"Zexpected_packagesZroot_directorypyproject_toml_pathrpyproject_tomlZpyproject_section_keysinfokeyZpyproject_build_min_versions requirementrrZexpected_min_versionrrr check_pyproject_sectionBsF       r3z'min_dependencies_tag, pyproject_sectioncCs$|dkrdgnd}t|||ddS)zFCheck versions in pyproject.toml is consistent with _min_dependencies.rnumpyN)r-)r3)r+r,r-rrr $test_min_dependencies_pyproject_tomlxs  r5)N)__doc__rr collectionsrpathlibrrr Zsklearn._min_dependenciesrZsklearn.utils.fixesrlistr%r*rrextrasr'extraappendZ)min_dependencies_tag_to_pyproject_sectiontagr!r3markZ parametrizer5rrrr s8    & 6