관리-도구
편집 파일: test_generator_nested.cpython-38.pyc
U �g� � @ s� d dl m Z ddlmZ ddlmZ G dd� de �Zd dd�Zd d � Zd d � Zd d� Z ee �Z dd� Z dd� Z ee �Z dd� Z ee �Z dd� Zee�Zdd� Zee�Zdd� Zee�ZG dd � d e�Zd 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 �T/opt/hc_python/lib64/python3.8/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 t�d 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|t�s"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 � fdd�dt �}|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� � fdd�t� fdd�| D ��D �}|st�qnt| � 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 ttd���t ttd����D ]}|�|� 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 }td�D ] }|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 d�D ]}|�|� 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( '