o
    8ήcK                     @   sB   d dl Z d dlZd dlZG dd de jZedkre   dS dS )    Nc                   @   s   e Zd Zdd ZdS )NumpyGeneratorUsageTestc           
         s   t jd}t jd}d tj fdd}|j d}||}|jt d d}t ||f}t||D ]
\}}	| ||	 q7d S )N   
   c                    s   | j t d dS )N   size)randomint)genr    U/tmp/pip-target-vg8gfxp4/lib/python/numba/tests/doc_examples/test_numpy_generators.pydo_stuff   s   z>NumpyGeneratorUsageTest.test_numpy_gen_usage.<locals>.do_stuffr   r   )	npr   default_rngnumbanjitr	   concatenatezipassertEqual)
selfxyr   originalnumba_func_resafter_numba	numba_res_np_res_nb_resr   r   r   test_numpy_gen_usage
   s   z,NumpyGeneratorUsageTest.test_numpy_gen_usageN)__name__
__module____qualname__r   r   r   r   r   r      s    r   __main__)unittestnumpyr   r   TestCaser   r   mainr   r   r   r   <module>   s   