Edit File: test_stride_tricks.cpython-37.pyc
B ��Fd�: � @ s$ d dl mZmZmZ d dlZd dlmZ d dlm Z m Z m Z m Z m Z d dlmZmZmZmZ dd� Zdd � Zd0d d �Zd d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Z dd � Z d d � Z d d � Z d!d"� Z d#d$� Z!G d%d&� d&ej"�Z#G d'd(� d(e#�Z$d)d*� Z%d+d,� Z&d-d.� Z'e(d/k�r e � dS )1� )�division�absolute_import�print_functionN)�rational)�run_module_suite� assert_equal�assert_array_equal� assert_raises�assert_)� as_strided�broadcast_arrays�_broadcast_shape� broadcast_toc C s@ dd� | D �}t |� }dd� |D �}|gt|� }t||� d S )Nc S s g | ]}t �|��qS � )�np�zeros)�.0�sr r ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/lib/tests/test_stride_tricks.py� <listcomp> s z)assert_shapes_correct.<locals>.<listcomp>c S s g | ] }|j �qS r )�shape)r �ar r r r s )r �lenr )� input_shapes�expected_shape�inarraysZ outarraysZ outshapes�expectedr r r �assert_shapes_correct s r c C s" dd� | D �}t ttf|�� d S )Nc S s g | ]}t �|��qS r )r r )r r r r r r s z4assert_incompatible_shapes_raise.<locals>.<listcomp>)r � ValueErrorr )r r r r r � assert_incompatible_shapes_raise s r Fc C s� t j| td�}tt j�|��}t �|��|�}|r>|j}|j}|r^|d d d� }|d d d� }|| }t||�\}} t || � d S )N)�dtype���) r r �int�multiply�reduce�arange�reshape�Tr r ) Zshape0Zshape1Z transposedZflipped�x0�n�x1�y�b0�b1r r r �assert_same_as_ufunc s r. c C s: t �d�} t �d�}t| |�\}}t| |� t||� d S )N� )r r% r r )�xr+ �bx�byr r r � test_same6 s r3 c C sv t �dddgg�} t �dgdgdgg�}t| |�\}}t �dddgdddgdddgg�}|j}t||� t||� d S )N� � � )r �arrayr r' r )r0 r+ r1 r2 Zbx0Zby0r r r � test_one_off> s " r8 c C s` ddddddddd d g } xB| D ]:}|g}t ||� ||g}t ||� |||g}t ||� q W d S ) Nr )r4 )r6 )r r4 )r r6 )r4 r )r6 r )r4 r6 )r6 r4 )r6 r6 )r )�datar r Z input_shapes2Z input_shapes3r r r �test_same_input_shapesH s" r: c C s� ddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgd dgdgdd gdgddgdgddgdgdd gd gg } x.| D ]&\}}t ||� t |d d d � |� q�W d S ) N)r4 )r6 )r4 r6 )r6 r6 )r6 r4 )r4 r4 )r4 r )r r )r r4 r! )r )r9 r r r r r �(test_two_compatible_by_ones_input_shapesc s r; c C s ddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgdd gd gdd gd gdd gd gdd gd gddgd gdd gd gdd gd gdd gd gdd gd gg} x.| D ]&\}}t ||� t |d d d � |� q�W d S ) Nr )r6 )r6 r6 )r6 r4 )r4 )r4 r4 )r4 r6 )r )r r )r r4 )r4 r r! )r )r9 r r r r r �3test_two_compatible_by_prepending_ones_input_shapes| s. r< c C sJ ddgddgdddgddgg} x&| D ] }t |� t |d d d� � q$W d S )N)r6 )� )r5 r6 )r5 )r4 r6 r= )r5 r6 r6 r! )r )r9 r r r r �)test_incompatible_shapes_raise_valueerror� s r> c " C s� ddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgd dgdgdd gdgddgdgddgdgdd gd gd dgdgddgdgddgdgddgdgd dgdgddgdgddgdgddgdgd dgdgd dgdgd d gd gd dgdgd d gdgddgdgd dgdgdd gdgdd gd gddgdgd dgdgd d gd gg!} x�| D ]�\}}t |d |d d|d |d f � t |d |d � t |d |d d� d |k�rTt |d |d dd� t |d |d dd� �qTW d S )N)r4 )r6 )r4 r6 )r6 r6 )r6 r4 )r4 r4 )r4 r )r r )r r4 r )r r r4 z Shapes: %s %sTF)r. )r9 r r r r r �test_same_as_ufunc� sR r? c C sv t �d�dt �d�gt �d�dt �d�gt �d�dt �d�gt �d�dt �d�gt �d�dt �d�gt �d�d t �d �gt �d�dt �d�gt �d�d t �d��dd �gt �d�d t �dddgdddgg�gt �d�dt �d�gt �d�dt �d�gt �d�dt �d�gt �d�dt �d�gt �d �dt �d�gt �d�dt �d�gg} x(| D ] \}}}t||�}t||� �qNW d S )Nr )r )r4 r4 )r6 r6 )r5 r5 )r4 r5 r6 )r4 r6 r! )r5 r6 )r4 r5 )r r5 )r5 r4 )r5 r )r r7 r �onesr% r& r r )r9 � input_arrayr r �actualr r r �test_broadcast_to_succeeds� s$ $ rC c sz ddgddgddgddgddgddgddgd dgdd gdd gdd gg } x.| D ]&\}�t �|�� tt� �fd d�� qLW d S )N)r r )r4 )r6 )r5 )r= )r4 r5 )r5 r4 )r4 r4 r! )r! )r! r5 c s t � ��S )N)r r )�arr� target_shaper r �<lambda> � z*test_broadcast_to_raises.<locals>.<lambda>)r r r r )r9 � orig_shaper )rD rE r �test_broadcast_to_raises� s rI c s� t t� d� t tddg�d� t tt�d��d� t tt�d�t�d��d� t tt�d�gd � d� t tt�d�gd � d� t tt�d�gd dg � d� t�d�gd t�d �gd � tt� fd d �� d S ) Nr r4 r5 )r5 )r4 r4 )r6 r= )r4 r5 � �d r6 c s t � � S )N)r r )�bad_argsr r rF rG z&test_broadcast_shape.<locals>.<lambda>)r r r r@ r r r r )rL r �test_broadcast_shape s rM c C s" t �d g�} t| �}t �d g�}t|t �d g�� t �ddddg�} t| dd| j fd�}t �ddg�}t||� t �ddddg�} t| ddd| j fd�}t �ddddgddddgddddgg�}t||� t �d d g�}t jd |d �} t �dd �| d< t| dd| jfd�}ddddggd }d gd gd }t|j|� t||d � t||d � t jd dd �} t| dd| jfd�}t| j|j� t �dgdgd��}t jd |d �} t| dd| jfd�}t| j|j� dd� t d�D �}t j|t d �} t| dd| jfd�}t| j|j� t|gd |� d S )Nr4 r5 r6 r= )r5 )r �strides)r6 r= r )�num�i4)�obj�O)r= )r � rO rQ �V4� )�names�formatsc S s g | ] }t |��qS r )r )r �ir r r r B s z#test_as_strided.<locals>.<listcomp>) r r7 r r �itemsizer �emptyr% r �ranger )r �a_viewr �dtZ expected_numZ expected_obj�rr r r �test_as_strided sB ( r_ c C s� t �d�} t| dd�}t|jj � t| dd�}t|jj� d|d< t| t �| d�� d| j_t| dd�}t| dd�}t|jj � d S )Nr/ F)� writeableTr6 .)r r@ r r �flagsr` r � full_like)rD �viewr r r �as_strided_writeableH s rd c @ s e Zd Zdd� ZdS )�VerySimpleSubClassc O s d|d< t j||��| �S )NT�subok)r r7 rc )�cls�args�kwargsr r r �__new__[ s zVerySimpleSubClass.__new__N)�__name__� __module__� __qualname__rj r r r r re Z s re c @ s e Zd Zdd� Zdd� ZdS )�SimpleSubClassc O s$ d|d< t j||��| �}d|_|S )NTrf �simple)r r7 rc �info)rg rh ri �selfr r r rj a s zSimpleSubClass.__new__c C s t |dd�d | _d S )Nrp rU z finalized)�getattrrp )rq rQ r r r �__array_finalize__g s z!SimpleSubClass.__array_finalize__N)rk rl rm rj rs r r r r rn ` s rn c C s� t ddddg�} tt| �t k� t| dd| j fd�}tt|�tjk� t| dd| j fdd�}tt|�t k� tddddg�} t| dd| j fdd�}tt|�tk� t|jd k� t� t | ��� d d�}t | |�\}}tt|�tjk� tt|�tjk� t|j |j k� t | |dd �\}}tt|�tk� t|jd k� tt|�tjk� t|j |j k� d }t| |�}tt|�tjk� t|j |k� t| |dd �}tt|�tk� t|jd k� t|j |k� d S ) Nr4 r5 r6 r= )r5 )r rN T)r rN rf zsimple finalizedr! )rf )r5 r= )re r �typer rY r �ndarrayrn rp r% r r& r r r )r r\ �bZb_viewr r r r �test_subclassesk s8 rw c C s� t �dddg�} t| d�}t|jjd� tt|jt d �d� x6t | �t d| �gD ] }x|D ]}t|jjd� qZW qPW d| j_t d| �\}}t|jjd� d}dg}t t �d�||�}t � d �}t ||�\}} t |j| jk� d S ) Nr4 r5 r6 )r5 r6 Fr T)r5 )r4 )r r7 r r ra r` r r � __setitem__�slicer r r r r ) �original�result�results�_r rN Z tricky_array�other�first�secondr r r �test_writeable� s" r� c C sX t jdtd�} t jdgd td�}t| d�}t||� t| t �d��\}}t||� d S )Nr )r r6 )r6 )r r7 �objectr r r r@ )rA r rB r} r r r �test_reference_types� s r� �__main__)FF))� __future__r r r �numpyr �numpy.core.test_rationalr Z numpy.testingr r r r r �numpy.lib.stride_tricksr r r r r r r. r3 r8 r: r; r< r> r? rC rI rM r_ rd ru re rn rw r� r� rk r r r r �<module> s4 4- %