§
    ƒÛg]  ã                   ó  — d dl mZ d dlZd dlZd dlZd dlmZ ed„ ¦   «         Zed„ ¦   «         Z	ed„ ¦   «         Z
d„ Zd„ Zd	„ Zd
„ Zd„ Zd„ Zd„ Zd„ Zd„ Z G d„ deej        ¦  «        Zedk    r ej        ¦   «          dS dS )é    )ÚjitN)ÚMemoryLeakMixinc                 ód   — t          t          | ¦  «        ¦  «        D ]}| |xx         dz  cc<   Œ| S )Né   )ÚrangeÚlen)ÚaÚis     úU/granary/edev/quantum/lib/python3.11/site-packages/numba/tests/test_chained_assign.pyÚincr      s9   € å•3�q‘6”6‰]Œ]ð ð ˆØ	ˆ!ˆˆŒ�‰	ˆˆ‰ˆØ€Hó    c                 ó2   — | dxx         dz  cc<   | d         S ©Nr   r   © ©r	   s    r   Úinc1r      s    € à€a€D€D„DˆA�I€D€D�DØˆQŒ4€Kr   c                 óH   — | dxx         dz  cc<   | d         | d         dz   fS r   r   r   s    r   Úinc2r      s-   € à€a€D€D„DˆA�I€D€D�DØˆQŒ4��1”˜‘ˆ>Ðr   c                 ó8   — t          | ¦  «        x}x}}||z   |z   S ©N)r   )r	   ÚxÚyÚzs       r   Úchain1r      s%   € Ý�A‘”Ð€AÐˆˆAØˆq‰5�1‰9Ðr   c                 óp   — t          j        d¦  «        }| x|d<   x}|d<   |d         |d         z   |dz  z   S )Né   r   r   )ÚnpÚzeros)Úvr	   r   s      r   Úchain2r       sA   € Ý
Œ�‰Œ€AØÐ€A€a�DÐˆ1ˆq�‰tØˆQŒ4�!�A”$‰;˜!˜a™%Ñ Ð r   c                 ó   — | |}}||dz  z   S ©Nr   r   )r   r   r	   Úbs       r   Úunpack1r$   $   s   € Øˆa€q€AØˆq�1‰u‰9Ðr   c                 ón   — t          | ¦  «        t          |¦  «        fx\  }}\  }}||dz  z   ||dz  z   fS r"   ©r   ©r   r   r	   r#   ÚcÚds         r   Úunpack2r*   )   s@   € Ý�q‘'”'�4 ™7œ7Ð"Ð"�D€A€q‰4ˆ1ˆaØˆq�1‰u‰9�a˜!˜a™%‘iÐÐr   c                 óö   — t          | ¦  «        t          |¦  «        fx}\  }}t          | ¦  «        t          |¦  «        fx\  }}}|d         |dz  z   |z   |d         z   |d         |z   |dz  z   |d         z   fS ©Nr   r   r   r&   )r   r   r	   r#   r(   r)   ÚeÚfs           r   Úchain3r/   .   s   € Ý�q‘'”'�4 ™7œ7Ð#Ð#€A‰ˆˆAÝ�q‘'”'�4 ™7œ7Ð#Ð#�F€QˆˆQØˆaŒD�1�q‘5‰L˜1Ñ˜q œtÑ# q¨¤t¨a¡x°!°a±%Ñ'7¸!¸A¼$Ñ'>Ð?Ð?r   c                 ó6   — t          | ¦  «        \  }}||dz  z   S r"   ©r   )r   r	   r#   s      r   Úunpack3r2   4   s   € Ý�‰7Œ7�D€A€qØˆq�1‰u‰9Ðr   c                 óP   — t          | ¦  «        x\  }}\  }}||dz  z   ||dz  z   fS r"   r1   )r   r	   r#   r(   r)   s        r   Úunpack4r4   9   s5   € Ý�q‘'”'Ð�D€A€q‰4ˆ1ˆaØˆq�1‰u‰9�a˜!˜a™%‘iÐÐr   c                 óº   — t          | ¦  «        x}\  }}t          | ¦  «        x\  }}}|d         |dz  z   |z   |d         z   |d         |z   |dz  z   |d         z   fS r,   r1   )r   r	   r#   r(   r)   r-   r.   s          r   Úunpack5r6   >   si   € Ý�A‰wŒwÐ€A‰ˆˆ1Ý�A‰wŒwÐ�D€A€qˆ1ØˆaŒD�1�q‘5‰L˜1Ñ˜q œtÑ# q¨¤t¨a¡x°!°a±%Ñ'7¸!¸A¼$Ñ'>Ð?Ð?r   c                 óL   — | |f|dz   | dz   fc\  }}\  }}||dz  z   |dz  |z   fS )Nr   r   r   r'   s         r   Úunpack6r8   D   s@   € Ø˜�V˜a !™e Q¨¡U˜^€N�F€Qˆ‰FˆQ�Øˆq�1‰u‰9�a˜!‘e˜a‘iÐÐr   c                   óJ   — e Zd Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Z	d„ Z
d	„ Zd
„ ZdS )ÚTestChainedAssignc                 ó¨   — t          j        d¦  «        gt          j        dt           j        ¬¦  «        gg}|                      t          |¦  «         d S )Nr   é   )Údtype)r   ÚarangeÚdoubleÚ_test_templater   ©ÚselfÚargss     r   Útest_chain1zTestChainedAssign.test_chain1J   sK   € åŒY�q‰\Œ\ˆNÝŒY�q¥¤	Ð*Ñ*Ô*Ð+ð
ˆð 	×Ò�F DÑ)Ô)Ð)Ð)Ð)r   c                 óH   — dgdgg}|                       t          |¦  «         d S )Né   ç      @)r@   r    rA   s     r   Útest_chain2zTestChainedAssign.test_chain2Q   s1   € àˆCØˆEð
ˆð 	×Ò�F DÑ)Ô)Ð)Ð)Ð)r   c                 óL   — ddgddgg}|                       t          |¦  «         d S )Nr   rG   ç      ð?rF   )r@   r$   rA   s     r   Útest_unpack1zTestChainedAssign.test_unpack1X   s5   € à�ˆHØ�!ˆHð
ˆð 	×Ò�G TÑ*Ô*Ð*Ð*Ð*r   c                 óä   — t          j        dg¦  «        t          j        dg¦  «        gt          j        dg¦  «        t          j        dg¦  «        gg}|                      t          |¦  «         d S )Nr   g      @ç       @r<   )r   Úarrayr@   r*   rA   s     r   Útest_unpack2zTestChainedAssign.test_unpack2_   s_   € åŒX�q�c‰]Œ]�BœH c U™OœOÐ,ÝŒX�s�e‰_Œ_�bœh¨ s™mœmÐ,ð
ˆð 	×Ò�G TÑ*Ô*Ð*Ð*Ð*r   c                 óä   — t          j        dg¦  «        t          j        dg¦  «        gt          j        dg¦  «        t          j        dg¦  «        gg}|                      t          |¦  «         d S )Nr   g      ø?g      à?r   )r   rN   r@   r/   rA   s     r   Útest_chain3zTestChainedAssign.test_chain3f   s_   € åŒX�q�c‰]Œ]�BœH c U™OœOÐ,ÝŒX�s�e‰_Œ_�bœh¨ s™mœmÐ,ð
ˆð 	×Ò�F DÑ)Ô)Ð)Ð)Ð)r   c                 ó”   — t          j        dg¦  «        gt          j        dg¦  «        gg}|                      t          |¦  «         d S ©Nr   rJ   )r   rN   r@   r2   rA   s     r   Útest_unpack3zTestChainedAssign.test_unpack3m   óF   € åŒX�q�c‰]Œ]ˆOÝŒX�s�e‰_Œ_Ðð
ˆð 	×Ò�G TÑ*Ô*Ð*Ð*Ð*r   c                 ó”   — t          j        dg¦  «        gt          j        dg¦  «        gg}|                      t          |¦  «         d S rS   )r   rN   r@   r4   rA   s     r   Útest_unpack4zTestChainedAssign.test_unpack4t   rU   r   c                 ó”   — t          j        dg¦  «        gt          j        dg¦  «        gg}|                      t          |¦  «         d S )Nr   rM   )r   rN   r@   r6   rA   s     r   Útest_unpack5zTestChainedAssign.test_unpack5{   rU   r   c                 óH   — d}d}|                       t          ||g¦  «         d S )N)rG   r   )rG   rM   )r@   r8   )rB   Úargs1Úargs2s      r   Útest_unpack6zTestChainedAssign.test_unpack6‚   s,   € ØˆØˆØ×Ò�G e¨U ^Ñ4Ô4Ð4Ð4Ð4r   c                 óÊ   — t          |¦  «        }|D ]P}t          j        |¦  «        }t          j        |¦  «        }t          j                              ||Ž  ||Ž ¦  «         ŒQd S r   )r   ÚcopyÚdeepcopyr   ÚtestingÚassert_allclose)rB   ÚpyfuncÚargcasesÚcfuncrC   Úa1Úa2s          r   r@   z TestChainedAssign._test_template‡   so   € Ý�F‘”ˆØð 	@ð 	@ˆDÝ”˜tÑ$Ô$ˆBÝ”˜tÑ$Ô$ˆBÝŒJ×&Ò& v v¨r {°E°E¸2°JÑ?Ô?Ð?Ð?ð	@ð 	@r   N)Ú__name__Ú
__module__Ú__qualname__rD   rH   rK   rO   rQ   rT   rW   rY   r]   r@   r   r   r   r:   r:   I   s¯   € € € € € ð*ð *ð *ð*ð *ð *ð+ð +ð +ð+ð +ð +ð*ð *ð *ð+ð +ð +ð+ð +ð +ð+ð +ð +ð5ð 5ð 5ð
@ð @ð @ð @ð @r   r:   Ú__main__)Únumbar   ÚunittestÚnumpyr   r_   Únumba.tests.supportr   r   r   r   r   r    r$   r*   r/   r2   r4   r6   r8   ÚTestCaser:   rh   Úmainr   r   r   ú<module>rr      s‹  ðØ Ð Ð Ð Ð Ð Ø €€€Ø Ð Ð Ð Ø €€€Ø /Ð /Ð /Ð /Ð /Ð /ð ðð ñ „ðð
 ðð ñ „ðð ðð ñ „ðð
ð ð ð
!ð !ð !ðð ð ð
 ð  ð  ð
@ð @ð @ðð ð ð
 ð  ð  ð
@ð @ð @ð ð  ð  ð
C@ð C@ð C@ð C@ð C@˜¨Ô):ñ C@ô C@ð C@ðL ˆzÒÐØ€H„M�O„O€O€O€Oð Ðr   