U _{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_objmethodr\/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sklearn/tests/test_docstrings.pyget_all_methodss      rccs6t}|D]&\}}d|jkr |jd|jVq dS)Nz utils.fixes.)r __module____name__)Z functionsr funcrrrget_all_functions_names%s  r!ccsf|D]\\}}|dkrq|dkrD|dk rD|dk rDt||}t|trDq|dk rV|dkrVq||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|dkr.t|drd}n|dkr*tdnt|dk rzt||}ztt|}Wntk rhd}YnX|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 fileZ docstringz# Errors css|]\}}d||VqdS)z - {}: {}N)format).0r#r$rrr szrepr_errors..r") r ValueErrorNotImplementedErrorrrr TypeErrorrjoin)resrrobjZ obj_signatureobj_namemsgrrr repr_errorsZs6      r6 function_namecCsHt|}tt|ddd|d<|drDt|d|d}t|dS)z)Check function docstrings using numpydoc.r"functionrzTested function: N)numpydoc_validationvalidatelistr%r6r.)r7requestr2r5rrrtest_function_docstrings  r>z Klass, methodcCsp|j}||jg}|dk r"||d|}t|}tt|d||d|d<|drlt|||}t |dS)Nrr")r) rrrr1r:r;r<r%r6r.)rrr=Zbase_import_path import_pathr2r5rrrtest_docstrings     r@__main__z!Validate docstring with numpydoc.) descriptionr?zImport path to validate)helprz(?:[A-Z][a-z]*)+r"r9z#All docstring checks passed for {}!)N)NN)-reinspectrtypingrZpytestZsklearn.experimentalrrZsklearn.utils.discoveryrrrZ importorskipr:rr!r%rr6markZ parametrizer>r@rargparsesysArgumentParserparser add_argument parse_argsargsr;r?r2splitZimport_path_sectionslenmatchrr<r5printexitr+rrrrsD    -7