
    `i(	                     t    d dl mZ d dlmZmZ d dlZ G d de          Zedk    r ej	                     dS dS )    )cuda)unittestCUDATestCaseNc                   ,    e Zd Zd Zd Zd Zd Zd ZdS )TestIteratorsc                     t           j        d             }t          j        d          }t          j        dt          j                  } |d         ||           |                     |d         d           d S )Nc                     d}t          |           D ]&\  }}||k    rd|d<   || |         k    rd|d<   |dz  }'|t          |           k    rd|d<   d S d S Nr            )	enumeratelen)xerrorcountivs        z/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/numba/cuda/tests/cudapy/test_iterators.py
enumeratorz0TestIterators.test_enumerate.<locals>.enumerator
   sv    E!!  1A:: E!H!99 E!H
Aa     
   	            r   dtyper   r   r   )r   jitnpasarrayzerosint32assertEqual)selfr   r   r   s       r   test_enumeratezTestIterators.test_enumerate	   s|    		 	 
	 J'(("(+++
4E"""q1%%%%%r   c                     t          j        d          }t          j        d          }t          j        dt           j                  } |d         |||           |                     |d         d           d S )Nr   )r   r   r         r   r   r    r   )r"   r#   r$   r%   r&   )r'   fr   yr   s        r   _test_twoarg_functionz#TestIterators._test_twoarg_function   so    J'((J''"(+++$1eq1%%%%%r   c                 Z    t           j        d             }|                     |           d S )Nc                     d}t          | |          D ],\  }}|| |         k    rd|d<   |||         k    rd|d<   |dz  }-|t          |           k    rd|d<   d S d S r
   )zipr   )r   r-   r   r   xvyvs         r   zipperz&TestIterators.test_zip.<locals>.zipper(   s|    Aa))  B1:: E!H1:: E!HQCFF{{a {r   r   r!   r.   )r'   r4   s     r   test_zipzTestIterators.test_zip'   s9    		 	 
	 	""6*****r   c                 Z    t           j        d             }|                     |           d S )Nc                     d}t          t          | |                    D ]:\  }\  }}||k    rd|d<   || |         k    rd|d<   |||         k    rd|d<   |dz  };|t          |           k    rd|d<   d S d S Nr   r   r   r   r*   )r   r1   r   r   r-   r   r   r   r2   r3   s          r   enumerator_zipperz;TestIterators.test_enumerate_zip.<locals>.enumerator_zipper:   s    E(Q33  8B:: E!H1:: E!H1:: E!H
Aa r   r5   )r'   r;   s     r   test_enumerate_zipz TestIterators.test_enumerate_zip9   :    		 	 
	  	""#455555r   c                 Z    t           j        d             }|                     |           d S )Nc                     d}t          t          |           |          D ]:\  \  }}}||k    rd|d<   || |         k    rd|d<   |||         k    rd|d<   |dz  };|t          |           k    rd|d<   d S d S r9   )r1   r   r   r:   s          r   zipper_enumeratorz;TestIterators.test_zip_enumerate.<locals>.zipper_enumeratorN   s    E"9Q<<33  B:: E!H1:: E!H1:: E!H
Aa r   r5   )r'   r@   s     r   test_zip_enumeratez TestIterators.test_zip_enumerateM   r=   r   N)__name__
__module____qualname__r(   r.   r6   r<   rA    r   r   r   r      s_        & & &,& & &+ + +$6 6 6(6 6 6 6 6r   r   __main__)
numbar   numba.cuda.testingr   r   numpyr"   r   rB   mainrE   r   r   <module>rK      s          5 5 5 5 5 5 5 5    X6 X6 X6 X6 X6L X6 X6 X6v zHMOOOOO r   