U md@sddlZddlmZddlmZddlZddlmZddlmZddl m Z ddl m Z ddl m Z e d Zd d Zd d Zd(ddZd)eeedddZejdeddZejdeddZedkrddlZddlZejddZejdddeZe ej!Z"ej!#dZ$e%e$d krLe&d!e$d"rLe$d#Z'ndZ'e(ee"d$e'e"d$<e"d$ree"ej!d%Z)e*e)e+d&ne*d',ej!dS)*N) signature)Optional)enable_iterative_imputer)enable_halving_search_cv)all_estimators) all_displays) 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_objmethodrV/home/sam/Atlas/atlas_env/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_errors\s6      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@rsysargparseArgumentParserparser add_argument parse_argsargsr;r?r2splitZimport_path_sectionslenmatchrr<r5printexitr+rrrrsJ         -7