
    J/Ph	                         d dl mZ d dlZd dlZd dlZd dlmZ  G d deej                  Z	e
dk    r ej                     dS dS )    )jitN)MemoryLeakMixinc                   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                     ddl m} t          j        d          gt          j        dt          j                  gg}|                     ||           d S )Nr   )chain1      )dtype)#numba.tests.chained_assign_usecasesr   nparangedouble_test_template)selfr   argss      _/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numba/tests/test_chained_assign.pytest_chain1zTestChainedAssign.test_chain1	   s]    >>>>>>Yq\\NYq	***+
 	FD)))))    c                 J    ddl m} dgdgg}|                     ||           d S )Nr   )chain2         @)r   r   r   )r   r   r   s      r   test_chain2zTestChainedAssign.test_chain2   sC    >>>>>>CE
 	FD)))))r   c                 N    ddl m} ddgddgg}|                     ||           d S )Nr   )unpack1   r         ?r   )r   r   r   )r   r   r   s      r   test_unpack1zTestChainedAssign.test_unpack1   sG    ??????H!H
 	GT*****r   c                     ddl m} t          j        dg          t          j        dg          gt          j        dg          t          j        dg          gg}|                     ||           d S )Nr   )unpack2r	   g      @       @r
   )r   r!   r   arrayr   )r   r!   r   s      r   test_unpack2zTestChainedAssign.test_unpack2!   sq    ??????Xqc]]BHcUOO,Xse__bhsmm,
 	GT*****r   c                     ddl m} t          j        dg          t          j        dg          gt          j        dg          t          j        dg          gg}|                     ||           d S )Nr   )chain3g      ?g      ?r   )r   r&   r   r#   r   )r   r&   r   s      r   test_chain3zTestChainedAssign.test_chain3)   sq    >>>>>>Xqc]]BHcUOO,Xse__bhsmm,
 	FD)))))r   c                     ddl m} t          j        dg          gt          j        dg          gg}|                     ||           d S )Nr   )unpack3r   r   )r   r)   r   r#   r   )r   r)   r   s      r   test_unpack3zTestChainedAssign.test_unpack31   X    ??????Xqc]]OXse__
 	GT*****r   c                     ddl m} t          j        dg          gt          j        dg          gg}|                     ||           d S )Nr   )unpack4r   r   )r   r-   r   r#   r   )r   r-   r   s      r   test_unpack4zTestChainedAssign.test_unpack49   r+   r   c                     ddl m} t          j        dg          gt          j        dg          gg}|                     ||           d S )Nr   )unpack5r	   r"   )r   r0   r   r#   r   )r   r0   r   s      r   test_unpack5zTestChainedAssign.test_unpack5A   r+   r   c                 J    ddl m} d}d}|                     |||g           d S )Nr   )unpack6)r   r	   )r   r"   )r   r3   r   )r   r3   args1args2s       r   test_unpack6zTestChainedAssign.test_unpack6I   s>    ??????GeU^44444r   c                     t          |          }|D ]P}t          j        |          }t          j        |          }t          j                             ||  ||            Qd S )N)r   copydeepcopyr   testingassert_allclose)r   pyfuncargcasescfuncr   a1a2s          r   r   z TestChainedAssign._test_templateO   so    F 	@ 	@Dt$$Bt$$BJ&&vvr{EE2J????	@ 	@r   N)__name__
__module____qualname__r   r   r   r$   r'   r*   r.   r1   r6   r    r   r   r   r      s        * * ** * *+ + ++ + +* * *+ + ++ + ++ + +5 5 5@ @ @ @ @r   r   __main__)numbar   unittestnumpyr   r8   numba.tests.supportr   TestCaser   rA   mainrD   r   r   <module>rL      s                / / / / / /L@ L@ L@ L@ L@): L@ L@ L@^ zHMOOOOO r   