a j@sddlZddlmZddlmZddlZddlmZmZddl m Z m Z m Z e dZddZd d Zd%d d Zd&eeed ddZejdeddZejdeddZedkrddlZddlZejddZejdddeZe ej!Z"ej!#dZ$e%e$dkr4e&de$dr4e$d Z'ndZ'e(ee"d!e'e"d!<e"d!rzee"ej!d"Z)e*e)e+d#ne*d$,ej!dS)'N) signature)Optional)enable_halving_search_cvenable_iterative_imputer) all_displaysall_estimators all_functionsznumpydoc.validateccst}t}||D]\}}|dr(qg}t|D]8}|drDq4t||}t|dsbt|tr4||q4|dt |t dD]}||fVqqdS)N___call__)key) rr startswithdirgetattrhasattr isinstancepropertyappendsortedstr)Z estimatorsZdisplaysnameKlassmethods method_objmethodrCE:\soccer\ml_env\Lib\site-packages\sklearn/tests/test_docstrings.pyget_all_methodss      rccs6t}|D]&\}}d|jvr |jd|jVq dS)Nz utils.fixes.)r __module____name__)Z functionsr funcrrrget_all_functions_names%s  r!ccsf|D]\\}}|dvrq|dvrD|durD|durDt||}t|trDq|durV|dvrVq||fVqdS)za Ignore some errors based on the method type. These rules are specific for scikit-learn.)ZRT02ZGL01ZGL02)ZPR02ZGL08N)ZEX01ZSA01ZES01)rrr)errorsrrcodemessagerrrr filter_errors-s    r%)rreturnc Cs|dur.t|drd}n|dur*tdnt|durxt||}ztt|}Wntyfd}Yn0|jd|}nd}|}ddt|d|||d d d d d |dDg}|S)a~Pretty print original docstring and the obtained errors Parameters ---------- res : dict result of numpydoc.validate.validate Klass : {Estimator, Display, None} estimator object or None method : str if estimator is not None, either the method name or None. Returns ------- str String representation of the error. N__init__z0At least one of Klass, method should be providedzM Parsing of the method signature failed, possibly because this is a property.rz file docstringz# Errors css|]\}}d||VqdS)z - {}: {}N)format).0r#r$rrr szrepr_errors..r") r ValueErrorNotImplementedErrorrrr TypeErrorrjoin)resrrobjZ obj_signatureobj_namemsgrrr repr_errorsZs6       r7 function_namecCsHt|}tt|ddd|d<|drDt|d|d}t|dS)z)Check function docstrings using numpydoc.r"functionrzTested function: N)numpydoc_validationvalidatelistr%r7r/)r8requestr3r6rrrtest_function_docstrings  r?z Klass, methodcCsp|j}||jg}|dur"||d|}t|}tt|d||d|d<|drlt|||}t |dS)Nrr")r) rrrr2r;r<r=r%r7r/)rrr>Zbase_import_path import_pathr3r6rrrtest_docstrings     rA__main__z!Validate docstring with numpydoc.) descriptionr@zImport path to validate)helprz(?:[A-Z][a-z]*)+r"r:z#All docstring checks passed for {}!)N)NN)-reinspectrtypingrZpytestZsklearn.experimentalrrZsklearn.utils.discoveryrrrZ importorskipr;rr!r%rr7markZ parametrizer?rArargparsesysArgumentParserparser add_argument parse_argsargsr<r@r3splitZimport_path_sectionslenmatchrr=r6printexitr,rrrrsB    -7