
    sh(	                     n    S SK Jr  S SKJrJr  S SKr " S S\5      r\S:X  a  \R                  " 5         gg)    )cuda)unittestCUDATestCaseNc                   2    \ rS rSrS rS rS rS rS rSr	g)	TestIterators   c                     [         R                  S 5       n[        R                  " S5      n[        R                  " S[        R
                  S9nUS   " X#5        U R                  US   S5        g )Nc                     Sn[        U 5       H!  u  p4X#:w  a  SUS'   X@U   :w  a  SUS'   US-  nM#     U[        U 5      :w  a  SUS'   g g Nr            )	enumeratelen)xerrorcountivs        z/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/cuda/tests/cudapy/test_iterators.py
enumerator0TestIterators.test_enumerate.<locals>.enumerator
   s[    E!!: E!H!9 E!H
 % Aa     
   	      r      r   dtyper   r   r   )r   jitnpasarrayzerosint32assertEqual)selfr   r   r   s       r   test_enumerateTestIterators.test_enumerate	   s^    		 
	 JJ'("((+4"q1%r   c                     [         R                  " S5      n[         R                  " S5      n[         R                  " S[         R                  S9nUS   " X#U5        U R	                  US   S5        g )Nr   )r   r   r         r   r   r!   r   )r#   r$   r%   r&   r'   )r(   fr   yr   s        r   _test_twoarg_function#TestIterators._test_twoarg_function   sU    JJ'(JJ'"((+	$eq1%r   c                 R    [         R                  S 5       nU R                  U5        g )Nc                     Sn[        X5       H$  u  pEX@U   :w  a  SUS'   XQU   :w  a  SUS'   US-  nM&     U[        U 5      :w  a  SUS'   g g r   )zipr   )r   r/   r   r   xvyvs         r   zipper&TestIterators.test_zip.<locals>.zipper(   s_    Aa)1: E!H1: E!HQ $ CF{a r   r   r"   r0   )r(   r7   s     r   test_zipTestIterators.test_zip'   s(    		 
	 	""6*r   c                 R    [         R                  S 5       nU R                  U5        g )Nc                     Sn[        [        X5      5       H1  u  nu  pVXC:w  a  SUS'   XPU   :w  a  SUS'   XaU   :w  a  SUS'   US-  nM3     U[        U 5      :w  a  SUS'   g g Nr   r   r   r   r,   )r   r4   r   r   r/   r   r   r   r5   r6   s          r   enumerator_zipper;TestIterators.test_enumerate_zip.<locals>.enumerator_zipper:   sv    E(Q38B: E!H1: E!H1: E!H
  4 Aa r   r9   )r(   r@   s     r   test_enumerate_zip TestIterators.test_enumerate_zip9   )    		 
	  	""#45r   c                 R    [         R                  S 5       nU R                  U5        g )Nc                     Sn[        [        U 5      U5       H1  u  u  pEnXC:w  a  SUS'   XPU   :w  a  SUS'   XaU   :w  a  SUS'   US-  nM3     U[        U 5      :w  a  SUS'   g g r>   )r4   r   r   r?   s          r   zipper_enumerator;TestIterators.test_zip_enumerate.<locals>.zipper_enumeratorN   sx    E"9Q<3: E!H1: E!H1: E!H
  4 Aa r   r9   )r(   rG   s     r   test_zip_enumerate TestIterators.test_zip_enumerateM   rD   r    N)
__name__
__module____qualname____firstlineno__r)   r0   r:   rB   rI   __static_attributes__rK   r   r   r   r      s    &,&+$6(6r   r   __main__)
numbar   numba.cuda.testingr   r   numpyr#   r   rL   mainrK   r   r   <module>rV      s5     5 X6L X6v zMMO r   