o
    Z۷i	                     @   s\   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 deejZ	e
dkr,e  dS dS )    )jitN)MemoryLeakMixinc                   @   s\   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd ZdS )TestChainedAssignc                 C   s:   ddl m} tdgtjdtjdgg}| || d S )Nr   )chain1      )dtype)#numba.tests.chained_assign_usecasesr   nparangedouble_test_template)selfr   args r   U/home/ubuntu/vllm_env/lib/python3.10/site-packages/numba/tests/test_chained_assign.pytest_chain1	   s
   
zTestChainedAssign.test_chain1c                 C   s(   ddl m} dgdgg}| || d S )Nr   )chain2         @)r	   r   r   )r   r   r   r   r   r   test_chain2   s
   zTestChainedAssign.test_chain2c                 C   s,   ddl m} ddgddgg}| || d S )Nr   )unpack1   r         ?r   )r	   r   r   )r   r   r   r   r   r   test_unpack1   s
   zTestChainedAssign.test_unpack1c                 C   sL   ddl m} tdgtdggtdgtdggg}| || d S )Nr   )unpack2r   g      @       @r   )r	   r   r
   arrayr   )r   r   r   r   r   r   test_unpack2!   
   zTestChainedAssign.test_unpack2c                 C   sL   ddl m} tdgtdggtdgtdggg}| || d S )Nr   )chain3g      ?g      ?r   )r	   r    r
   r   r   )r   r    r   r   r   r   test_chain3)   r   zTestChainedAssign.test_chain3c                 C   8   ddl m} tdggtdggg}| || d S )Nr   )unpack3r   r   )r	   r#   r
   r   r   )r   r#   r   r   r   r   test_unpack31   
   zTestChainedAssign.test_unpack3c                 C   r"   )Nr   )unpack4r   r   )r	   r&   r
   r   r   )r   r&   r   r   r   r   test_unpack49   r%   zTestChainedAssign.test_unpack4c                 C   r"   )Nr   )unpack5r   r   )r	   r(   r
   r   r   )r   r(   r   r   r   r   test_unpack5A   r%   zTestChainedAssign.test_unpack5c                 C   s(   ddl m} d}d}| |||g d S )Nr   )unpack6)r   r   )r   r   )r	   r*   r   )r   r*   args1args2r   r   r   test_unpack6I   s   zTestChainedAssign.test_unpack6c                 C   s@   t |}|D ]}t|}t|}tj|| ||  qd S )N)r   copydeepcopyr
   testingassert_allclose)r   pyfuncargcasescfuncr   a1a2r   r   r   r   O   s   

z TestChainedAssign._test_templateN)__name__
__module____qualname__r   r   r   r   r!   r$   r'   r)   r-   r   r   r   r   r   r      s    r   __main__)numbar   unittestnumpyr
   r.   numba.tests.supportr   TestCaser   r7   mainr   r   r   r   <module>   s    O