gKpdZddlZddlZddlZddlZddlmZmZgdZ dZ d dZ Gdd Z d Z dS) ze Nose test running. This module implements ``test()`` and ``bench()`` functions for NumPy modules. N) import_nosesuppress_warnings)get_package_namerun_module_suite NoseTester _numpy_testerrrrcx|dd}g}d|vsd|vrDtj|\}}|dvrn||d|v@d|vD|sd|vrdSdS||ddr|dd |S) a& Given a path where a package is installed, determine its name. Parameters ---------- filepath : str Path to a file. If the determination fails, "numpy" is returned. Examples -------- >>> np.testing.nosetester.get_package_name('nonsense') 'numpy' N site-packages dist-packages)r r scipynumpyrz.egg.)ospathsplitappendreverseendswithpopjoin)filepathfullpathpkg_namep2s W/granary/edev/quantum/lib/python3.11/site-packages/numpy/testing/_private/nosetester.pyrrs {HH X % %H)D)Dw}}X.. " 3 3 3  X % %H)D)D  h  77 {F## Q 88H  c|8tjd}|jdd}|t|tj|gz}n||gz}t }ddlm}| ||gdS)a Run a test module. Equivalent to calling ``$ nosetests `` from the command line Parameters ---------- file_to_run : str, optional Path to test module, or None. By default, run the module from which this function is called. argv : list of strings Arguments to be passed to the nose test runner. ``argv[0]`` is ignored. All command line arguments accepted by ``nosetests`` will work. If it is the default value None, sys.argv is used. .. versionadded:: 1.9.0 Examples -------- Adding the following:: if __name__ == "__main__" : run_module_suite(argv=sys.argv) at the end of a test module will run the tests when that module is called in the python interpreter. Alternatively, calling:: >>> run_module_suite(file_to_run="numpy/tests/test_matlib.py") # doctest: +SKIP from an interpreter will run all the test routine in 'test_matlib.py'. Nr__file__)KnownFailurePluginargv addplugins) sys _getframef_localsgetAssertionErrorr"r noseclassesr run) file_to_runr"fnoser s rrr=sF M!  jnnZ66    |x;-'{m# ==D//////HH$$6$6$8$8#9H:::::rcReZdZdZ ddZdZdZd Z dd Z dd Z ddZ dS)ra Nose test runner. This class is made available as numpy.testing.Tester, and a test function is typically added to a package's __init__.py like so:: from numpy.testing import Tester test = Tester().test Calling this test function finds and runs all tests associated with the package and all its sub-packages. Attributes ---------- package_path : str Full path to the package to test. package_name : str Name of the package to test. Parameters ---------- package : module, str or None, optional The package to test. If a string, this should be the full path to the package. If None (default), `package` is set to the module from which `NoseTester` is initialized. raise_warnings : None, str or sequence of warnings, optional This specifies which warnings to configure as 'raise' instead of being shown once during the test execution. Valid strings are: - "develop" : equals ``(Warning,)`` - "release" : equals ``()``, don't raise on any warnings. Default is "release". depth : int, optional If `package` is None, then this can be used to initialize from the module of the caller of (the caller of (...)) the code that initializes `NoseTester`. Default of 0 means the module of the immediate caller; higher values are useful for utility routines that want to initialize `NoseTester` objects on behalf of other code. NreleaserFc*|d}d}|vtjd|z}|jdd}|tt j|}|jdd}ngt|tt r6t j|j }t|dd}nt|}||_ |t|}||_||_||_dS)Nr/rr__name__)r$r%r&r'r(rrdirname isinstancetypergetattrstr package_pathr package_nameraise_warningscheck_fpu_mode)selfpackager9depthr:r8r,r7s r__init__zNoseTester.__init__s  !&N ? a%i((A:>>*d;;L#$$7??<88L:>>*d;;LL b * * (7??7+;<K 6HHH// .u../// d' ' ' LLL 65#466 7 7 7GOOEN33E 2522 3 3 3K''b11  + ++,,, %(<<=====rc"ddlm}|S)z Return instantiated plugin for doctests Allows subclassing of this class to override doctester A return value of None means use the nose builtin doctest plugin r) NumpyDoctest)r)r[)r;r[s r_get_custom_doctesterz NoseTester._get_custom_doctesters" .-----|~~rrArct||||}|r|d|jdddgz }|r5|dur|dgz }n*t|tr|ddt |gz }dd l}dd lm} d d l m } m } m } | g} | d |j jj Dz } |jr| | gz } |dgz } | }|| d|j Dz } n#t"$rYnwxYwd|v}|dkr|rd}|}| |r|s|dgz }n7|r|d| | d|gz } |r|d|jzgz }|| fS)z Run tests for module using nose. This method does the heavy lifting for the `test` method. It takes all the same arguments, for details see `test`. See Also -------- test z--cover-package=z--with-coveragez --cover-testsz --cover-eraseTz --with-timerz --timer-top-nrN)EntryPointPluginManagerr)r UnpluggerFPUModeCheckPluginc"g|] }| Srb.0ps r z0NoseTester.prepare_test_args..(s>>>AAACC>>>rz--with-fpumodecheckplugincg|]}|Srbrbrcs rrfz0NoseTester.prepare_test_args..0s>>>a>>>rz--with-doctestFdoctestz--with-)rrFr8r3intr6nose.plugins.builtin nose.pluginsr^r)r r_r`pluginsbuiltinr: loadPlugins ImportErrorr\removename)r;rCrDrEdoctestscoveragetimerr"r-r^r r_r`rlentrypoint_manager doctest_argvplugs rprepare_test_argszNoseTester.prepare_test_argssd  ugz::  5 ;(9;;=N"O5 5D  F}}((E3'' F#e**EE $###888888 6 6 6 6 6 6 6 6 6 6%%''(>>!5!=>>>>   2 **,,- -G 01 1D !8!8!:!:   * * , , , >>#5#=>>> >GG    D  (4/ u   H))++ < +  +)** . ,---  ),,d3 3G 0TY.//W}s 2C<< D D ct|d}ddlm}||_|||||||\} } |rt d|jnt d|j|ddl} d| _ ||j }ttfd } t|tr| |}td 5} t!jt!jd |D]}t!jd | | d| d| d| t(j | d| dt!j5t!jd ddlm}dddn #1swxYwY| t4|t!jddt6dddlm}|| d| }dddn #1swxYwY|jS)a Run tests for module using nose. Parameters ---------- label : {'fast', 'full', '', attribute identifier}, optional Identifies the tests to run. This can be a string to pass to the nosetests executable with the '-A' option, or one of several special values. Special values are: * 'fast' - the default - which corresponds to the ``nosetests -A`` option of 'not slow'. * 'full' - fast (as above) and slow tests as in the 'no -A' option to nosetests - this is the same as ''. * None or '' - run all tests. * attribute_identifier - string passed directly to nosetests as '-A'. verbose : int, optional Verbosity value for test outputs, in the range 1-10. Default is 1. extra_argv : list, optional List with any extra arguments to pass to nosetests. doctests : bool, optional If True, run doctests in module. Default is False. coverage : bool, optional If True, report coverage of NumPy code. Default is False. (This requires the `coverage module `_). raise_warnings : None, str or sequence of warnings, optional This specifies which warnings to configure as 'raise' instead of being shown once during the test execution. Valid strings are: * "develop" : equals ``(Warning,)`` * "release" : equals ``()``, do not raise on any warnings. timer : bool or int, optional Timing of individual tests with ``nose-timer`` (which needs to be installed). If True, time tests and report on all of them. If an integer (say ``N``), report timing results for ``N`` slowest tests. Returns ------- result : object Returns the result of running the tests as a ``nose.result.TextTestResult`` object. Notes ----- Each NumPy module exposes `test` in its namespace to run all tests for it. For example, to run all tests for numpy.lib: >>> np.lib.test() #doctest: +SKIP Examples -------- >>> result = np.lib.test() #doctest: +SKIP Running unit tests for numpy.lib ... Ran 976 tests in 3.933s OK >>> result.errors #doctest: +SKIP [] >>> result.knownfail #doctest: +SKIP [] r)utilsz$Running unit tests and doctests for zRunning unit tests for rNrb)developr/locationalwayserror)categoryzNot importing directory)messageznumpy.dtype size changedznumpy.ufunc size changedz.*boolean negative.*z.*boolean subtract.*)cpuinfo)rmoduleignorez.*getargspec.*znose\.)rrr)NumpyTestProgramF)r"exitrl)minrLr{rDrxrMr8rYrhmasterr9dictWarningr3r6rwarnings resetwarningsfilterwarningsfilternpModuleDeprecationWarningcatch_warnings simplefilter distutilsr UserWarningDeprecationWarningr)rresult)r;rCrDrErrrsr9rtr{r"rlrh _warn_optssup warningtyperrts rtestzNoseTester.testFs<Ngq// ..w HhGG g  A L9JLL M M M M ?D,=?? @ @ @       !!0N7*"$&&& nc * * 8'7N z * *% Ic  " $ $ $  #H - - -- G G '+FFFFF JJ8J 9 9 9 JJ9J : : : JJ9J : : : JJ ;J < < < JJ5J 6 6 6 JJ5J 6 6 6(** 1 1%h///000000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 JJ GJ < < <  #H6F-?+4 6 6 6 6 6 5 5 5 5 5  dHHHAK% I% I% I% I% I% I% I% I% I% I% I% I% I% I% INxs8C$I;G" I"G& &I)G& *AII  I ctd|j|||||}|ddtjzgz }t }ddlm}|dg}| ||S)a Run benchmarks for module using nose. Parameters ---------- label : {'fast', 'full', '', attribute identifier}, optional Identifies the benchmarks to run. This can be a string to pass to the nosetests executable with the '-A' option, or one of several special values. Special values are: * 'fast' - the default - which corresponds to the ``nosetests -A`` option of 'not slow'. * 'full' - fast (as above) and slow benchmarks as in the 'no -A' option to nosetests - this is the same as ''. * None or '' - run all tests. * attribute_identifier - string passed directly to nosetests as '-A'. verbose : int, optional Verbosity value for benchmark outputs, in the range 1-10. Default is 1. extra_argv : list, optional List with any extra arguments to pass to nosetests. Returns ------- success : bool Returns True if running the benchmarks works, False if an error occurred. Notes ----- Benchmarks are like tests, but have names starting with "bench" instead of "test", and can be found under the "benchmarks" sub-directory of the module. Each NumPy module exposes `bench` in its namespace to run all benchmarks for it. Examples -------- >>> success = np.lib.bench() #doctest: +SKIP Running benchmarks for numpy.lib ... using 562341 items: unique: 0.11 unique1d: 0.11 ratio: 1.0 nUnique: 56230 == 56230 ... OK >>> success #doctest: +SKIP True zRunning benchmarks for z--matchz(?:^|[\\b_\\.%s-])[Bb]enchr)r_rhr!) rMr8rYrFrseprr)r_r*)r;rCrDrEr"r-r_ add_pluginss rbenchzNoseTester.benchst ;(9;;<<<    ugz:: 9BFBCC}} +***** y++, xxTkx:::r)Nr/rF)rArNFFF)rArNFFNF)rArN) r1 __module__ __qualname____doc__r>rFrYr\rxrrrbrrrrps((RFG %'-'-'-'-R###J>>>(EI@E>>>>@8<<@IIIIVG;G;G;G;G;G;rrcvttdrdtjvrd}nd}t|ddS)NrNz.dev0r|r/rT)r9r=r:)hasattrrrNr)modes rr r sJr=!!g&?&? T%) + + ++r)NN)rrr$rrrr{rr__all__rrrr rbrrrs  11111111    &&&R0;0;0;0;fh;h;h;h;h;h;h;h;V +++++r