o
    0 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      )Zdtype)#numba.tests.chained_assign_usecasesr   npZarangedouble_test_template)selfr   args r   k/home/app/PaddleOCR-VL-test/.venv_paddleocr/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(   Zargs1Z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	   testingZassert_allclose)r   ZpyfuncZargcasesZcfuncr   Za1Z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__)Znumbar   Zunittestnumpyr	   r*   Znumba.tests.supportr   ZTestCaser   r-   mainr   r   r   r   <module>   s    O