
    xKg	                         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                          yy)    )jitN)MemoryLeakMixinc                   H    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zy)TestChainedAssignc                     ddl m} t        j                  d      gt        j                  dt        j                        gg}| j                  ||       y )Nr   )chain1      )dtype)#numba.tests.chained_assign_usecasesr   nparangedouble_test_template)selfr   argss      c/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/tests/test_chained_assign.pytest_chain1zTestChainedAssign.test_chain1	   s@    >YYq\NYYq		*+
 	FD)    c                 @    ddl m} dgdgg}| j                  ||       y )Nr   )chain2         @)r   r   r   )r   r   r   s      r   test_chain2zTestChainedAssign.test_chain2   s(    >CE
 	FD)r   c                 D    ddl m} ddgddgg}| j                  ||       y )Nr   )unpack1   r         ?r   )r   r   r   )r   r   r   s      r   test_unpack1zTestChainedAssign.test_unpack1   s,    ?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}| j	                  ||       y )Nr   )unpack2r	   g      @       @r
   )r   r!   r   arrayr   )r   r!   r   s      r   test_unpack2zTestChainedAssign.test_unpack2!   sV    ?XXqc]BHHcUO,XXse_bhhsm,
 	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}| j	                  ||       y )Nr   )chain3g      ?g      ?r   )r   r&   r   r#   r   )r   r&   r   s      r   test_chain3zTestChainedAssign.test_chain3)   sV    >XXqc]BHHcUO,XXse_bhhsm,
 	FD)r   c                     ddl m} t        j                  dg      gt        j                  dg      gg}| j	                  ||       y )Nr   )unpack3r   r   )r   r)   r   r#   r   )r   r)   r   s      r   test_unpack3zTestChainedAssign.test_unpack31   =    ?XXqc]OXXse_
 	GT*r   c                     ddl m} t        j                  dg      gt        j                  dg      gg}| j	                  ||       y )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}| j	                  ||       y )Nr   )unpack5r	   r"   )r   r0   r   r#   r   )r   r0   r   s      r   test_unpack5zTestChainedAssign.test_unpack5A   r+   r   c                 @    ddl m} d}d}| j                  |||g       y )Nr   )unpack6)r   r	   )r   r"   )r   r3   r   )r   r3   args1args2s       r   test_unpack6zTestChainedAssign.test_unpack6I   s#    ?GeU^4r   c                     t        |      }|D ]R  }t        j                  |      }t        j                  |      }t        j                  j                   ||  ||        T y )N)r   copydeepcopyr   testingassert_allclose)r   pyfuncargcasescfuncr   a1a2s          r   r   z TestChainedAssign._test_templateO   sM    FDt$Bt$BJJ&&vr{E2J? r   N)__name__
__module____qualname__r   r   r   r$   r'   r*   r.   r1   r6   r    r   r   r   r      s5    **++*+++5@r   r   __main__)numbar   unittestnumpyr   r8   numba.tests.supportr   TestCaser   rA   mainrD   r   r   <module>rL      sG        /L@):): L@^ zHMMO r   