Edit File: test_nditer.cpython-37.pyc
B ��Fd�� � @ s< d dl mZmZmZ d dlZd dlZd dlZd dlmZm Z m Z m Z d dl m Z d dlmZmZmZmZmZmZmZmZmZ dd� Zdd � Zd d � Ze�e d �d d� �Z dd� Z dd� Z dd� Z dd� Z dd� Z!dd� Z"dd � Z#d d � Z$d d � Z%d!d"� Z&d#d$� Z'd%d&� Z(d'd(� Z)d)d*� Z*d+d,� Z+d-d.� Z,d/d0� Z-d1d2� Z.d3d4� Z/d5d6� Z0d7d8� Z1d9d:� Z2d;d<� Z3d=d>� Z4d?d@� Z5dAdB� Z6dCdD� Z7dEdF� Z8dGdH� Z9dIdJ� Z:dKdL� Z;dMdN� Z<dOdP� Z=dQdR� Z>dSdT� Z?dUdV� Z@dWdX� ZAdYdZ� ZBd[d\� ZCd]d^� ZDd_d`� ZEdadb� ZFdcdd� ZGdedf� ZHdgdh� ZIdidj� ZJdkdl� ZKdmdn� ZLdodp� ZMdqdr� ZNdsdt� ZOdudv� ZPdwdx� ZQdydz� ZRd{d|� ZSd}d~� ZTdd�� ZUejVd�d�� �ZWd�d�� ZXd�d�� ZYd�d�� ZZd�d�� Z[d�d�� Z\d�d�� Z]d�d�� Z^d�d�� Z_d�d�� Z`d�d�� Zad�d�� Zbd�d�� Zcd�d�� Zdd�d�� Zed�d�� Zfd�d�� Zgd�d�� Zhd�d�� Zid�d�� Zjd�d�� Zkd�d�� Zlemd�k�r8e� dS )�� )�division�absolute_import�print_functionN)�array�arange�nditer�all)�test_nditer_too_large) �run_module_suite�assert_� assert_equal�assert_array_equal� assert_raises� assert_warns�dec� HAS_REFCOUNT�suppress_warningsc C s( g }x | j s"|�| j� | �� qW |S )N)�finished�append� multi_index�iternext)�i�ret� r ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/core/tests/test_nditer.py�iter_multi_index s r c C s( g }x | j s"|�| j� | �� qW |S )N)r r �indexr )r r r r r � iter_indices s r c C s( g }x | j s"|�| j� | �� qW |S )N)r r � iterindexr )r r r r r �iter_iterindices s r z$python does not have sys.getrefcountc C sf t d�} t�d��� }t�| �}t�|�}t| g ddggd|gd�}t|j � tt�| �|k� tt�|�|k� d }t t�| �|� t t�|�|� t ddd�} t�d�}t�| �}t�|�}t| g dgg|gd�}t�| �}t�|�}|� � }tt�| �|k� tt�|�|k� d }t t�| �|� t t�|�|� d }t t�| �|� t t�|�|� ~d S ) N� �f4� readwrite� updateifcopy�unsafe)�casting� op_dtypes)�dtype)r& ) r �npr' � newbyteorder�sys� getrefcountr r �iterationneedsapir �copy)�a�dtZrc_aZrc_dt�itZrc2_aZrc2_dt�it2r r r �test_iter_refcount$ sB r2 c C s �xdD �]} t t�| ��}x�tdt| � �D ]�}td �gt| � }x0tt| ��D ] }d| |@ rRtd d d�||< qRW t|�}|�| �| }t|g dgg�}t dd� |D �|� t|j g dgg�}t dd� |D �|� t| �dkr.t|� dd �g dgg�}t d d� |D �|� q.W qW d S ) N))� )� � )� r4 r5 )r6 r4 r5 r4 )r6 r4 r6 r6 r4 r6 ����readonlyc S s g | ]}|�qS r r )�.0�xr r r � <listcomp>` s z(test_iter_best_order.<locals>.<listcomp>c S s g | ]}|�qS r r )r9 r: r r r r; c s r � c S s g | ]}|�qS r r )r9 r: r r r r; g s ) r r( �prod�range�len�slice�tuple�reshaper r �T�swapaxes)�shaper. �dirs� dirs_index�bit�aviewr r r r �test_iter_best_orderN s rJ c C s2 �x*dD �] } t t�| ��}�x tdt| � �D ]�}td �gt| � }x0tt| ��D ] }d| |@ rTtd d d�||< qTW t|�}|�| �| }t|dd�}t dd� |D �|j dd�� t|j dd�}t dd� |D �|j j dd�� t| �dkr0t|� d d �dd�}t d d� |D �|� d d �j dd�� q0W qW d S ) N))r3 )r4 r5 )r6 r4 r5 )r6 r4 r5 r4 )r6 r4 r6 r6 r4 r6 r7 �C)�orderc S s g | ]}|�qS r r )r9 r: r r r r; z s z%test_iter_c_order.<locals>.<listcomp>c S s g | ]}|�qS r r )r9 r: r r r r; } s r r< c S s g | ]}|�qS r r )r9 r: r r r r; � s ) r r( r= r> r? r@ rA rB r r �ravelrC rD )rE r. rF rG rH rI r r r r �test_iter_c_orderi s" rN c C s2 �x*dD �] } t t�| ��}�x tdt| � �D ]�}td �gt| � }x0tt| ��D ] }d| |@ rTtd d d�||< qTW t|�}|�| �| }t|dd�}t dd� |D �|j dd�� t|j dd�}t dd� |D �|j j dd�� t| �dkr0t|� d d �dd�}t d d� |D �|� d d �j dd�� q0W qW d S ) N))r3 )r4 r5 )r6 r4 r5 )r6 r4 r5 r4 )r6 r4 r6 r6 r4 r6 r7 �F)rL c S s g | ]}|�qS r r )r9 r: r r r r; � s z%test_iter_f_order.<locals>.<listcomp>c S s g | ]}|�qS r r )r9 r: r r r r; � s r r< c S s g | ]}|�qS r r )r9 r: r r r r; � s ) r r( r= r> r? r@ rA rB r r rM rC rD )rE r. rF rG rH rI r r r r �test_iter_f_order� s" rP c C s2 �x*dD �] } t t�| ��}�x tdt| � �D ]�}td �gt| � }x0tt| ��D ] }d| |@ rTtd d d�||< qTW t|�}|�| �| }t|dd�}t dd� |D �|j dd�� t|j dd�}t dd� |D �|j j dd�� t| �dkr0t|� d d �dd�}t d d� |D �|� d d �j dd�� q0W qW d S ) N))r3 )r4 r5 )r6 r4 r5 )r6 r4 r5 r4 )r6 r4 r6 r6 r4 r6 r7 �A)rL c S s g | ]}|�qS r r )r9 r: r r r r; � s z*test_iter_c_or_f_order.<locals>.<listcomp>c S s g | ]}|�qS r r )r9 r: r r r r; � s r r< c S s g | ]}|�qS r r )r9 r: r r r r; � s ) r r( r= r> r? r@ rA rB r r rM rC rD )rE r. rF rG rH rI r r r r �test_iter_c_or_f_order� s" rR c C sf t d�} t| dgdgg�}tt|�ddddg� t| d d d� dgdgg�}tt|�ddddg� d S ) Nr5 r r8 )r )r< )r6 )r4 r7 )r r r r )r. r r r r �#test_iter_best_order_multi_index_1d� s rS c C s, t d�} t| �dd�dgdgg�}tt|�dddd d d g� t| �dd�jd d �dgdgg�}tt|�dd dd dd g� t| �dd�d d d� dgdgg�}tt|�d d d dddg� t| �dd�d d �d d d�f dgdgg�}tt|�dddd d d g� t| �dd�d d d�d d d�f dgdgg�}tt|�d d d dddg� t| �dd�jd d �d d d� dgdgg�}tt|�d dd dd dg� t| �dd�jd d �d d �d d d�f dgdgg�}tt|�dd dd dd g� t| �dd�jd d �d d d�d d d�f dgdgg�}tt|�d dd dd dg� d S )Nr r6 r4 r r8 )r r )r r< )r r6 )r< r )r< r< )r< r6 rO )rL r7 )r r rB r r r- )r. r r r r �#test_iter_best_order_multi_index_2d� s& "$,.,&(rT c C s� t d�} t| �ddd�dgdgg�}tt|�dddd d d d d ddddg � t| �ddd�jdd�dgdgg�}tt|�dd ddd ddd d dd dg � t| �ddd�d d d� dgdgg�}tt|�d d dddddddd d d g � t| �ddd�d d �d d d�f dgdgg�}tt|�d d dd ddddddd d g � t| �ddd�d d �d d �d d d�f dgdgg�}tt|�ddd dd d d d ddddg � t| �ddd�jdd�d d d� dgdgg�}tt|�d ddddd d ddd dd g � t| �ddd�jdd�d d �d d d�f dgdgg�}tt|�d ddddd d dd ddd g � t| �ddd�jdd�d d �d d �d d d�f dgdgg�}tt|�dd d dd ddd ddd dg � d S )N� r6 r4 r r8 )r r r )r r r< )r r< r )r r< r< )r r6 r )r r6 r< )r< r r )r< r r< )r< r< r )r< r< r< )r< r6 r )r< r6 r< rO )rL r7 )r r rB r r r- )r. r r r r �#test_iter_best_order_multi_index_3d� sH $ & . 4 ( . rV c C sf t d�} t| dgdgg�}tt|�ddddg� t| d d d� dgdgg�}tt|�ddddg� d S ) Nr5 �c_indexr8 r r< r6 r4 r7 )r r r r )r. r r r r � test_iter_best_order_c_index_1d s rX c C s, t d�} t| �dd�dgdgg�}tt|�dddddd g� t| �dd�jd d �dgdgg�}tt|�dddddd g� t| �dd�d d d � dgdgg�}tt|�ddd dddg� t| �dd�d d �d d d �f dgdgg�}tt|�dddd ddg� t| �dd�d d d �d d d �f dgdgg�}tt|�d dddddg� t| �dd�jd d �d d d � dgdgg�}tt|�ddddd dg� t| �dd�jd d �d d �d d d �f dgdgg�}tt|�dd ddddg� t| �dd�jd d �d d d �d d d �f dgdgg�}tt|�d dddddg� d S ) Nr r6 r4 rW r8 r r< r5 r3 rO )rL r7 )r r rB r r r- )r. r r r r � test_iter_best_order_c_index_2d s* $,. &(rY c C s� t d�} t| �ddd�dgdgg�}tt|�dddddd d d d d ddg � t| �ddd�jdd�dgdgg�}tt|�dd dd dddd dd d dg � t| �ddd�d d d� dgdgg�}tt|�d d d d dddddddd g � t| �ddd�d d �d d d�f dgdgg�}tt|�dd ddddddd d d d g � t| �ddd�d d �d d �d d d�f dgdgg�}tt|�ddddd dd d d d ddg � t| �ddd�jdd�d d d� dgdgg�}tt|�d dd dddd dd ddd g � t| �ddd�jdd�d d �d d d�f dgdgg�}tt|�dddd dd d ddd dd g � t| �ddd�jdd�d d �d d �d d d�f dgdgg�}tt|�dd dd d ddd dd ddg � d S )NrU r6 r4 rW r8 r r< r5 r3 r � � � � � rO )rL r7 )r r rB r r r- )r. r r r r � test_iter_best_order_c_index_3d4 s: & . 4 ( .r_ c C sf t d�} t| dgdgg�}tt|�ddddg� t| d d d� dgdgg�}tt|�ddddg� d S ) Nr5 �f_indexr8 r r< r6 r4 r7 )r r r r )r. r r r r � test_iter_best_order_f_index_1dY s ra c C s, t d�} t| �dd�dgdgg�}tt|�dddddd g� t| �dd�jd d �dgdgg�}tt|�dddddd g� t| �dd�d d d � dgdgg�}tt|�ddd dddg� t| �dd�d d �d d d �f dgdgg�}tt|�dddd ddg� t| �dd�d d d �d d d �f dgdgg�}tt|�d dddddg� t| �dd�jd d �d d d � dgdgg�}tt|�ddddd dg� t| �dd�jd d �d d �d d d �f dgdgg�}tt|�dd ddddg� t| �dd�jd d �d d d �d d d �f dgdgg�}tt|�d dddddg� d S ) Nr r6 r4 r` r8 r r5 r< r3 rO )rL r7 )r r rB r r r- )r. r r r r � test_iter_best_order_f_index_2dd s* $,. &(rb c C s� t d�} t| �ddd�dgdgg�}tt|�ddddd d d d dd ddg � t| �ddd�jdd�dgdgg�}tt|�dd ddd ddd dd d dg � t| �ddd�d d d� dgdgg�}tt|�d d dd ddddddd d g � t| �ddd�d d �d d d�f dgdgg�}tt|�d d dddddddd d d g � t| �ddd�d d �d d �d d d�f dgdgg�}tt|�ddddd d d d d dddg � t| �ddd�jdd�d d d� dgdgg�}tt|�d ddddd d dd ddd g � t| �ddd�jdd�d d �d d d�f dgdgg�}tt|�d ddddd d ddd dd g � t| �ddd�jdd�d d �d d �d d d�f dgdgg�}tt|�dd dd d ddd ddd dg � d S )NrU r6 r4 r` r8 r r r[ r5 r] r< rZ r\ r3 r^ rO )rL r7 )r r rB r r r- )r. r r r r � test_iter_best_order_f_index_3d� s: & . 4 ( .rc c C sF �x>dD �]4} t �| �}t|�}�xtdt| � �D �]}td �gt| � }x0tt| ��D ] }d| |@ rZtd d d�||< qZW t|�}|�| �| }t|dgdgg�}t |j d� t |d j |f� t|j dgdgg�}t |j d� t |d j |f� t| �dkr4t|� dd�dgdgg�}t |j d� t |d j |f� q4W qW d S )N))r3 )r4 r5 )r6 r4 r5 )r6 r4 r5 r4 )r6 r4 r6 r6 r4 r6 r7 � external_loopr8 r< r )r( r= r r> r? r@ rA rB r r �ndimrE rC rD )rE �sizer. rF rG rH rI r r r r � test_iter_no_inner_full_coalesce� s* rg c C sV t d��ddd�d d �d d �d d�f } t| dgdgg�}t|jd� t|d jd � t d��ddd�d d �d d�d d �f } t| dgdgg�}t|jd� t|d jd � t d��ddd�d d�d d �d d �f } t| dgdgg�}t|jd � t|d jd � t d��d d dd d dd d dd d � } t| dgdgg�}t|jd � t|d jd � d S )N� r6 r4 r5 r7 rd r8 r )r4 )r[ r< )rU )rh )r rB r r re rE )r. r r r r �!test_iter_no_inner_dim_coalescing� s ( ( ( " ri c C s� t d��ddd�} t| dgdgg�}t|jd� t d��ddd�}t|dgdgg�}t|jd� t|�d d�dgdgg�}t|jd� t|jdgdgg�}t|jd� t|jd gdgg�}t|jd� t|j�d d�d gdgg�}t|jd� t d��ddd�}t|d d �}t|jd� t|jd d �}t|jd� t|d d �}t|jd� t|jd d �}t|jd� t|dd �}t|jd� t|jdd �}t|jd� d S )Nrh r6 r4 r5 r r8 rW r< r r` rK )rL rO rQ )r rB r r re rD rC )r. r Za3dr r r �test_iter_dim_coalescing� s6 rj c C s� t td�t�d�gdgdggd �} t| jd� t| jd� t td��dd�t�d�gdgdggd �} t| jd� t| jd� t td��dd�td�gdgdggd �} t| jd� t| jd� t td��dd�td�gdgdggd �} t| jd� t| jd� t td��dd�td��dd�gdgdggd �} t| jd� t| jd� t t�d�td ��d dd�gdgdggd �} t| jd � t| jd � t td�td ��d dd�gdgdggd �} t| jd � t| jd � t td�td ��d dd�gdgdggd �} t| jd � t| jd � t td��dd�td ��d dd�gdgdggd �} t| jd � t| jd � t td��dd�td ��d dd�gdgdggd �} t| jd � t| jd � t td��dd�td ��d dd�gdgdggd �} t| jd � t| jd � t td��ddd�td��ddd�td ��d dd�gdgdggd �} t| jd � t| jd � t td��ddd�td ��d dd�gdgdggd �} t| jd � t| jd � t td ��d dd�td ��d dd�gdgdggd �} t| jd � t| jd � d S )Nr r6 r r8 )r r4 )r6 r4 r< rh r5 )r5 r6 r4 r[ rU )r r r( �int32r �itersizerE rB )r r r r �test_iter_broadcasting sp $ " " " " $ $ rm c C sR t jddd��dd�} t| d gg dgddggd d d gd gd d �}t|jd jd � t|jd jd� t| jd gg dgddggd d d gd gd d �}t|jd jd� t|jd jd� t| jd gg dgddggdd d d gd gd d�}t|jd jd� t|jd jd� t t t| d gg dgddggd d d gd gdd � t jt � d�d d gdd�}d S )Nr �i2)r' r6 r4 r8 � writeonly�allocater r< )r7 r7 r5 )�op_axes� itershape)r6 r4 r5 )rh r[ r6 )r4 r6 r5 )r[ rh r6 rO )rL rq rr )r6 r rU )r7 r< r5 )r6 )rr ) r( r rB r r �operandsrE �stridesrC r � ValueError�ones)r. r r r r �test_iter_itershapeF s, rw c C s* t tttd�td�gg dggd � t tttd��dd�td�gg dggd � t tttd��dd�td��dd�gg dggd � t tttd��dd�td��dd�gg dggd � t tttd��ddd�td��ddd�gg dggd � t tttd ��ddd �td��ddd�gg dggd � yRttd��d dd �td��d d�td��dd�gg dgdgd d gg� td ��W n\ tk �r� } z<t| �}t|�d�dkd| � t|�d�dkd| � W d d } ~ X Y nX yHttd��dd�td�gg dgdggdd gdt j ggdd� td ��W nt tk �r� } zTt| �}t|�d�dkd| � t|�d�dkd| � t|�d�dkd| � W d d } ~ X Y nX y2tt � d�t � d �gg d d gdgg� td ��W n\ tk �r$ } z<t| �}t|�d �dkd | � t|�d �dkd | � W d d } ~ X Y nX d S )!Nr6 r4 r8 r r\ r5 �$ rh r[ r< ro � no_broadcastz$Should have raised a broadcast errorz(2,3)r z0Message "%s" doesn't contain operand shape (2,3)z(1,2,3)z4Message "%s" doesn't contain broadcast shape (1,2,3))r5 r4 )rq rr z (2,3)->(2,3)z7Message "%s" doesn't contain operand shape (2,3)->(2,3)z(2,)->(2,newaxis)zDMessage "%s" doesn't contain remapped operand shape(2,)->(2,newaxis)z(4,3)z6Message "%s" doesn't contain itershape parameter (4,3))r6 r< r< )r6 z(2,1,1)z2Message "%s" doesn't contain operand shape (2,1,1)z(2,1,2)z8Message "%s" doesn't contain the broadcast shape (2,1,2)) r ru r r rB �AssertionError�strr �findr( �newaxis�zeros)�e�msgr r r � test_iter_broadcasting_errorse sr "" r� c C s� t d�} tttg g g � ttt| gd g dggd � ttt| gdgdgg� ttt| gg ddgg� ttt| gg dggdd� ttt| gg dggdd� ttt| gd g dggd � ttt| d d gdgg� ttt| d dgdgg� ttt| d d gdgg� ttt| d d gdgg� ttt| g g g� ttt| g ddgg� ttt| g ddgg� ttt| g ddgg� ttt| g dddgg� tttdg dgg� tttdg dgg� tttt�d�g dgg� tttt�d�g dgg� d| j_ttt| g dgg� ttt| g dgg� d| j_tt d�g dgg�}ttdd� |� ttdd� |� dd� }dd� }d d � }d d � }tt d�d g�}tt||� tt||� tt||� tt||� tt d�d g�}tt||� tt||� tt||� tttt� g �� d S )!Nr �d r8 zbad flag�G)rL Znoon)r% r4 r6 rW r` rd r ro r" g �?r< FTc S s | j S )N)r )r r r r �<lambda>� � z(test_iter_flags_errors.<locals>.<lambda>c S s | j S )N)r )r r r r r� � r� c S s d| _ d S )N)r )r )r r r r �assign_multi_index� s z2test_iter_flags_errors.<locals>.assign_multi_indexc S s d| _ d S )Nr )r )r r r r � assign_index� s z,test_iter_flags_errors.<locals>.assign_indexc S s d| _ d S )Nr )r )r r r r �assign_iterindex� s z0test_iter_flags_errors.<locals>.assign_iterindexc S s d| _ d S )N)r r< )� iterrange)r r r r �assign_iterrange� s z0test_iter_flags_errors.<locals>.assign_iterrange�buffered) r r ru r � TypeErrorr( rk �flags� writeabler )r. r r� r� r� r� r r r �test_iter_flags_errors� s^ r� c C s� t �d�t �d�t �d� } }}t| ||gg dg�}d|dd�< t| dddg� t|dddg� t|dddg� d|d< t|dd� ddg� d S ) Nr4 g @r" )r4 r4 r r6 r< rU )r( r r r )r. �b�cr r r r �test_iter_slice� s " r� c C s� t jddd�} | �� �� }t| jj|jjk� t|g ddggdt �d�gd�}t|j d j| jj� t|j d jj| jj� t|j d | � d |j d d d �< d }t|d gd � t jddd�} | �� �� }t| jj|jjk� t|g ddd ggdd �}t|j d j| jj� t|j d jj| jj� t|j d | � d |j d d d �< d }t|d gd � t j d d d�dd � } d| _t jddd�| d d �< t| j j � t| g dgg�}t|j d j j � t|j d | � t| g dddgg�}t|j d j j � t|j d | � d|j d d d �< d }t| dgd � td�} t| d d� g dgg�}t|j d j j� t|j d | d d� � t| d d d � ddgddggdd�}t|d j j� t|d | d d d � � d S )Nr r! )r' r" r# �equiv)r% r& r r6 �nbo)r% )� �i1r< r8 �alignedr4 rU r� rd Zcontigr] )� buffersize)r( r �byteswapr) r r' � byteorderr r �dtypesrs r~ r� r� � contiguous)r. Zaur r r r �test_iter_nbo_align_contig sX r� c C s� t jddd��dd�} t| g dggt �d�gd�}t|jd | � t|jd jt �d�� t jdd d��dd�} t| g dd ggd t �d �gd �}t|jd | � t|jd jt �d �� t jddd��ddd��dd�} t| g ddggdt �d�gd �}t|jd | � t|jd jt �d�� t|jd jd� | d d d�d d �d d d�f } t| g ddggdt �d�gd �}t|jd | � t|jd jt �d�� t|jd jd� t jddd��ddd�j } t| g dd ggdt �d�gd �}t|jd | � t|jd jt �d�� t|jd jd� d|jd d< t | d dk� d }t| d d� t jddd�d d d � } t| g d d ggd t �d�gd �}t|jd jt �d�� t|jd jd � dddg|jd d d �< d }t| dddg� d S ) Nr r! )r' r6 r4 r" )r& r z<f4r# r� z>f4)r% r& rh r5 r< r8 r- �safe�f8)�` r[ � r7 � same_kind)r5 � �0 g )�)r6 r< r< �i4���ro r$ )r5 ) r( r rB r r' r rs rD rt rC r )r. r r r r �test_iter_array_castB s\ r� c C s� t tttddd�g dggt�d�gd� t tttddd�g ddggdt�d�gd � t tttddd�g ddggd t�d�gd � t tttddd�g d d ggdt�d�gd � t tttddd�g d d ggd t�d�gd � t tttdd d�g ddggdt�d�gd � t tttddd�g dd ggdt�d�gd � t tttddd�g dd ggdt�d�gd � t tttddd�g ddggdt�d�gd � t tttddd�g d d ggdt�d�gd � d S )Nr6 r! )r' r8 r� )r& r- �no)r% r& r� ro r# z<f4z>f4r" r� r� r� )r r� r r r( r' r r r r �test_iter_array_cast_errors� sF r� c C s� t t�d�g dggt�d�gd�} t| jd t�d�� t| jjt�d�� t| jd� t t�d�g ddggdt�d�gd �} t| jd t�d�� t| jjt�d�� t| jd� t t�d�g ddggd t�d�gd �} t| jd t�d�� t| jjt�d�� t| jd� t t�d �g ddggd t�d �gd �} t| jd t�d �� t| jjt�d �� t| jd� t dg dggt�d�gd�} t| d jt�d�� t| d d � d S )Ng @r8 r! )r& r r- r� r� )r% r& r� g @r$ r� r4 )r r( �float32r' r r� �value�float64)r r r r �test_iter_scalar_cast� s: r� c C s� t ttt�d�g dggt�d�gd� t ttdg dggt�d�gd� t ttt�d�g dggd t�d�gd � t ttt�d�g dggd t�d �gd � d S ) Nr6 r" r� )r& g @r! gD�����cLr8 r� )r% r& r� r� )r r� r r( r� r' r� r r r r � test_iter_scalar_cast_errors� s r� c C s� ddd�} t jdddgd | d gdd�}tr4t�| �}ttt|� trTtt�| �|� t|dgd g�}d d � |D �}tt j|dd�|� d gd \}}}tr�tt�| �|� t|� dd�j dd gd gd d�}t |j � dd � |D �}tt j|dd�|� dd�j dd�� d gd \}}}t�r,tt�| �|� t|� dd�j dd gdgd d�}x|D ]}d |d< �qRW d gd \}}}t�r�t t�| �|d k� t|t jd gd dd�� d S )Nr4 �d)r. r� r< r6 �O)r' �refs_okr8 c S s g | ] }|d �qS )r r )r9 �x_r r r r; � s z1test_iter_object_arrays_basic.<locals>.<listcomp>r� rK )rL c S s g | ] }|d �qS )r r )r9 r� r r r r; � s rO r" .r5 )r( r r r* r+ r r� r r rB rC r r, rM )�objr. �rcr �valsr: r r r � test_iter_object_arrays_basic� s: $ r� c C s� t jddd�} t| ddgdgddd �}x |D ]}|d d 7 < q,W t| t �d�d � t jddd�} t| ddgdgddd �}x |D ]}|d d 7 < q�W t| t �d�d � t jd d dgd�} | d } t �d�| d d �< t| ddgdgddd �}x |D ]}|d d 7 < q�W t| t �d�d � t jd d dgd�} | d } t �d�d | d d �< t| ddgdgddd �}|d d }t�r�t�|�}x |D ]}|d d 7 < �q�W t�r�tt�|�|d k� t| t �d�d � d S )Nr r� )r' r� r� r" r$ r� )r% r& .r< )r )�pr� )r. r� r. )r. r� iH��r r iI��) r( r r r r~ r r* r+ r )r. r r: �obr� r r r �#test_iter_object_arrays_conversions sB r� c C s� t tdgdd�tdgdd�gdgddggd d d �} t| jd t�d�� t| jd t�d�� t tdgd d�tdgdd�gdgddggd d d �} t| jd t�d�� t| jd t�d�� t tdgdd�tddd�gdgddggd dd �} t| jd t�d�� t| jd t�d�� t tdgdd�tdd d�gdgddggd d d �} t| jd t�d�� t| jd t�d�� t tdgdd�tdd d�gdgddggd d d �} t| jd t�d�� t| jd t�d�� t tdgdd�tdd d�tdgdd�tdgdd�gdgddggd d d �} t| jd t�d�� t| jd t�d�� t| jd t�d�� t| jd t�d�� t| jd� t tdgd d�d tdgdd�gg ddgddgdggd d �} t| jd t�d �� t| jd t�d �� t| jd t�d�� t tdgd d�d tdgdd�gdgddgddgdggd d �} t| jd t�d�� t| jd t�d�� t| jd t�d�� d S )Nr4 r! )r' r r� Z common_dtyper8 r- r6 r� )r% r< r� r� �u4i��i8y @�c8r\ r5 �c16)r4 i�y @r\ ro rp )r r r r� r( r' r� )r r r r �test_iter_common_dtype5 sr r� c C s� x:dD ]2} t d�}t|gdg| gg�}t|jd |k� qW t d�}|dd � }|d d� }t||gdgdgdgg�}ttj|j� � t d�}|}|}t||gdgdd gdd gg�}t|jd |ko�|jd |k� t||gdgdgdgg�}t|jd |k�ot�|jd |� � t d�}|d d d � }|dd d � }t||gdgdgd gg�}t|jd |k�ot|jd |k� t d tjd �}|dd � }|�tj�d d� }t||gdgdgd gg�}ttj|j� � xzdD ]r} t� ddg�}|}|j }|}t|||gdgdgdg| gg�}|j\}}}tt�||� � tt�||� � �q�W t� ddg�}|}|j }|}t|||gdgdgdgdgg�}|j\}}}t||k� t||k� t||k� t� ddg�}|}t� ddg�}|j }t|||gdgdgd gdgg�}|j\}}}t||k� t||k� t||k� t� d�}|d d� }|dd� }|d d� }t|||gdgdgd gd gg�}|j\}}}t||k� t||k� t||k� d S )N)r8 ro r" r] Zcopy_if_overlapr r< r7 r8 r" Zoverlap_assume_elementwiser6 ro r5 )r' r4 )ro r" rZ r ) r r r rs r( � shares_memory�int8�viewrk rv rC ) �flagr. r r: r� r� �a2�b2�c2r r r �test_iter_copy_if_overlapu s� (" r� c C sv t d��dd�} t| | jgg dggd ddgddggd�}ttdd � |D ��� t d ��ddd �} t| j| gg dggd dddgd gd�}ttd d � |D ��� t dd ��ddd�} t dd�}t| |gg dggd d dddggd�}tdd � |D �| |�ddd� �� � t dd �}t| |gg dggd d dddggd�}tdd � |D �| |�ddd� �� � t dd�}t| |gg dggd d tj tj dggd�}tdd � |D �| |�ddd� �� � t d ��ddd �} t d��ddd �}t| |gdgdggd ddddgddddggd�}t|j d� t d��dd �} t d��d d�}t| |gdgdggd ddgddggd�}t|j d� d S )Nr r6 r4 r8 r r< )rq c S s g | ]\}}||k�qS r r )r9 r: �yr r r r; � s z%test_iter_op_axes.<locals>.<listcomp>rh r5 c S s g | ]\}}||k�qS r r )r9 r: r� r r r r; � s � r3 r7 c S s g | ]\}}|| �qS r r )r9 r: r� r r r r; � s c S s g | ]\}}|| �qS r r )r9 r: r� r r r r; � s c S s g | ]\}}|| �qS r r )r9 r: r� r r r r; � s �( r )r6 r4 r3 r6 rU � )r4 r3 ) r rB r rC r r r rM r( r} rE )r. r r� r r r �test_iter_op_axes� s6 *( && && & r� c C sJ t d��dd�} ttt| | gg dggd dgdgdggd� ttt| | gg dggd ddgddggd� ttt| | gg dggd ddgddggd� ttt| | gg dggd ddgddggd� ttt| | gg dggd ddgddggd� ttt| | gg dggd ddgdddggd� ttt| | gg dggd ddgddggd� d S ) Nr r6 r4 r8 r r< )rq r7 )r rB r ru r )r. r r r �test_iter_op_axes_errors� s r� c C s� t d��ddd�} t| �}|�� }tdd� |D �dd� |D �� d|_|�� }tdd� |D �dd� |D �� t| dd gd dd �}|�� }td d� |D �d d� |D �� d|_|�� }td d� |D �d d� |D �� d|_|�� }tdd� |D �dd� |D �� d|_t|� t|� |�� }tdd� |D �dd� |D �� t| dgd dddd�}|�� }d }tdd� |D �| jd d�� t ddd��ddd�} t| dgd dddd�}|�� }d }tdd� |D �| jd d�� d S ) Nrh r6 r4 r5 c S s g | ] }|d �qS )r r )r9 r: r r r r; s z"test_iter_copy.<locals>.<listcomp>c S s g | ] }|d �qS )r r )r9 r: r r r r; s r� �rangedrO )rL r� c S s g | ] }|d �qS )r r )r9 r: r r r r; s c S s g | ] }|d �qS )r r )r9 r: r r r r; ! s )r4 r\ c S s g | ] }|d �qS )r r )r9 r: r r r r; % s )r6 � c S s g | ] }|d �qS )r r )r9 r: r r r r; + s r$ r� r3 )rL r% r& r� c S s g | ] }|d �qS )r r )r9 r: r r r r; 2 s )rL z<i4)r' z>f8c S s g | ] }|d �qS )r r )r9 r: r r r r; 9 s ) r rB r r- r r r� �nextrM )r. r �jr r r �test_iter_copy s@ r� c C s` t d�} t| d gg dgddggd t�d�gd�}t|jd j| j� t|jd jt�d�� d S )Nr r8 ro rp r! )r& r< )r r r( r' r rs rE )r. r r r r � test_iter_allocate_output_simple; s r� c C s� t d�} t| d gddgdgddgg�}d|jd d d �< |�� x(|D ] }|d d |d d 7 < qFW t|jd | d � d S ) Nr r� �delay_bufallocr8 rp r" r< .r )r r rs �resetr )r. r r: r r r �,test_iter_allocate_output_buffered_readwriteE s r� c C s� t ddd��dd�} t| d gg dgddggd t�d �gd �}t|jd j| j� t|jd j| j� t|jd jt�d �� t d dd��ddd �j } t| d gg dgddggd t�d �gd �}t|jd j| j� t|jd j| j� t|jd jt�d �� t d dd��ddd �� dd �} t| d gg dgddggdd t�d �gd�}t|jd j| j� t|jd jd� t|jd jt�d �� d S )Nr r� )r' r6 r4 r8 ro rp r! )r& r< rh r5 r rK )rL r& )r� r� r5 ) r rB r r( r' r rs rE rt rC rD )r. r r r r �!test_iter_allocate_output_itorderQ s( r� c C s� t ddd��ddd�} td | gg ddgd ggt�d �d gd dd gd gd �}t|jd jd� t|jd jd� t|jd jt�d �� d S )Nrh r� )r' r6 r4 r5 ro rp r8 r� r< r )r& rq )r5 r6 r4 )r5 r� r� ) r rB r r( r' r rs rE rt )r. r r r r � test_iter_allocate_output_opaxesl s r� c C s� t tdgdd�tdgdd�d gg dggd dd gg �} t| jd t�d�� t tdgd d�tdgdd�d gg dggd dd gg �} t| jd t�d�� t tdgdd�tddd�d gg dggd dd gg �} t| jd t�d�� t tdgd d�tdd d�d gg dggd dd gg �} t| jd t�d �� t tdgd d�td d d�d gg dggd dd gg �} t| jd t�d �� d S )Nr4 r! )r' r r� r8 r6 ro rp r� r� i�r� )r r r r� r( r' )r r r r �)test_iter_allocate_output_types_promotionw s r� c C s� t dgdd��� } t| d gg dgddgg�}t|jd |jd � t| | d gg dgdgddgg�}t|jd |jd k� t|jd �d �|jd � d S ) Nr4 r� )r' r8 ro rp r r< r6 �=)r r) r r r� r )r. r r r r �*test_iter_allocate_output_types_byte_order� s r� c C sb t d ddt�d�t�d�gg ddggdggd �} t| jd jt�d �� t| jd jd � d S ) Nr< gffffff@rU r4 ro rp r8 r5 r � complex128)r r( r� r� r rs r' re )r r r r �&test_iter_allocate_output_types_scalar� s r� c C s0 t �ddgddgg�} t �d��dd�j}t| |d gg dgdgddgg�}tt| �t|jd �� t t|�t|jd �k� t|jd j d� t �d��ddd�}t t t| |d gg dgdgddgg� t| |d gg dgdgddd gg�}tt|�t|jd �� t t| �t|jd �k� t|jd j d � d S ) Nr< r6 r4 r5 r8 ro rp )r6 r6 Z no_subtype)r< r6 r6 ) r( �matrixr rB rC r r �typers r rE r � RuntimeError)r. r� r r r r �!test_iter_allocate_output_subtype� s r� c C sX t d�} ttt| d gg dgddgg� ttt| d gg dgddgg� ttt| d gdgddg� tttd d gg ddgddggt�d�t�d�gd� t d d d ��d d d�} ttt| d gg dgddggd t�d�gd dtjdggd� ttt| d gg dgddggd t�d�gd dd dggd� ttt| d gg dgddggd t�d�gd dd ddggd� d S )Nr ro rp r8 r� r" r! )r& rh r� )r' r6 r4 r5 r r< )r& rq ) r r r� r ru r( r' rB r} )r. r r r � test_iter_allocate_output_errors� s0 r� c C s� t d��ddd�} t| dg�}|�d� tdd� |D �| d d �d d d �f �� � | d d d �d d �d d �f } t| dg�}|�d � td d� |D �| d d d �d d �f �� � d S ) Nrh r6 r4 r5 r r< c S s g | ]}|�qS r r )r9 r: r r r r; � s z)test_iter_remove_axis.<locals>.<listcomp>r r7 c S s g | ]}|�qS r r )r9 r: r r r r; � s )r rB r � remove_axisr rM )r. r r r r �test_iter_remove_axis� s * r� c C s t d��ddd�} t| dg�}t|jd� t|jd� t|jd jd� dd � |D �}|�� d d � |D �}t||� t|jd � tt d d � |� t|jd jd� |� � t|j d� t|d jt � � |� � t|j d� t|d jd� t|jt d�� d S )Nrh r6 r4 r5 r )r6 r4 r5 r c S s g | ]}|�qS r r )r9 r: r r r r; � s z;test_iter_remove_multi_index_inner_loop.<locals>.<listcomp>c S s g | ]}|�qS r r )r9 r: r r r r; � s r< c S s | j S )N)rE )r r r r r� � r� z9test_iter_remove_multi_index_inner_loop.<locals>.<lambda>)rh )r rB r r re rE �itviews�remove_multi_indexr ru r� rl rA �enable_external_loopr� )r. r �before�afterr r r �'test_iter_remove_multi_index_inner_loop� s& r� c C s� d} t d��ddd�}�x�g dgfD �]�}t||| d�}tt|�ttd��� d|_tt|�ttdd��� t||d| d �}tt|�ttd��� d|_tt|�ttdd��� t|d d d � |d| d �}tt|�ttd��� d |_tt|�ttd d��� t|d d d �d d d �f |d | d �}tt|�ttd��� d |_tt|�ttd d��� t|d d d�d d d �f || d�}tt|�ttd��� d|_tt|�ttdd��� |�� d|_tt|�ttdd��� q$W d S )Nr3 rh r5 r4 r6 r� )r� rO )rL r� r7 r\ rK � r< � ) r rB r r r �listr> r r� )r� r. r� r r r r �test_iter_iterindex s4 $"r� c C s� d} t ddd��ddd�}|jdd �}t|d gd gd| d �}t|jd � tdd� |D �|� xBdD ]:}||_t|j|� tdd� |D �||d |d � � qbW t|d dgd gdd| d�}t|jd � tdd� |D �|� xBdD ]:}||_t|j|� tdd� |D �||d |d � � q�W dd� }t|d ddgd gdd| d�}t|jd � t||�|� x>dD ]6}||_t|j|� t||�||d |d � � �qbW d S ) Nr3 rh r� )r' r5 r4 r6 rO )rL r� r8 )rL r� )r rh c S s g | ] }|d �qS )r r )r9 r: r r r r; / s z'test_iter_iterrange.<locals>.<listcomp>))r rh )r< r6 )r4 rh )r3 r3 )r r� )r� rh c S s g | ] }|d �qS )r r )r9 r: r r r r; 3 s r r< r� r� )rL r& r� c S s g | ] }|d �qS )r r )r9 r: r r r r; 8 s c S s g | ] }|d �qS )r r )r9 r: r r r r; < s c S s. t jg dd�}x| D ]}t �||f�}qW |S )Nr� )r' )r( r � concatenate)r �valr: r r r � get_array> s z&test_iter_iterrange.<locals>.get_arrayrd )r rB rM r r r� )r� r. Za_fortr �rr� r r r �test_iter_iterrange% s: ( ( r� c C s4 g } | � tjddd��ddd�j�� �� � | � tjddd�� tjd d d�d d � }d |_tjd d d�|d d �< | � |� | � tjdd d��dddd�j� x�| D ]�}x�dD ]z}g }t |ddgdddggdd|d�}x8|j �st |d j |k� |� |d � � � |�� q�W tt�|�|jdd�� q�W q�W d S )Nrh r� )r' r6 r4 r5 r] r! )�A r� r< r� r� �x r3 )r< r6 r4 r3 r[ r^ r� i r� rd r8 r� r� rK r� )rL r% r� r )rL )r r( r rB rC r) r� r~ r' r r r rf r- r r r� rM )�arraysr. r� r� r r r r �test_iter_bufferingN s, " r� c C s� t �d��ddd�j�� �� } t| dgdddggd d d d �}d }x"|jsb||d < |d7 }|�� qBW t | j d d�t �d�� d S )Nrh r6 r4 r5 r� r" r� r� r� rK r� )r% rL r� r r< )rL ) r( r rB rC r) r� r r r r rM )r. r r: r r r �test_iter_write_bufferingk s r� c C s� t �d�} t jddd�}t| |gddddgd gd dd �}t|j� ttd d � |� ttdd � |� ttdd � |� dd� }tt||� |�� t|j � t|j d� t|d d� d|d< t|dd� ddg� tdd� |D �t t t d�dgd ��� d S )Nr r< r! )r' r� r� r � reduce_okr" r$ )r% r& c S s | j S )N)r )r r r r r� � r� z3test_iter_buffering_delayed_alloc.<locals>.<lambda>c S s | d S )Nr r )r r r r r� � r� c S s | dd� S )Nr r6 r )r r r r r� � r� c S s d| d< d S )Nr r )r r r r � assign_iter� s z6test_iter_buffering_delayed_alloc.<locals>.assign_iter)r r r6 c S s$ g | ] }|d d |d d g�qS )r r r< r )r9 r: r r r r; � s z5test_iter_buffering_delayed_alloc.<locals>.<listcomp>) r( r r r �has_delayed_bufallocr ru r� r r r� �zipr> )r. r� r r� r r r �!test_iter_buffering_delayed_alloc| s& r� c C sp t jddd�} t| ddgdddggd t �d �gd d �}x |D ]}|d d9 < q<W t| dt jddd� � d S )Nr] r! )r' r� rd r" r� r� r� r� r4 )r% r& r� .r6 )r( r r r' r )r. r �vr r r � test_iter_buffered_cast_simple� s r� c C s t jddd��� �� } t| ddgdddggd t �d ��� gd d �}x |D ]}|d d9 < qHW t| dt jddd� � t� ��}|�t j � t jdd d��� �� } t| ddgdddggdt �d��� gd d �}x |D ]}|d d9 < q�W t| dt jdd d� � W d Q R X d S )Nr] r! )r' r� rd r" r� r� r� r� r4 )r% r& r� .r6 r$ r� ) r( r r) r� r r' r r �filter�ComplexWarning)r. r r� �supr r r �#test_iter_buffered_cast_byteswapped� s( r c C s t jddd��� �� } | d7 } t| ddgddd ggd t �d �gd d �}x |D ]}|d d9 < qLW t| dt jddd� d � t jddd�} | d7 } t| ddgddd ggd t �d ��� gd d �}x |D ]}|d d9 < q�W t| dt jddd� d � t jdt jd��� �� } | d7 } t| ddgddd ggd t �d �gd d �}x |D ]}|d d9 < �qJW t| dt jdt jd� d � t jdt jd��� �� } t| ddgddd ggd t �d�gdd �}x |D ]}|d d9 < �q�W t| dt jdt jd� � d S )Nr] r� )r' y @r� rd r" r� r� r� r� r4 )r% r& r� .r6 y @r! rZ ) r( r r) r� r r' r � clongdouble� longdouble)r. r r� r r r �+test_iter_buffered_cast_byteswapped_complex� sN r c C s ddddg} t jddd�d }t|d d gd gd | d �}dd� |D �}t|d d d� t|d d d� t|d d dgd gd � t|d d d� t|d d d� t|d d d� t|d d dgd gd � t|d d d� t|d jt �| �� ddddg} t jddd�}dddddgdddggdf|d< dddddgdddggdf|d< dddddgdddggdf|d< t�r�t�|d �}t|d d gd gd | d �}dd� |D �}t|d d d� t|d d d� t|d d dgd gd � t|d d d� t|d d d� t|d d d� t|d d dgd gd � t|d d d� t|d jt �| �� d gd \}}}t�r�tt�|d �|� dddg} t j d d g| d�}t|d d gd gd d d �}td d� |D �d d!g� dddg}d"d#d$g}t j d%d&g|d�}t t �� t|d d gd gd |d �}W d Q R X t|d jt �|�� td'd� |D �t j d(|d�t j d)|d�g� dddg}d*d+g}t j d%d&g|d�}t t �� t|d d gd,gd |d �}W d Q R X t|d jt �|�� g }x.|D ]&}|� t � |�� |d d |d< �q�W t|t j d-|d�t j d.|d�g� t|t j d/d0g|d�� dddd1d2gfg}d*d+g}t j d3d4g|d�}t t �� t|d d gd,gd |d �}W d Q R X t|d jt �|�� g }x.|D ]&}|� t � |�� |d d |d< �q�W t|t j d-|d�t j d.|d�g� t|t j d5d6g|d�� dddd1d*gfg}d*d+g}t j d3d4g|d�}t t �� t|d d gd,gd |d �}W d Q R X t|d jt �|�� g }x.|D ]&}|� t � |�� |d d |d< �q�W t|t j d-|d�t j d.|d�g� t|t j d7d8g|d�� d*d+g}dddd1d*gfg}t j d9d:g|d�}t t �� t|d d gd,gd |d �}W d Q R X t|d jt �|�� g }xJ|D ]B}|� t � |�� t|d t j d;d1d*gd�� |d d |d< �q�W t|t j d<|d�t j d=|d�g� t|t j d>d?g|d�� d S )@N)r. r! )r� r� )r� r� )r6 r4 )r� r� r4 r! )r' g �?r� r� r8 r$ )r% r& c S s g | ]}t �|��qS r )r( r )r9 r: r r r r; � s z;test_iter_buffered_cast_structured_type.<locals>.<listcomp>r r. r� r� r6 r� r< g �?)r4 r� g @c S s g | ] }|� � �qS r )r- )r9 r: r r r r; s )g @rZ �test)r[ r] r^ r� c S s g | ] }|d �qS )r r )r9 r� r r r r; ! s r3 r[ )r� �u2)r. r� )r� r� )r< r6 r4 )r5 r3 r c S s g | ]}t �|��qS r )r( r )r9 r� r r r r; - s )r4 r< r6 )r r5 r3 )r� r� )r. r� r" )r6 r< )r3 r5 )r3 r6 N)r[ r3 N)r. rn )r� r� )r< r6 )r r\ )r5 r3 )r� � )r3 r6 )r r )r[ r3 )r r )r3 r6 )r N)r[ r3 )r N)r< r6 )r5 r3 )r N)r6 r< )r N)r3 r5 )r N)r< r5 )r5 rZ ) r( r r r r' r~ r r* r+ r r � FutureWarningr )�sdtr. r r� r� r: �sdt1�sdt2r r r �'test_iter_buffered_cast_structured_type� s� r c C s� dg} dg}t jd| d�}t �d�|d< t|ddgd gd |d �}t|d jt �|�� x4t|ttd���D ] \}}t t � |d |k�� qjW d g} dg}t jd| d�}t �d�|d d d �d d f< t|ddgdgd |d �}t|d jt �|�� d }x>|D ]6}t t � |d |k�� |d d d7 < |d7 }q�W t|d t �d�� ddd�d � dg} dg}t jd| d�}t �d�|d d d �d d d f< t|ddgdgd |d �}t|d jt �|�� d }x4|D ],}t|d |� |d d7 < |d7 }�q�W t|d t �d�� dddd�t � d� d � dg} dg}t jd| d�}t �d�|d d d �d d d f< t|ddgd gd |d �}t|d jt �|�� d }x$|D ] }t|d |� |d7 }�q�W dg} dg}t jd| d�}t �d�|d d d �d d d f< t|ddgd gd |d �}t|d jt �|�� d }x$|D ] }t|d |� |d7 }�q$W dg} dg}t jd| d�}t �d�� dddd�|d< t|ddgd gd |d �}t|d jt �|�� d }x,|D ]$}t|d || d � |d7 }�q�W dg} dg}t jd| d�}t �d�� dd�|d< t|ddgd gd |d �}t|d jt �|�� d }x4|D ],}t|d || d d d� � |d7 }�q@W dg} d g}t jd| d�}t �d �� dd�|d< t|ddgd gd |d �}t|d jt �|�� d }xR|D ]J}t|d d d� || d � t|d dd � d d d d g� |d7 }�q�W dg} d g}t jd| d�}t �d �� dd�|d< t|ddgd gd |d �}t|d jt �|�� d }xJ|D ]B}t|d d || d � t|d d || d � |d7 }�q�W d g} d g}t jd| d�}t �d �� ddd�|d< t|ddgd gd |d �}t|d jt �|�� d }x�|D ]�}t|d d d�d f || d d d �d f � t|d d d�df || d d d �d f � t|d dd d �f d d g� |d7 }�q>W d!g} d g}t jd| d�}t �d�� ddd�|d< t|ddgd gd |d �}t|d jt �|�� d }x�|D ]�}t|d d d�d f || d d d �d f � t|d d d�df || d d d �df � t|d dd d �f d d g� |d7 }�q4W d S )"N)r. r! )r. r� )r4 r6 r6 )r )r' r r. r� r� r8 r$ )r% r& r )r. r� )r< r< )r. r� )r4 r6 r6 r" r6 r< )r. r� )r< )r< r4 r6 r6 )r. r! )r< )r. r! )r4 r6 r6 �H r4 )r. r� )r )r. r! )r6 rx )r. r� )r6 )r. r! )r rU )r. r! )r6 r6 )r. r� )r6 r< )r. r! )r4 r6 )r. r� )r6 r4 ) r( r~ r r r r' r� r� r> r r rB rv )r r r. r r: �countr r r � test_iter_buffered_cast_subarrayz s " . .. .. r c C s t �d��ddd�} t �d��ddd�}ttt| |gddgdgd ggd d � t| |gddgd gd ggd d � t �d��ddd�} t| |gddd gdgd ggd d � t �d��ddd�} ttt| |gddgdgd ggd d � t �d��ddd�} ttt| |gddgdgd ggd d � d S )Nr r6 r4 r< rU r� rd r" ro rK )rL r8 r� r5 )r( r rB r ru r )r. r� r r r � test_iter_buffering_badwriteback s. r c C s� t jdddgt jd�} t| jt �d�� ttt| dgdgdd � t| dgdgd d �}t|d d � t|d jt �d �� t jdddgt jd�} t| jt �d �� ttt| dgdgdd � t| dgdgdd �}t|d d� t|d jt �d�� d S )N�abcr. �abcd)r' �S4r� r8 �S2)r& ZS6r s abc�U4�U2ZU6) r( r �bytes_r r' r r� r �unicode)r. r r r r �test_iter_buffering_stringA s r c C s4 t �d�} t| dddgdd�}t|d j| j� d S )N� r� Z growinnerrd r3 )r� r )r( r r r rf )r. r r r r � test_iter_buffering_growinnerS s r c sR t �d�dd� � dddddg} d d g}d d gd d gg�t j� jg}� �fdd�}x�|� D ]�\}}}t j|�� d g|| ||d�}d|jd d< |�� ||_x |D ]\}} | |� t j �7 } q�W |jd } x�t dd�D ]r} t j|d g|| || |d�} d| jd d< | �� || _x"| D ]\} }|| � t j �7 }�qW | jd }t || � q�W qZW d S )Ni� �� �� r� r� r r� r� )r8 )r" rp )r r< r6 )r r< r7 )r r7 r7 c 3 s~ xxt dd�D ]j} xdt | d�D ]V}xP�D ]H}| � j |� j � jf}tjj�� d|�}xdD ]}|||fV qZW q&W q W q W d S )Ni����r] )r4 r4 r4 )r r< )r> �itemsizer( �lib� stride_tricks� as_strided)�xsZysrq rt �arr�skip)r. � op_axes_listr r � get_paramsf s z3test_iter_buffered_reduce_reuse.<locals>.get_params)rq r� �op_flagsr& r r7 .� )rq r� r( r� r& ) r( r �floatr' r r- rs r� r �astype�int_r> r )r� r( r& r' r$ rq r% Znditer2Za2_inZb2_inZcomp_res�bufsizeZnditer1Za1_inZb1_in�resr )r. r& r � test_iter_buffered_reduce_reuse\ s6 r/ c C s� t �d��ddd�} t �d��ddd�}t �d��dd�}t| ||gg dd gdgdgg� ttt| ||gg dgdd gdgg� ttt| ||gg dgdgdd gg� d S ) Nrh r6 r4 r5 r r< rU r8 ry )r( r rB r r ru )r. r� r� r r r �test_iter_no_broadcast� s r0 c C sV t d��ddd�} t�| dgddgg�\}}g }x |D ]}|�dd� |D �� q6W t|dddddd gd d d d ddgg� t�| ddgdgg�\}}g }x |D ]}|�dd� |D �� q�W t|ddgddgdd gd d gd d gddgg� t�| ddgdgg�\}}g }x"|D ]}|�dd� |D �� �q W t|dddgddd gd d dgd d dgg� d S )NrU r6 r4 r r< c S s g | ]}|�qS r r )r9 r� r r r r; � s z0test_iter_nested_iters_basic.<locals>.<listcomp>r5 r3 r rZ r[ r\ r] r^ c S s g | ]}|�qS r r )r9 r� r r r r; � s c S s g | ]}|�qS r r )r9 r� r r r r; � s )r rB r( � nested_itersr r )r. r r� r� r: r r r � test_iter_nested_iters_basic� s & . r2 c C s� t d��ddd�} t�| dgddgg�\}}g }x |D ]}|�dd� |D �� q6W t|dddddd gd d d d ddgg� t�| ddgdgg�\}}g }x |D ]}|�dd� |D �� q�W t|ddgddgdd gd d gd d gddgg� t�| ddgdgg�\}}g }x"|D ]}|�dd� |D �� �q W t|dddgddd gd d dgd d dgg� tj| dgddggdd�\}}g }x"|D ]}|�dd� |D �� �qzW t|dddddd gd d dd d dgg� tj| ddgdggdd�\}}g }x"|D ]}|�dd� |D �� �q�W t|ddgd d gddgd d gdd gddgg� tj| ddgdggdd�\}}g }x"|D ]}|�dd� |D �� �qZW t|dddgd d dgddd gd d dgg� d S )NrU r6 r4 r r< c S s g | ]}|�qS r r )r9 r� r r r r; � s z2test_iter_nested_iters_reorder.<locals>.<listcomp>r5 r3 r rZ r[ r\ r] r^ c S s g | ]}|�qS r r )r9 r� r r r r; � s c S s g | ]}|�qS r r )r9 r� r r r r; � s rK )rL c S s g | ]}|�qS r r )r9 r� r r r r; � s c S s g | ]}|�qS r r )r9 r� r r r r; � s c S s g | ]}|�qS r r )r9 r� r r r r; � s )r rB r( r1 r r )r. r r� r� r: r r r � test_iter_nested_iters_reorder� s> & . * & . r3 c C s� t d��ddd�d d d�d d d�d d d�f } t�| dgddgg�\}}g }x |D ]}|�dd� |D �� qRW t|ddddd d gd d d dddgg� t�| ddgdgg�\}}g }x |D ]}|�dd� |D �� q�W t|ddgddgd d gd d gd dgddgg� t�| ddgdgg�\}}g }x"|D ]}|�dd� |D �� �q&W t|ddd gddd gd d dgd ddgg� tj| dgddggdd�\}}g }x"|D ]}|�dd� |D �� �q�W t|dddd d d gd d ddddgg� tj| ddgdggdd�\}}g }x"|D ]}|�dd� |D �� �qW t|ddgdd gd d gd d gddgddgg� tj| ddgdggdd�\}}g }x"|D ]}|�dd� |D �� �qvW t|ddd gdd d gd ddgd ddgg� d S )NrU r6 r4 r7 r r< c S s g | ]}|�qS r r )r9 r� r r r r; � s z4test_iter_nested_iters_flip_axes.<locals>.<listcomp>r5 r3 r rZ r[ r\ r] r^ c S s g | ]}|�qS r r )r9 r� r r r r; � s c S s g | ]}|�qS r r )r9 r� r r r r; � s rK )rL c S s g | ]}|�qS r r )r9 r� r r r r; � s c S s g | ]}|�qS r r )r9 r� r r r r; � s c S s g | ]}|�qS r r )r9 r� r r r r; s )r rB r( r1 r r )r. r r� r� r: r r r � test_iter_nested_iters_flip_axes� s> . & . * & . r4 c C s t d��dd�} t d��dd�}t�| |gdgdgg�\}}g }x |D ]}|�dd� |D �� qFW t|ddgddgddggddgddgddggg� t�| |gdgdgg�\}}g }x |D ]}|�dd� |D �� q�W t|ddgddggddgddggddgddggg� d S )Nr6 r< r4 r c S s g | ]}|�qS r r )r9 r� r r r r; s z4test_iter_nested_iters_broadcast.<locals>.<listcomp>c S s g | ]}|�qS r r )r9 r� r r r r; s )r rB r( r1 r r )r. r� r r� r� r: r r r � test_iter_nested_iters_broadcast s 2 r5 c C sN t ddd��dd�} tj| dgdggdd gd d �\}}t|d jt�d �� g }x |D ]}|�d d � |D �� qVW t|dddgdddgg� d }t ddd��dd�} tj| dgdggddgdd d�\}}t|d jt�d �� x*|D ]"}x |D ]}|d d7 < q�W q�W t| dddgdddgg� d\}}}}t| dddgdddgg� d S )Nr r� )r' r6 r4 r r< r8 r- r� )r( r& c S s g | ]}|�qS r r )r9 r� r r r r; " s z5test_iter_nested_iters_dtype_copy.<locals>.<listcomp>r5 r3 r! r" r# r� )r( r% r& .)NNNN)r rB r( r1 r r' r )r. r r� r� r: r� r r r �!test_iter_nested_iters_dtype_copy s, r6 c C s� t ddd��dd�} tj| dgdggdgd gd d d �\}}t|d jt�d �� x*|D ]"}x |D ]}|d d7 < q`W qVW t| dddgdddgg� d S )Nr r! )r' r6 r4 r r< r� r" r� r� )r� r( r% r& .r5 r3 )r rB r( r1 r r' )r. r r� r: r� r r r �%test_iter_nested_iters_dtype_buffered4 s r7 c C sz t �d�} ttt| d gg dgddggdgdggd� t �d��dd �} ttt| d gd gdgddggdd gddggd� d S ) Nr r8 r" rp r r7 )rq r6 r4 rd r< )r( r r ru r rB )r. r r r �test_iter_reduction_errorC s r8 c C sn t �d�} t| d gdgdgddggdgdggd�}d|jd d < x |D ]\}}|d |7 < qFW t|jd jd� t|jd t �| �� t �d��d d �} t| d gdd gdgddggdd gddggd�}d|jd d < t|d jd� t|d j d� x<|D ]4\}}x*t t |��D ]}|| || 7 < �qW q�W t|jd jd� t|jd t �| �� t � d�} t| d gdd gdgddggd ddd ggd�}t| d gdd ddgdgddggd ddd ggdd�}|jd � d� |jd � d� |�� x&|D ] }|d d |d 7 < �q�W x&|D ] }|d d |d 7 < �q W t|jd |jd � t|jd �� | j� d S )Nr r� r8 r" rp r r7 )rq r< .r6 r4 rd )r )r )r6 r4 r3 r� r� r] )rq r� )r( r r rs r re �sumrB rE rt r> r? rv �fillr� rf )r. r r: r� r� Zit1r1 r r r �test_iter_reductionO sL r; c C s� t �d�} t jddd��� �� }t| |gddgdgdd ggd gd ggd �}t|d jt �d�� t|d j|jk� x |D ]\}}|d |7 < qzW t|t � | �� t �d�� dd�} t jd d gdd��� �� }t| |gdddgdgdd ggd d gd d ggd �}t|d j d� t|d j d� x>|D ]6\}}x*t t|��D ]}|| || 7 < �q>W �q(W t|t j | d d�� t �d�d }t j|d gddddgdgddggd d gd d ggdd�}|jd �d � |�� t|d d dd dg� t �dt j�dd�d dd�d d �f �d dd �}t �|j�� |j �|d< t jd t jd�� dd�}|�� } |d d d�d d �d f }t �||gdddgdgdgg�}x |D ]\} }| �d� �q�W t|d d d� | d d d� � t|d d d� d� d S ) Nr g r� )r' r� r� r8 r" r� r r7 )rq r< .r6 r4 rd )r4 )r )�axisr� rp )r6 r6 )rq rr )rZ r� r[ r5 r^ r3 r� )r( r r r� r) r r r' r r9 rB rE rt r> r? rs r: r� rv r� � transposerf r- ) r. r� r r: r� r� r� r0 Zy_baseZ y_base_copyr r r � test_iter_buffering_reduction� sT 0 r> c C s� t �d�} t �d�}t j| |gdddgdgdggdd �}g }x |D ]\}}|�|jd � qBW t|dd dd g� tt|�| j� d S ) N)r6 rZ )r< rZ r� rd r� r8 r" r3 )r� r( r� r r6 )r( r~ r r rE r r9 rf )r. r� r0 Zbufsizesr: r� r r r �0test_iter_buffering_reduction_reuse_reduce_loops� s r? c C s t �d�} t �d�}t �dddgdddgg�}t �dddg�}t jdddgdd�}t jdddgd d�}t jdddgd d�}ttt| |gg d d gd gg� ttt| |gg d d gd dgg� ttt| |gg d gd dd gg� ttt| ||gg d d gd dgd dgg� ttt| |gg d gd dgg� ttt| ||gdgd gd d gd dgg� t �| ||gdgd gd d gd dgg� ttt| ||gdgd gd d gd dgg� t j| |gdgd d gd dggd d gdd� ttt j| |gdgd d gd dggd d gdd� ttt j| |gdgd d gd dggd d gdd� d S )N)r6 r4 )r4 TFr r< �u1)r' r� r! r" � writemaskedr8 � arraymaskr� r� r� )r& r% )r( r~ r r ru r r� )r. r� �m�m2�m3Zmbad1Zmbad2r r r � test_iter_writemasked_badinput� s\ rF c C s t jddd�} t �dddg�}t �| |gg ddgdd gg�}x|D ]\}}d |d < qBW t| d d d g� t �| |gd gddgdd gg�}x|D ]\}}d |d < q�W t| d d d g� t j| |gd gddgdd ggdd gdd�}x|D ]\}}d|d < q�W t| ddd g� d S )N)r4 r� )r' TFr" rA r8 rB r< .r� g @r� r$ )r& r% r4 )r( r~ r r r )r. �mskr0 r: rC r r r �test_iter_writemasked s, rH c C sN t �t �d��} dddddddd d d d d dg }x|D ]}ttt| |� q4W d S )Nr6 r� rE rs r� r� r, �has_multi_index� has_indexr� re �noprl r )r( r rv r �AttributeError�delattr)r0 �attr�sr r r �)test_iter_non_writable_attribute_deletion< s rP c C s< t �t �d��} ddddg}x|D ]}ttt| |� q"W d S )Nr6 r r r� r )r( r rv r rL rM )r0 rN rO r r r �%test_iter_writable_attribute_deletionF s rQ c C sL t �t �d��} y| d= | dd�= W n" tk r8 Y n t�Y nX d S )Nr4 r< r6 )r( r rv r� rz )r0 r r r �test_iter_element_deletionM s rR c C s� t jdddgd fd dgd�} x(| D ] \}}|d |d< |d |d< q"W t| jd ddgddgd d gg� t jd d d d gd�} xB| D ]:\}}}|| |d < || |d< || |d< || |d< q�W t| jd ddgddgg� d S )Nr< r4 r� )r� )r6 )r& r r6 r5 � r )r] r6 N)r� )r6 r6 )r r )r r< )r< r )r< r< r[ rU r3 )r( r r rs )r0 r. r� r� r r r � test_iter_allocated_array_dtypesW s " rT c C s4 t ddgdgdggd �} t| jd� tt| �d� t| jd� t| jd� ttt| � | �� tt| �d� ttt| � t t � d�dgdggdgd �} t| jd� tt | �d � d d d dg}t j ddd�}t |ddgdgd|d�} t| �}t|d d� t|d d� t|d dgd gd � t|d d� d S )Nr6 r4 r r8 r )r6 r4 r r3 )rq r< )r. r! )r� r� )r� r� )r6 r4 )r� r� g �?r! )r' r� r� r$ )r% r& r. r� r� r� ) r r re r� r r r � StopIterationr� r( r r? r )r r r. r� r r r � test_0d_iterj s* rV c C sv t �d��ddd�} t �| g dddgg�\}}g }x |D ]}|�dd� |D �� q8W t|dddddd d d d d ddg g� t �| dddgg g�\}}g }x |D ]}|�dd� |D �� q�W t|dgdgdgdgdgd gd gd gd gd gdgdgg � t �| ddgg dgg�\}}}g }x2|D ]*}x"|D ]}|�dd� |D �� �q$W �qW t|dddgddd gd d dgd d dgg� d S )NrU r6 r4 r< r c S s g | ]}|�qS r r )r9 r� r r r r; � s z'test_0d_nested_iter.<locals>.<listcomp>r5 r3 r rZ r[ r\ r] r^ c S s g | ]}|�qS r r )r9 r� r r r r; � s c S s g | ]}|�qS r r )r9 �zr r r r; � s )r( r rB r1 r r )r. r r� r� r: �kr� r r r �test_0d_nested_iter� s" $ : rY c C sn t �t j�jd } t jj�t �d�| fd�}tt t ||d d �d f f� tt t ||d d �d f fdgd� d S )Ni r< )r r )r� ) r( �iinfo�intp�maxr r! r" r~ r ru r )rf r$ r r r �test_iter_too_large� s r] c C s� d} d}x"| | t �t j�jk r*|d7 }q W ddg| }g }x8t|�D ],}|d d � }d||d < |�t �|�� qFW t|�}x td�D ]}tt t |d|� q�W t |dd� xJt|�D ]>}x8td�D ],}t ||d |� tt t ||d d |� q�W q�W d S )Ni r< r6 r r7 rZ ) r( rZ r[ r\ r> r �emptyrA r ru r )Z base_size�numZshape_templater� r rE �moder r r �#test_iter_too_large_with_multiindex� s& ra �__main__)n� __future__r r r r* �warnings�numpyr( r r r r �numpy.core.multiarray_testsr Z numpy.testingr r r r r r r r r r r r �skipifr2 rJ rN rP rR rS rT rV rX rY r_ ra rb rc rg ri rj rm rw r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r �slowr/ r0 r2 r3 r4 r5 r6 r7 r8 r; r> r? rF rH rP rQ rR rT rV rY r] ra �__name__r r r r �<module> s� ,* , % % $F KM ;?,&&)@X&. # ") . &" 2** 8<B& "