o
    Df                     @   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   sB  t jt d tdd } tdd t| D }tt|}t |}||_	td| }t
||}|D ]}tt|j|p>|k q4t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 qzt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/home/ubuntu/webapp/venv/lib/python3.10/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   
<listcomp>   s    ztest.<locals>.<listcomp>c                 S      g | ]}t |d  qS r
   r   r   r   r   r   r           c                 S   r   r   r   r   r   r   r   r   !   r   c                 s   s    | ]}t | V  qd S )N)slice)r   tr   r   r   r   "   s    )nprandomseedaranger   tupleranger   r   shaper   r   lenflatlistzipall	__array__)ndimsr   elsabuf_sizebblockstartstopstepslice_cdr   r   r   test
   s*   


r1   )operatorr   	functoolsr   numpyr   numpy.randomr   	numpy.libr   numpy.testingr   r1   r   r   r   r   <module>   s    