U
|
¸ý°d ã @ sÊ d dl m Z ddlmZ ddlmZ G dd de Zd ddZd d
|
Zd d Zd d Z ee Z dd Z
|
dd Z ee Z dd Z ee Z dd ZeeZdd ZeeZdd ZeeZG dd d eZd S ) é )Úgreenleté )ÚTestCase)Úfails_leakcheckc @ s< e Zd ZdZdd Zdd Zdd Zdd Zd
|
d ZeZ dS ) ÚgenletNc O s || _ || _d | _d S ©N)ÚargsÚkwdsÚchild)Úselfr r © r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\greenlet/tests/test_generator_nested.pyÚ__init__ s zgenlet.__init__c C s | j \}|| j| j d S r )Úfnr r )r r r r r Úrun s z
|
genlet.runc C s | S r r ©r r r r Ú__iter__ s zgenlet.__iter__c C s
|
|| _ d S r )r
|
)r r
|
r r r Ú set_child s zgenlet.set_childc C sP | j r.| j }|j r$|}|j }d |_ q | ¡ }nt ¡ | _| ¡ }| rH|S td S r )r
|
Úswitchr Ú
|
getcurrentÚparentÚ StopIteration)r r
|
ÚtmpÚresultr r r Ú__next__ s
|
|
zgenlet.__next__)
|
Ú__name__Ú
|
__module__Ú __qualname__r r r r r r Únextr r r r r s r c C sT t ¡ }|dkrFt|ts"td|dkr6|j |¡ |j}|d8 }q| | ¡ d S )Nr zyield outside a genletr )r r Ú
|
isinstancer Ú RuntimeErrorr r r )ÚvalueÚlevelÚgr r r ÚYield- s
|
|
r$ c s G fdddt }|S )Nc s e Zd Z fZdS )zGenlet.<locals>.TheGenletN)r r r r r ©Úfuncr r Ú TheGenlet< s r' )r )r& r' r r% r ÚGenlet; s r( c c s& t | D ]}| |d ¡ |V qd S ©Nr )ÚrangeÚappend©ÚnÚseenÚir r r Úg1C s r0 c C s( t | D ]}| |d ¡ t| qd S r) )r* r+ r$ r, r r r Úg2I s r1 c C s t | d S r ©r$ )r/ r r r ÚnestedQ s r3 c C s( t | D ]}| |d ¡ t| qd S r) )r* r+ r3 r, r r r Úg3U s r4 c C s2 | dkr d S t | d D ] }t| qt| d S )Nr r )Úaxr$ ©r- Úiir r r Úa\ s
|
|
r8 c sP t | dkrD| D ]0 fddt fdd| D D }|stqnt| d S )Nr c s g | ]}t g| qS r r2 )Ú.0Úp©Úer r Ú
|
<listcomp>i s zperms.<locals>.<listcomp>c s g | ]}| kr|qS r r )r9 Úxr; r r r= i s )ÚlenÚpermsÚAssertionErrorr$ )Úlr> r r; r r@ e s
|
$ r@ c C s* t d| D ]}t| t|| d q
|
d S )Nr é )r* r$ r6 r r r Úgr1p s rD c C s t | D ]}| |¡ qd S r )rD r+ )r- r. r7 r r r Úgr2x s rE c @ s8 e Zd Zdd Zedd Zdd Zdd Zd d
|
Zd S ) ÚNestedGeneratorTestsc C s> g }t d|D ]}| |¡ q| |ddddddddg¡ d S )Né r rC é é é é )rE r+ Ú assertEqual©r r. r7 r r r Útest_layered_genlets s z)NestedGeneratorTests.test_layered_genletsc C s t ttd}t|}| t|d¡ | ddddg|¡ | ddddg|¡ g }tt ttdt ttdD ]}| |¡ qv| |ddddgdddgfddddgdddgfddddgdddgfddddgdddgfddddgdddgfddddgdddgfg¡ d S )NrH é r r rC rI )r@ Úlistr* rL r? ZassertInÚzipr+ )r Z gen_permsÚ permutationsÚresr7 r r r Útest_permutations s" & ( þþz&NestedGeneratorTests.test_permutationsc C sd t ttfD ]T}g }tdD ] }|d|D ]}| |¡ q(q| |dddddddddddg
|
¡ q
|
d S )NrI rG r r rC rH )r0 r1 r4 r* r+ rL )r r# r. Ú_Újr r r Útest_genlet_simple s z'NestedGeneratorTests.test_genlet_simplec C s& z t d W n tk
|
r Y nX d S )Né
|
)r$ r r r r r Útest_genlet_bad s z$NestedGeneratorTests.test_genlet_badc C s g }t dD ]}| |¡ q d S )NrG )r5 r+ rM r r r Útest_nested_genlets¥ s z(NestedGeneratorTests.test_nested_genletsN) r r r rN r rT rW rY rZ r r r r rF s
|
rF N)r )r Ú r Z leakcheckr r r$ r( r0 r1 r3 r4 r8 r5 r@ rD rE rF r r r r Ú<module> s( '
|
|
