U
    O8c                     @   sP   d dl mZ d dlmZ d dlZd dlmZ d dlm	Z	 d dl
mZ dd ZdS )	    )mul)reduceNrandint)Arrayterator)assert_c                  C   sD  t jt d tdd } tdd t| D }tt|}t |}||_	td| }t
||}|D ]}tt|j|p||k qhtt|jt|jk dd |D }d	d |D }d
d |D }	tdd t|||	D }
||
 }||
 }|D ]}tt|j|p
|k qtt | |k tt|jt|jk d S )N
         c                 s   s   | ]}t d d V  qdS )r   r
   Nr   .0Zdim r   E/tmp/pip-unpacked-wheel-fd_gsd75/numpy/lib/tests/test_arrayterator.py	<genexpr>   s     ztest.<locals>.<genexpr>   c                 S   s   g | ]}t |qS r   r   r   r   r   r   
<listcomp>   s     ztest.<locals>.<listcomp>c                 S   s   g | ]}t |d  qS r
   r   r   r   r   r   r       s     c                 S   s   g | ]}t |d  qS r   r   r   r   r   r   r   !   s     c                 s   s   | ]}t | V  qd S )N)slice)r   tr   r   r   r   "   s     )nprandomseedZaranger   tupleranger   r   shaper   r   lenZflatlistzipallZ	__array__)Zndimsr   Zelsabuf_sizebblockstartstopstepZslice_cdr   r   r   test
   s*    


r(   )operatorr   	functoolsr   Znumpyr   Znumpy.randomr   Z	numpy.libr   Znumpy.testingr   r(   r   r   r   r   <module>   s   