o Rh@sddlZddlmZddlmZddlZddlmZmZddl m Z m Z m Z e dZddZd d Zd&d d Zd'd eedefddZejdeddZejdeddZedkrddlZddlZejddZejdddeZe ej!Z"ej!#dZ$e%e$dkre&de$d re$d!Z'ndZ'e(ee"d"e'e"d"<e"d"ree"ej!d#Z)e*e)e+d$dSe*d%,ej!dSdS)(N) signature)Optional)enable_halving_search_cvenable_iterative_imputer) all_displaysall_estimators all_functionsznumpydoc.validateccst}t}||D]@\}}|drq g}t|D]}|dr#qt||}t|ds2t|tr7||q|dt |t dD]}||fVqCq dS)N___call__)key) rr startswithdirgetattrhasattr isinstancepropertyappendsortedstr) estimatorsdisplaysnameKlassmethods method_objmethodr_/home/air/sanwanet/backup_V2/venv/lib/python3.10/site-packages/sklearn/tests/test_docstrings.pyget_all_methodss&       rccs8t}|D]\}}d|jvr|jd|jVqdS)Nz utils.fixes.)r __module____name__) functionsr funcrrrget_all_functions_names%s  r$ccsh|D].\}}|dvr q|dvr#|dur#|dur#t||}t|tr#q|dur,|dvr,q||fVqdS)za Ignore some errors based on the method type. These rules are specific for scikit-learn.)RT02GL01GL02)PR02GL08N)EX01SA01ES01)rrr)errorsrrcodemessagerrrr filter_errors-s     r0rreturnc Cs|durt|dr d}n |durtdt|dur;t||}ztt|}Wn ty2d}Ynw|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 s zrepr_errors..r-) r ValueErrorNotImplementedErrorrrr TypeErrorr!join)resrrobj obj_signatureobj_namemsgrrr repr_errorsZs8      rC function_namecCsHt|}tt|ddd|d<|dr"t|d|d}t|dS)z)Check function docstrings using numpydoc.r-functionrzTested function: N)numpydoc_validationvalidatelistr0rCr:)rDrequestr>rBrrrtest_function_docstrings rKz Klass, methodcCsp|j}||jg}|dur||d|}t|}tt|d||d|d<|dr6t|||}t |dS)Nrr-)r) r r!rr=rGrHrIr0rCr:)rrrJbase_import_path import_pathr>rBrrrtest_docstrings     rN__main__z!Validate docstring with numpydoc.) descriptionrMzImport path to validate)helprz(?:[A-Z][a-z]*)+r-rFz#All docstring checks passed for {}!)N)NN)-reinspectrtypingrpytestsklearn.experimentalrrsklearn.utils.discoveryrrr importorskiprGrr$r0rrCmark parametrizerKrNr!argparsesysArgumentParserparser add_argument parse_argsargsrHrMr>splitimport_path_sectionslenmatchrrIrBprintexitr7rrrrsF    -7