
    wg(	                     l    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                          yy)    )cuda)unittestCUDATestCaseNc                   *    e Zd Zd Zd Zd Zd Zd Zy)TestIteratorsc                     t         j                  d        }t        j                  d      }t        j                  dt        j
                        } |d   ||       | j                  |d   d       y )Nc                     d}t        |       D ]!  \  }}||k7  rd|d<   || |   k7  rd|d<   |dz  }# |t        |       k7  rd|d<   y y Nr            )	enumeratelen)xerrorcountivs        k/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/numba/cuda/tests/cudapy/test_iterators.py
enumeratorz0TestIterators.test_enumerate.<locals>.enumerator
   sf    E!! 1A: E!H!9 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	   sb    		 
	 JJ'("((+
4E"q1%r   c                     t        j                  d      }t        j                  d      }t        j                  dt         j                        } |d   |||       | j	                  |d   d       y )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   sY    JJ'(JJ'"((+$1eq1%r   c                 R    t         j                  d        }| j                  |       y )Nc                     d}t        | |      D ]$  \  }}|| |   k7  rd|d<   |||   k7  rd|d<   |dz  }& |t        |       k7  rd|d<   y y r
   )zipr   )r   r-   r   r   xvyvs         r   zipperz&TestIterators.test_zip.<locals>.zipper(   sl    Aa) B1: E!H1: E!HQ CF{a r   r   r!   r.   )r'   r4   s     r   test_zipzTestIterators.test_zip'   s(    		 
	 	""6*r   c                 R    t         j                  d        }| j                  |       y )Nc                     d}t        t        | |            D ]1  \  }\  }}||k7  rd|d<   || |   k7  rd|d<   |||   k7  rd|d<   |dz  }3 |t        |       k7  rd|d<   y y 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 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5r   c                 R    t         j                  d        }| j                  |       y )Nc                     d}t        t        |       |      D ]1  \  \  }}}||k7  rd|d<   || |   k7  rd|d<   |||   k7  rd|d<   |dz  }3 |t        |       k7  rd|d<   y y r9   )r1   r   r   r:   s          r   zipper_enumeratorz;TestIterators.test_zip_enumerate.<locals>.zipper_enumeratorN   s    E"9Q<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(6r   r   __main__)
numbar   numba.cuda.testingr   r   numpyr"   r   rB   mainrE   r   r   <module>rK      s7     5 X6L X6v zHMMO r   