a h@sdZddlZddlZddlmZddlmZddlZddlZddl m Z ddl m Z ee Ze D]*\Z\ZZedD]Zeeeqzqddd d ZeD]Zd eee<qd d ZdddZejdeddZdS)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|s6t d| d~}|D]h}| |}|sZqF| d| d}}|}|tvrFt|}tt|d}||ksFJ|dqFWdn1s0YdS) NzN(\.\. \|)(([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_versionr f/var/www/html/assistant/venv/lib/python3.9/site-packages/sklearn/tests/test_min_dependencies_readme.pytest_min_dependencies_readmes$    r"cCsZtd}|durg}t|}ttjjj}|d}|sFtd| d}| |}Wdn1sp0Y| d} |} | D] } | | } qi} | D]J} d| vr| d\}}n&d| vr| d\}}nt | d|| |<qt | t |ksJ| D]H\}}t|}tt|d }||vr:q ||ks J|d q dS) 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_versionrrrrrrrloadsplitNotImplementedErrorsorteditemsrr)pyproject_sectionmin_dependencies_tagskip_version_check_forr#Zexpected_packagesZroot_directorypyproject_toml_pathrpyproject_tomlZpyproject_section_keysinfokeyZpyproject_build_min_versions requirementrrZexpected_min_versionr r r!check_pyproject_sectionBsB   (    r4z'min_dependencies_tag, pyproject_sectioncCs$|dkrdgnd}t|||ddS)zFCheck versions in pyproject.toml is consistent with _min_dependencies.rnumpyN)r.)r4)r,r-r.r r r!$test_min_dependencies_pyproject_tomlxs r6)N)__doc__rr collectionsrpathlibrrrZsklearn._min_dependenciesrZsklearn.utils.fixesrlistr&r+rrextrasr(extraappendZ)min_dependencies_tag_to_pyproject_sectiontagr"r4markZ parametrizer6r r r r!s4    & 6