3
Kd                 @   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   sL  t jjt jd tdd } tdd t| D }tt|}t j|}||_	td| }t
||}x"|D ]}tt|j|p~|k qjW tt|jt|jk dd |D }d	d |D }d
d |D }	tdd t|||	D }
||
 }||
 }x$|D ]}tt|j|p|k qW tt j|j |k tt|jt|jk d S )N
         c             s   s   | ]}t d d V  qdS )r   r	   N)r   ).0dim r   U/var/www/html/virt/lib64/python3.6/site-packages/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   r   
<listcomp>   s    ztest.<locals>.<listcomp>c             S   s   g | ]}t |d  qS )r	   )r   )r
   r   r   r   r   r       s    c             S   s   g | ]}t |d  qS )r	   )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aZ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   