a
    Lf                     @   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   .0dim r   Z/var/www/ai-form-bot/venv/lib/python3.9/site-packages/numpy/lib/tests/test_arrayterator.py	<genexpr>       ztest.<locals>.<genexpr>   c                 S   s   g | ]}t |qS r   r   r   r   r   r   
<listcomp>   r   ztest.<locals>.<listcomp>c                 S   s   g | ]}t |d  qS r
   r   r   r   r   r   r       r   c                 S   s   g | ]}t |d  qS r   r   r   r   r   r   r   !   r   c                 s   s   | ]}t | V  qd S )N)slice)r   tr   r   r   r   "   r   )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   numpyr   Znumpy.randomr   Z	numpy.libr   Znumpy.testingr   r)   r   r   r   r   <module>   s   