File: //lib/python3/dist-packages/numpy/lib/tests/__pycache__/test_arrayterator.cpython-310.pyc
o
6��a � @ sP d dl mZ d dlmZ d dlZd dlmZ d dlm Z d dl
m Z dd� Z dS ) � )�mul)�reduceN��randint)� Arrayterator)�assert_c
C sB t j�t �d�� td�d } tdd� t| �D ��}tt|�}t �|�}||_ td| �}t
||�}|D ]
}t t |j
�|p>|k� q4t t|j
�t|j
�k� dd� |D �}d d� |D �}d
d� |D �} td 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 qdS )r r
Nr ��.0�dim� r �C/usr/lib/python3/dist-packages/numpy/lib/tests/test_arrayterator.py� <genexpr> s � ztest.<locals>.<genexpr>� c S s g | ]}t |��qS r r r r r r �
<listcomp> s ztest.<locals>.<listcomp>c S � g | ]}t |�d �qS �r
r r r r r r � c S r r r r r r r r ! r c s s � | ]}t |� V qd 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