OwlCyberSecurity - MANAGER
Edit File: test_generator_nested.cpython-312.pyc
� [��g� � �� � d dl m Z ddlmZ ddlmZ G d� de � Zdd�Zd� Zd � Zd � Z ee � Z d � Z d � Z ee � Z d � Z ee � Z d� Z ee� Zd� Z ee� Zd� Z ee� Z G d� de� Zy)� )�greenlet� )�TestCase)�fails_leakcheckc �2 � e Zd ZdZd� Zd� Zd� Zd� Zd� ZeZ y)�genletNc �. � || _ || _ d | _ y �N)�args�kwds�child)�selfr r s �U/opt/hc_python/lib64/python3.12/site-packages/greenlet/tests/test_generator_nested.py�__init__zgenlet.__init__ s � ��� ��� ��� � c �X � | j \ } || j i | j �� y r )�fnr r )r r s r �runz genlet.run s$ � � �g�g��� �D�I�I�#����#r c � � | S r � �r s r �__iter__zgenlet.__iter__ s � �� r c � � || _ y r )r )r r s r � set_childzgenlet.set_child s � ��� r c � � | j rK| j }|j r"|}|j }d |_ |j r�"|j � }n)t j � | _ | j � }| r|S t �r )r �switchr � getcurrent�parent� StopIteration)r r �tmp�results r �__next__zgenlet.__next__ sm � � �:�:��J�J�E��+�+��� � � �� �� � �+�+� �\�\�^�F�"�-�-�/�D�K��[�[�]�F� ��M��r ) �__name__� __module__� __qualname__r r r r r r" �nextr r r r r s' � � �F�� $� �� �$ �Dr r c � � t j � }|dk7 rRt |t � s t d� �|dkD r|j j |� |j }|dz }|dk7 r�R|j | � y )Nr zyield outside a genletr )r r � isinstancer � RuntimeErrorr r r )�value�level�gs r �Yieldr- - sn � ���� �A� �1�*��!�V�$� � 7�8� 8� �1�9� �H�H� � �q� !� �H�H�� �� �� �1�*� �H�H�U�Or c �* � � G � fd�dt � }|S )Nc � �� e Zd ZW � fZy)�Genlet.<locals>.TheGenletN)r# r$ r% r )�funcs �r � TheGenletr0 < s �� ��W�r r2 )r )r1 r2 s ` r �Genletr3 ; s �� ��F� � �r c # �\ K � t | � D ] }|j |dz � |�� � y �w�Nr )�range�append��n�seen�is r �g1r< C s) � �� � �1�X�� � � �A��E���� �s �*,c �b � t | � D ]! }|j |dz � t |� �# y r5 )r6 r7 r- r8 s r �g2r> I s' � � �1�X�� � � �A��E�� �a�� r c � � t | � y r )r- )r; s r �nestedr@ Q s � � �!�Hr c �b � t | � D ]! }|j |dz � t |� �# y r5 )r6 r7 r@ r8 s r �g3rB U s' � � �1�X�� � � �A��E���q� � r c �b � | dk( ry t | dz � D ] }t |� � t | � y )Nr r )�axr- �r9 �iis r �arG \ s, � ��A�v����Q��i�� �b� � � �!�Hr c �� � t | � dkD rE| D ]? }t | D �cg c] }||k7 s� |�� c}� D �cg c] }t |g|z � �� }}|r�?J � y t | � y c c}w c c}w r5 )�len�permsr- )�l�e�x�ps r rJ rJ e sj � � �1�v��z��A�).�1�/G�1�a��Q���1�/G�)H�I�)H�A���s�Q�w� �)H�A�I��H�1� � �a��� 0H��Is � A �A �A%c �Z � t d| � D ] }t |� t ||z d� � y )Nr � )r6 r- rE s r �gr1rQ p s' � ��A�q�k�� �b� � �b�2�g�q�� r c �F � t | � D ] }|j |� � y r )rQ r7 )r9 r: rF s r �gr2rS x s � ��!�f�� � � �B�� r c �4 � e Zd Zd� Zed� � Zd� Zd� Zd� Zy)�NestedGeneratorTestsc �t � g }t d|� D ] }|j |� � | j |g d�� y )N� )r r rP � � � rX � )rS r7 � assertEqual�r r: rF s r �test_layered_genletsz)NestedGeneratorTests.test_layered_genlets� s3 � ����a��,�B� �K�K��O� � ���� 8�9r c � � t t t d� � � }t |� }| j t |� d� | j g d�|� | j g d�|� g }t t t t d� � � t t t d� � � � D ] }|j |� � | j |g d�g d�fg d�g d�fg d �g d �fg d �g d �fg d �g d�fg d�g d�fg� y )NrX � )r r rP rY )rY rP r r rY )r r rP )r r rY rP )r rP r )r rP r rY )r r rP )r rP rY r )r rP r )r rY r rP )rP r r )r rY rP r )rP r r )rJ �listr6 r\ rI �assertIn�zipr7 )r � gen_perms� permutations�resrF s r �test_permutationsz&NestedGeneratorTests.test_permutations� s� � ��$�u�Q�x�.�)� ��I�� � ����\�*�M�:� � � �l�L�1� � � �l�L�1����e�D��q��N�+�U�4��a��>�-B�C�B� �J�J�r�N� D� ��� ��I� &� �y�(A��I� &� �y�(A��I� &� �y�(A� C� Dr c � � t t t fD ]J }g }t d� D ]! } |d|� D ] }|j |� � �# | j |dg d�z � �L y )NrY rW ) r r rP r rY rP rX rY rW rX )r< r> rB r6 r7 r\ )r r, r: �_�js r �test_genlet_simplez'NestedGeneratorTests.test_genlet_simple� sX � ��R���A��D��1�X���1�d��A��K�K��N� $� � � � �T�1�'E�#E� F� r c �: � t d� y # t $ r Y y w xY w)N� )r- r) r s r �test_genlet_badz$NestedGeneratorTests.test_genlet_bad� s � � � �"�I��� � � �s � � �c �J � g }t d� D ] }|j |� � y )NrW )rD r7 r] s r �test_nested_genletsz(NestedGeneratorTests.test_nested_genlets� s � ����Q�%�B� �K�K��O� r N) r# r$ r% r^ r rg rk rn rp r r r rU rU s- � �:� � D� � D� G�� r rU N)r )r � r � leakcheckr r r- r3 r< r> r@ rB rG rD rJ rQ rS rU r r r �<module>rs s� �� � � &�%�X� %�N � �� � �B�Z�� �� �B�Z�� � �A�Y��� �u� ��� �S�k��� �S�k��) �8� ) r