a p=ic8@sddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZddlmZddlmZmZGd d d ejZGd d d ejZGd ddejZeZGdddejZGdddZGddde Z e Z!Gddde Z"Gdddej#j$Z%dS)N)doctests) ErrorClassErrorClassPlugin)Plugin)src)get_package_name)KnownFailureExceptionKnownFailureTestc@seZdZddZddZdS)NumpyDocTestFindercCs|dur dSt|r"|j|juSt|r8|j|jkSt|rN|j|jkSt|rh|j|j j jkSt |dur|t |uSt |dr|j|jkSt |trdStddS)zY Return true if the given object is defined in the given module. NT __module__z"object must be a class or function)inspect isfunction__dict__ __globals__ isbuiltin__name__r isclassismethod__self__ __class__ getmodulehasattr isinstanceproperty ValueError)selfmoduleobjectrs/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/numpy/testing/_private/noseclasses.py _from_modules"          zNumpyDocTestFinder._from_modulec CsDtj||||||||ddlm}m} m} m} m} | |r|j r|j D]H\} }|d| }||st| |rN| ||rN||||||||qN| |r@|j r@|j D]\} }t |trt|| }t |trt|| j}| |s| |s| |st |tr| ||r|d| } |||| ||||qdS)zm Find tests for the given object and any contained objects, and add them to `tests`. r) isroutinerismodulerr.N)doctest DocTestFinder_findr r"rr#rrZ_recurseritemsr!r staticmethodgetattr classmethod__func__r)rtestsobjnamerZ source_linesglobsseenr"rr#rrZvalnamevalZvalname1rrr r':s<      zNumpyDocTestFinder._findN)rr __qualname__r!r'rrrr r sr c@seZdZddZdS)NumpyOutputCheckercCsztj||||}|svd|vr"dS|dd}|dd}dD]$}|d|d}|d|d}q>tj||||}|S)Nz#randomTz'>z'<)z'dd|jjD|j_dS)NcSsg|]}|jdkr|qS)r%)r/.0prrr s z*NumpyDoctest.configure..) r configuretest_finder_classfinderr%Z DocTestParserrLenabledpluginsrrIconfigrrr rRs   zNumpyDoctest.configurecCsXttj|j}tddtd|_d|vrT|d}|d}t ||ji|g|j|<dS)a` Configure `test` object to set test context We set the numpy / scipy standard doctest namespace Parameters ---------- test : test object with ``globs`` dictionary defining namespace Returns ------- None Notes ----- `test` object modified in place __main__) __builtins____file__rnpZscipyr$N) rospathdirnamefilenamerZnumpyr0split __import__)rrFpkg_namerPp2rrr set_test_contexts  zNumpyDoctest.set_test_contextccs||jstjd|dSz|j|}WntyBYdS0|sLdS|t |j }|D]>}|j snqb|j sz||_ | ||j||j||jdVqbdS)NzDoctest doesn't want module %s)r<rCrG)matchesrnpdlogdebugrTfindAttributeErrorsortrr[Zexamplesrargdoctest_case_classdoctest_optflagsout_check_classrJ)rrr-Z module_filerFrrr loadTestsFromModules,    z NumpyDoctest.loadTestsFromModulecCstjfitdSN)rbZset_printoptions print_staterrrr afterContextszNumpyDoctest.afterContextcCs(tj|}||jvrdStj||S)NF)r^r_basenamedoctest_ignoreriDoctestwantFile)rfilebnrrr rzs  zNumpyDoctest.wantFileN)rr r3r/scorer%ZNORMALIZE_WHITESPACEELLIPSISrprxr?ror4rqr rSr^environrIrRrgrrrvrzrrrr rHs   *rHc@s6eZdZdZdZdZdZd ddZdd Zd d Z d S) Unpluggerzi Nose plugin to remove named plugin late in loading By default it removes the "doctest" plugin. Z unpluggerTir%cCs ||_dSrs) to_unplug)rrrrr rEszUnplugger.__init__cCsdSrsrrKrrr rIszUnplugger.optionscsfdd|jjD|j_dS)Ncsg|]}|jjkr|qSr)r/rrNrurr rQs z'Unplugger.configure..)rVrWrrur rRszUnplugger.configureN)r%) rr r3__doc__r/rUr}rErIrRrrrr r s rc@s8eZdZdZdZeedddZej fddZ dd Z d S) KnownFailurePlugina1Plugin that installs a KNOWNFAIL error class for the KnownFailureClass exception. When KnownFailure is raised, the exception will be logged in the knownfail attribute of the result, 'K' or 'KNOWNFAIL' (verbose) will be output, and the exception will not be counted as an error or failure.TZ KNOWNFAILF)labelZ isfailurecCs$d}|jddd||ddddS)NZNOSE_WITHOUT_KNOWNFAILz--no-knownfail store_true noKnownFailFz3Disable special handling of KnownFailure exceptions)actiondestdefaulthelp) add_optionget)rrLrMZenv_optrrr rI-s  zKnownFailurePlugin.optionscCs*|js dS||_t|dd}|r&d|_dS)NrF)Z can_configureconfr*rU)rrIrdisablerrr rR4s  zKnownFailurePlugin.configureN) rr r3rrUrr Z knownfailr^rrIrRrrrr r"src@seZdZdZddZdS)FPUModeCheckPluginzx Plugin that checks the FPU mode before and after each test, raising failures if the test changed the mode. csddlmfdd}|S)Nr) get_fpu_modecsZ}|}||krVztd||Wn"tyT|tYn0dS)Nz6FPU mode changed from {0:#x} to {1:#x} during the test)rFAssertionErrorformatZ addFailuresysexc_info)resultZold_modeZnew_moderrFrr runHs  z/FPUModeCheckPlugin.prepareTestCase..run)Znumpy.core._multiarray_testsr)rrFrrrr prepareTestCaseEs  z"FPUModeCheckPlugin.prepareTestCaseN)rr r3rrrrrr r?src@seZdZddZdS)NumpyTestProgramcCsh|jdur(tjj|jj|jj|jd|_|jj|j}|durF||_|j |j |_ |j |_ |j S)aRun Tests. Returns true on success, false on failure, and sets self.success to the same value. Because nose currently discards the test result object, but we need to return it to the user, override TestProgram.runTests to retain the result N)stream verbosityrX)Z testRunnernosecoreZTextTestRunnerrXrrrVZprepareTestRunnerrrFrZ wasSuccessfulsuccess)rZ plug_runnerrrr runTests[s   zNumpyTestProgram.runTestsN)rr r3rrrrr rZsr)&r^rr%r rbrZ nose.pluginsrriZnose.plugins.errorclassrrZnose.plugins.baserZ nose.utilrZ nosetesterrutilsr r r&r r8r4rDr?Zget_printoptionsrtryrHrrZ KnownFailurerrZ TestProgramrrrrr s*     O x