Why Gemfury?
Push, build, and install
RubyGems
npm packages
Python packages
Maven artifacts
PHP packages
Go Modules
Debian packages
RPM packages
NuGet packages
duality-group
duality-group / ray python
Repository URL to install this package:
|
Version:
2.0.0rc1 ▾
|
o
õbÄ+ ã @ sÄ d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl Z zddl
Z
W n e y5 dZ
Y nw zddl Z W n e yG dZ Y nw ddl
Z
ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ d
ZdZe
¡ pdZ e oe Z ej ej !e"¡¡Z#e j$j%Z%d$ddZ&G dd dZ'G dd de j(Z)G dd de j*Z+G dd de+Z,d%dd Z-d d Z.d d Z/d!d" Z0e1d#kràe0 dS dS )&aA
Unit test runner, providing new features on top of unittest module:
- colourized output
- parallel run (UNIX only)
- print failures/tracebacks on CTRL+C
- re-run failed tests only (make test-failed)
Invocation examples:
- make test
- make test-failed
Parallel:
- make test-parallel
- make test-process ARGS=--parallel
é )Úprint_functionN)Úhilite)Ú print_color)Úterm_supports_colors)Úsuper)Ú
CI_TESTING)Úimport_module_by_path)Ú
print_sysinfo)Ú
reap_children)Ú safe_rmpathé z.failed-tests.txté Fc C sD |d u r|dkr t jnt j}trt| |||d d S t| |d d S )NÚred)ÚboldÚfile)r )ÚsysÚstderrÚstdoutÚ
USE_COLORSr Úprint)ÚmsgÚcolorr r © r ú:/tmp/pip-target-wec6zj2z/lib/python/psutil/tests/runner.pyÚcprintA s
r c @ sV e Zd ZeZdgZdejv re g d¢¡ dd Z dd Z
dd Z d
d Z d d
Z
dS )Ú
TestLoaderztest_memleaks.pyZ WHEELHOUSE_UPLOADER_USERNAME)z test_osx.pyz
test_linux.pyz
test_posix.pyc s fddt j¡D S )Nc s: g | ]}| d ¡r| d¡r| jvrtj j|¡qS )Ztest_ú.py)Ú
startswithÚendswithÚ
skip_filesÚosÚpathÚjoinÚtestdir)Ú.0Úx©Úselfr r Ú
<listcomp>R s þþýz,TestLoader._get_testmods.<locals>.<listcomp>)r Úlistdirr# r& r r& r Ú
_get_testmodsQ s
ÿzTestLoader._get_testmodsc c sP | ¡ D ] }t|}t|D ]}t||}t|tr$t|tjr$|V qqdS )zgIterate over all test files in this directory and return
all TestCase classes in them.
N) r* r ÚdirÚgetattrÚ
isinstanceÚtypeÚ
issubclassÚunittestÚTestCase)r' r! ÚmodÚnameÚobjr r r Ú_iter_testmod_classesW s
ÿüþz TestLoader._iter_testmod_classesc C s, t ¡ }| ¡ D ] }t|}| |¡ q|S ©N)r0 Ú TestSuiter5 ÚloadTestsFromTestCaseÚaddTest)r' Úsuiter4 Útestr r r Úallc s
zTestLoader.allc C sr t ¡ }tj t¡s |S ttd}| ¡ ¡ }W d n1 s"w Y |D ]
}t j
|¡}| |¡ q)|S )NÚrt) r0 r7 r r! ÚisfileÚFAILED_TESTS_FNAMEÚopenÚreadÚsplitÚdefaultTestLoaderÚloadTestsFromNamer9 )r' r: ÚfÚnamesÚnr; r r r Ú last_failedj s ÿ zTestLoader.last_failedc C s. | d¡rtj tj |¡¡d }tj |¡S )Nr r )r r r! ÚsplitextÚbasenamer0 rC rD )r' r3 r r r Ú from_namev s
zTestLoader.from_nameN)Ú__name__Ú
__module__Ú __qualname__ÚHEREr# r r ÚenvironÚextendr* r5 r< rH rK r r r r r J s
r c s< e Zd Zdd Zdd Zdd Zdd Z fd d
Z ZS ) ÚColouredResultc C s t j | |¡ tdd d S )NÚOKÚgreen)r0 Ú
TestResultÚ
addSuccessr )r' r; r r r rV ~ s zColouredResult.addSuccessc C s" t j | ||¡ tdddd d S )NÚERRORr T©r )r0 rU ÚaddErrorr ©r' r; Úerrr r r rY s zColouredResult.addErrorc C s t j | ||¡ tdd d S )NÚFAILr )r0 rU Ú
addFailurer rZ r r r r] s zColouredResult.addFailurec C s& t j | ||¡ td| ¡ d d S )Nz skipped: %sZbrown)r0 rU ÚaddSkipr Ústrip)r' r; Úreasonr r r r^ s zColouredResult.addSkipc s$ t |d|dkd}t ||¡ d S )Nr rW rX )r r ÚprintErrorList)r' ÚflavourÚerrors©Ú __class__r r ra s z ColouredResult.printErrorList) rL rM rN rV rY r] r^ ra Ú
__classcell__r r rd r rR | s rR c sf e Zd ZdZer
enejZ fddZ fddZ
dd Z dd Z fd
d Z
d d
Zdd Z ZS )ÚColouredTextRunnerz
A coloured text runner which also prints failed tests on KeyboardInterrupt
and save failed tests in a file so that they can be re-run.
c s t j|i |¤ t | _d S r6 )r Ú__init__ÚsetÚ
failed_tnames)r' ÚargsÚkwargsrd r r rh s zColouredTextRunner.__init__c s t ¡ | _| jS r6 )r Ú _makeResultÚresultr& rd r r rm s z ColouredTextRunner._makeResultc C sT | j r(ttd}| j D ] }| |d ¡ q W d d S 1 s!w Y d S d S )NÚwtÚ
)rj r@ r? Úwrite)r' rE Útnamer r r Ú_write_last_failed¤ s
ÿ"ÿÿz%ColouredTextRunner._write_last_failedc C s: | ¡ s|j|j D ]}|d ¡ }| j |¡ q
d S d S )Nr )Ú
wasSuccessfulrc ÚfailuresÚidrj Úadd)r' rn Útrr r r r Ú _save_resultª s ýz ColouredTextRunner._save_resultc sL zt |¡}W n ttfy | jj}| ¡ t d¡w | |¡ |S )Nr
)
r ÚrunÚKeyboardInterruptÚ
SystemExitÚrunnerrn Ú printErrorsr Úexitry ©r' r: rn rd r r Ú_run° s
ý
zColouredTextRunner._runc C sL |rt dddd tt t d¡ d S t dddd | ¡ t d¡ d S ) NÚSUCCESSrT TrX r ZFAILEDr r
)r r r? r r rs )r' Úsuccessr r r Ú_exit» s zColouredTextRunner._exitc C s | |¡}| | ¡ ¡ d S r6 )r r rt r r r r rz Å s
zColouredTextRunner.run)rL rM rN Ú__doc__r rR r0 ÚTextTestResultZ resultclassrh rm rs ry r r rz rf r r rd r rg s
rg c @ s, e Zd Zedd Zedd Zdd ZdS )ÚParallelRunnerc s0 fdd}t j |tj _t t¡}t | |¡S )Nc s | |}t |j¡ |S r6 )ÚatexitÚregisterÚclose)ÚfdÚmodeÚkwdsÚstream©Z orig_fdopenr r ÚfdopenÎ s
z+ParallelRunner._parallelize.<locals>.fdopen)r r ÚconcurrencytestZfork_for_testsÚNWORKERSZConcurrentTestSuite)r: r Zforkerr r r Ú _parallelizeÌ s
zParallelRunner._parallelizec C s t ¡ }t ¡ }| D ]6}| ¡ dkrq
t|t jr |jd j}nt|t jr)|}ntd| t|ddr;| |¡ q
| |¡ q
||fS )Nr zcan't recognize type %rZ
_serialrunF)
r0 r7 ÚcountTestCasesr- Z_testsre r1 Ú TypeErrorr, r9 )r: ÚserialÚparallelr; Z
test_classr r r Ú _split_suiteÚ s zParallelRunner._split_suitec C sV | |¡\}}| |¡}tdt ddd t ¡ }| |¡}t ¡ | }t ¡ ¡ }tj |dd\}} | r?td| d t
t ¡ }| |¡}
t ¡ | } | ¡ s\| ¡ d kr\|
¡ tt|j|j|jf\} }
}tt|
j|
j|
jf\}}}tt d
|j| |
|||
j|||| f
¡ td |j|
j || tf | ¡ o£|
¡ }| |¡ d S ) Nz(starting parallel tests using %s workersrT TrX r
)Útimeoutzalive processes %sr r a=
+----------+----------+----------+----------+----------+----------+
| | total | failures | errors | skipped | time |
+----------+----------+----------+----------+----------+----------+
| parallel | %3s | %3s | %3s | %3s | %.2fs |
+----------+----------+----------+----------+----------+----------+
| serial | %3s | %3s | %3s | %3s | %.2fs |
+----------+----------+----------+----------+----------+----------+
z&Ran %s tests in %.3fs using %s workers)r r r r Útimer ÚpsutilÚProcessÚchildrenZ
wait_procsr
rt r r~ ÚmapÚlenru rc Úskippedr ÚtextwrapÚdedentZtestsRunr )r' r: Z ser_suiteZ par_suiterx ÚparZ par_elapsedZorphansÚgoneÚaliveZserZ ser_elapsedZ par_failsZpar_errsZ par_skipsZ ser_failsZser_errsZ ser_skipsÚokr r r rz î sH
ÿ
þ þ ÿø
ÿzParallelRunner.runN)rL rM rN Ú staticmethodr r rz r r r r r Ê s
r c C sT dd }| r%t jr|d ntd u r|d ntdkr |d nttdS ttdS )Nc S s t | d d d S )Nz Running serial tests instead.r )r )r r r r Úwarn s zget_runner.<locals>.warnz$Can't run parallel tests on Windows.z(concurrencytest module is not installed.r
zOnly 1 CPU available.)Ú verbosity)r ÚWINDOWSr r r Ú VERBOSITYrg )r r¨ r r r Ú
get_runner s
r¬ c C s* t rt t | ¡}t }| |¡ d S r6 )r r r rK r¬ rz )r3 r: r} r r r Ú
run_from_name/ s
r c C s t d¡ d S )NT)r Z
_set_debugr r r r Úsetup7 s r® c C sÈ t d} tj| dd}|jddddd |jd ddd
d | ¡ \}}|js+tt t }|rGt |d kr?|
¡ t d ¡S |
|d ¡}n |jrO| ¡ }n| ¡ }trXt t|j}| |¡ d S )
Nz*python3 -m psutil.tests [opts] [test-name]zrun unit tests)ÚusageÚ descriptionz
--last-failedÚ
store_trueFzonly run last failed tests)ÚactionÚdefaultÚhelpz
--parallelzrun tests in parallelr
r )r® ÚoptparseÚ OptionParserÚ
add_optionÚ
parse_argsrH r r? r r Ú print_usager r rK r< r r r¬ r rz )r¯ ÚparserÚoptsrk Úloaderr: r} r r r Úmain; s6 þþ
r½ Ú__main__)FN)F)2r
Ú
__future__r r rµ r r r¡ r r0 ÚctypesÚ ImportErrorr r Zpsutil._commonr r r Zpsutil._compatr Z psutil.testsr r r r
r r« r? Ú cpu_countr r r! ÚabspathÚdirnameÚ__file__rO rC r8 r r r rR ZTextTestRunnerrg r r¬ r r® r½ rL r r r r Ú<module> s^ ÿ ÿ
27
U"
ÿ