관리-도구
편집 파일: test_multiarray.cpython-37.pyc
B ��Fd� � @ s d dl mZmZmZ d dlZd dlZd dlZd dlZd dlZd dl Z d dl Z d dl Z d dl Z d dl Z d dlZd dlZd dlmZ ejd dkr�d dlZnd dlZd dlmZ d dlZd dlmZmZ d dl m Z d dl m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z% d d l&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8 d d l9m:Z:m9Z9 ejdd � d k�rnd Z;ndZ;e<ddfdd�Z=G dd� de'�Z>G dd� de'�Z?G dd� de'�Z@G dd� de'�ZAG dd� de'�ZBG dd � d e'�ZCG d d � d e'�ZDG d d � d e'�ZEG d!d"� d"e'�ZFG d#d$� d$e'�ZGG d%d&� d&e'�ZHG d'd(� d(e'�ZIG d)d*� d*eJ�ZKG d+d,� d,eJ�ZLG d-d.� d.e'�ZMG d/d0� d0e'�ZNG d1d2� d2e'�ZOG d3d4� d4e'�ZPG d5d6� d6e'�ZQG d7d8� d8e'�ZRG d9d:� d:e'�ZSG d;d<� d<e'�ZTG d=d>� d>e'�ZUG d?d@� d@e'�ZVG dAdB� dBe'�ZWG dCdD� dDe'�ZXG dEdF� dFeJ�ZYG dGdH� dHeJ�ZZG dIdJ� dJe'�Z[G dKdL� dLe'�Z\G dMdN� dNeJ�Z]G dOdP� dPe'�Z^G dQdR� dRe'�Z_G dSdT� dTe'�Z`G dUdV� dVe'�ZadWdX� ZbdYdZ� Zcd[d\� ZdG d]d^� d^e'�ZeG d_d`� d`e'�ZfG dadb� dbe'�ZgG dcdd� dd�ZhG dedf� dfehe'�Ziejdd � dgk�rHG dhdi� diehe'�Zjdjdk� ZkG dldm� dme'�ZlG dndo� doe'�ZmG dpdq� dqe'�ZnG drds� dse'�ZoG dtdu� due'�Zpd dvd ddwdx�ZqG dydz� dze'�ZrG d{d|� d|e'�ZsG d}d~� d~eJ�ZtG dd�� d�eJ�Zud d�lvmwZw G d�d�� d�eJ�ZxG d�d�� d�eJ�ZyG d�d�� d�eJ�Zzd�d�� Z{d�d�� Z|d�d�� Z}d�d�� Z~G d�d�� d�e'�ZG d�d�� d�e'�Z�G d�d�� d�e'�Z�G d�d�� d�e'�Z�G d�d�� d�e'�Z�e1�s�G d�d�� d�e'�Z�G d�d�� d�e'�Z�G d�d�� d�e'�Z�G d�d�� d�e'�Z�G d�d�� d�e'�Z�G d�d�� d�e'�Z�d�d�� Z�d�d�� Z�e�d�k�re(� dS )�� )�division�absolute_import�print_functionN)�contextmanager� )�Decimal)�strchar�unicode)�in_foreign_locale)�test_neighborhood_iterator� test_neighborhood_iterator_oob�!test_pydatamem_seteventhook_start� test_pydatamem_seteventhook_end�test_inplace_increment�get_buffer_info�test_as_c_array)�TestCase�run_module_suite�assert_� assert_raises� assert_warns� assert_equal�assert_almost_equal�assert_array_equal�assert_array_almost_equal�assert_allclose�IS_PYPY� HAS_REFCOUNT�assert_array_less� runstring�dec�SkipTest�temppath�suppress_warnings)� timedelta�datetime� )r r& � �Cc C s� t �|�}|t �t�kr8|dk r(td��t j| ||d�S |dkrF|j}t| d�sV| f} t�t j | �|j }t � || d t j �}|jd d | }|dkr�|| }|||| d � dd� }t j| |||d �}|�d� |S ) z+Allocate a new ndarray with aligned memory.Nz$object array alignment not supported)�dtype�order�__len__� �datar ���)r* )�npr) �object� ValueError�zeros� alignment�hasattr� functools�reduce�operator�mul�itemsize�empty�uint8�__array_interface__�ndarray�fill)�shaper) r* �align�size�buf�offsetr- r' 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_multiarray.py�_aligned_zeros3 s$ rE c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TestFlagsc C s t �d�| _d S )N� )r/ �arange�a)�selfr' r' rD �setUpN s zTestFlags.setUpc C sR t � }d| jj_| �ttd|� | �ttd|� d| jj_d| jd< d| jd< d S )NFz self.a[0] = 3zself.a[0:1].itemset(3)T� r )�localsrI �flags� writeable� assertRaisesr1 r )rJ Zmydictr' r' rD �test_writeableQ s zTestFlags.test_writeablec C s� t | jjjd� t | jjjd� t | jjjd� t | jjjd� t | jjjd� t | jjjd� t | jjj d� t | jjj d� t | jjj d� d S )NTF) r rI rN �carray�farray�behaved�fnc�forc�owndatarO �aligned� updateifcopy)rJ r' r' rD �test_otherflagsZ s zTestFlags.test_otherflagsc C sD t jdt �d�d�}t|jj� t jdt �d�d�}t|jj� d S )N� z|S4)r) rL )r/ r2 r) r rN rX )rJ rI r' r' rD �test_string_aligne s zTestFlags.test_string_alignc C s( t jdt �ddg�d�}t|jj� d S )Nr[ )rI �i4)�br] )r) )r/ r2 r) r rN rX )rJ rI r' r' rD �test_void_alignl s zTestFlags.test_void_alignN)�__name__� __module__� __qualname__rK rQ rZ r\ r_ r' r' r' rD rF M s rF c @ s e Zd Zdd� ZdS )�TestHashc C sp �xht jt jdft jt jdft jt jdft jt jdfgD �].\}}}�x t d|�D �]}t t |d| ��t d| �d||f d� t t |d|d ��t d|d �d ||d f d� t t |d| d ��t d| d �d ||f d� t |d d�}t t |d|d ��t d|d �d ||d f d� t t |d| d ��t d| d �d ||f d� qRW q8W d S ) N� � � �@ r, r& z %r: -2**%d)�err_msgz %r: 2**%dz %r: 2**%d - 1) r/ �int8r; �int16�uint16�int32�uint32�int64�uint64�ranger �hash�max)rJ �st�ut�s�ir' r' rD �test_ints s """"zTestHash.test_intN)r` ra rb rw r' r' r' rD rc q s rc c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )�TestAttributesc C s@ t �d�| _t �d��dd�| _t jdt jd��ddd�| _d S ) NrG � r[ rL �< )r) r& � )r/ rH �one�reshape�two�float64�three)rJ r' r' rD rK � s zTestAttributes.setUpc C s t | jjd� t | jjd� t | jjd� d| j_t | jjd� d| j_t | jj| jjf� | jj}t | jjd| |f� | jj}t | jjd| d| |f� t | jjd� t | jjd � t | jjd � | jj}t | jjd � t | jj d | � t | jj| jj j� t | jj t � d �� d S ) N)rG )r[ rL )r& rL r{ )rG r r& rL � r{ r, r& r ry )r r| r? r~ r� �stridesr9 �ndimrA �nbytesr) �baser/ rH )rJ �numr' r' rD �test_attributes� s& z TestAttributes.test_attributesc C s� t | jjt�tj�� t | jjt�tj�� t | jjjd� t | jjjd� | �| jjj d dk� t | jjj d d� t | jjj d d� d S )N�l�dr z<>r, rv �f) r r| r) r/ �int_r� �float_�char� assertTrue�str)rJ r' r' rD �test_dtypeattr� s z TestAttributes.test_dtypeattrc C sX t �d�}tjd dkr*tt|t�d� n*tt|t�d� ddlm} t||�d� d S )Nr r FT)�test_int_subclass) r/ r� �sys� version_infor � isinstance�int�numpy.core.multiarray_testsr� )rJ Z numpy_intr� r' r' rD �test_int_subclassing� s z#TestAttributes.test_int_subclassingc s� | j � � fdd�}t|ddd�t�ddddg�� | �t|ddd� | �t|ddd� | �t|d dd� t|d dd �t�dgd �� | �t|d d t�dd g�� |d d d� d S )Nc s t j| � t|� j |� j d�S )N)�bufferr) rC r� )r/ r= r� r9 )rA rC r� )�xr' rD � make_array� s z3TestAttributes.test_stridesattr.<locals>.make_arrayr[ r. r r& r, ���rd r )r& r rL ���rG )r| r r/ �arrayrP r1 )rJ r� r' )r� rD �test_stridesattr� s z TestAttributes.test_stridesattrc s0 | j � � fdd�}t|ddd�t�ddddg�� t|ddd�t�ddd d dd d g�� | �t|ddd � | �t|ddd� | �t|d dd� tjj� t� d�dd�� dd� }| �t|� d� j � j f� tjjj tj dtj d�d ddd�� | �t|� d d d� d� � d d d� }d|_ d|d d d� _ d S )Nc s^ y t j| gt� |� j d�}W n* tk rH } z t|��W d d }~X Y nX |� j |_}|S )N)r) r� rC )r/ r= r� r9 � Exception� RuntimeErrorr� )rA rC r� �r�e)r� r' rD r� � s z7TestAttributes.test_set_stridesattr.<locals>.make_arrayr[ r. r r& r, � rL r{ rd � r� )rG rG )r r c S s || _ d S )N)r� )�arrr� r' r' rD � set_strides� s z8TestAttributes.test_set_stridesattr.<locals>.set_stridesrG )r) )rG )r. )r? r� )r| r r/ r� rP r1 r� �lib� stride_tricks� as_stridedrH r9 ri r� )rJ r� r� rI r' )r� rD �test_set_stridesattr� s & z#TestAttributes.test_set_stridesattrc C sF x@dD ]8}t �d|�}t �d|�}|�d� d|d< t||� qW d S )Nz?bhilqpBHILQPfdgFDGO)r r& r, r, .)r/ r: r> r )rJ �tr� �yr' r' rD � test_fill� s zTestAttributes.test_fillc C sD t jdt jd�}t jdt jd�}d}||d< |�|� t||� d S )N)r r& r, )r) l ���� .)r/ r: ro r> r )rJ r� r� �valuer' r' rD �test_fill_max_uint64� s z#TestAttributes.test_fill_max_uint64c C s~ t jddgdd�}|�|d � t|d d |d d � t jdd d gd�}|�d � t|d d d g� t|d ddg� d S )N)r g )r, g �?zi4,f8)r) r �f1r, r& )rI �f8)r^ r] )g @r� rI g @r^ r� )r/ r� r> r r2 r )rJ r� r' r' rD �test_fill_struct_array� s z%TestAttributes.test_fill_struct_arrayN) r` ra rb rK r� r� r� r� r� r� r� r� r' r' r' rD rx � s "rx c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestArrayConstructionc C s� t �d�}t �||g�}t|t �d�� t �d�}t �d�}t �||g�}t||� d|d< t �||d g�}t||� t �d�}t �||gg�}t|t �d�� t �d�}t �||g||gg�}t|t �d�� t �d�}t �||g�}t|t �d�� t �d �}t �||d g|d g�}tt|�d� t|d ||d g� t|d |d � t jd t jd �}d |d< d |dd d�f< t �ddd gd d dgg�}t||� t �dd gdd gd dgg�}t||j� d S )Nr{ )r& r{ r& r, )r, r& r{ )r& r& r{ )r{ r{ )r& r{ r{ )r{ r )r& r )r) F)r r& T)r/ �onesr� r �len�bool�T)rJ r� r� �tgtr' r' rD � test_array s> z TestArrayConstruction.test_arrayc C s t ttj� d S )N)r � TypeErrorr/ r� )rJ r' r' rD �test_array_empty4 s z&TestArrayConstruction.test_array_emptyc C sz t �dddg�}t j|dd�}d|d< t|dddg� t j|ddd�}d|d< t|dddg� d |d< t|ddd g� d S ) Nr, r& r F)�copy�F)r� r* r[ r� )r/ r� r )rJ r� r� r' r' rD �test_array_copy_false7 s z+TestArrayConstruction.test_array_copy_falsec C s� t �dddgdddgg�}t j|dd�}d|d< d|d< t|dddgdddgg� t|dddgdddgg� t j|dd d �}d |d< d |d< t|ddd gdddgg� t|dd dgdddgg� d S ) Nr, r& r T)r� )r r, i����)r r& r� )r� r* rL r� )r/ r� r )rJ r� r� r' r' rD �test_array_copy_trueB s z*TestArrayConstruction.test_array_copy_truec C s� t �d�d d d� }tt �|�jj� tt �|�jj� tt �|�jj� tt �|�jj� t �d�d d d�d d d�f }tt �|�jj� tt �|�jj� d S )NrG r& )rG rG )r/ r� r �ascontiguousarrayrN � c_contiguous� f_contiguous�asfortranarray)rJ r� r' r' rD �test_array_contO s z%TestArrayConstruction.test_array_contN)r` ra rb r� r� r� r� r� r' r' r' rD r� s ' r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestAssignmentc C s� t �d��dd�}t �d�|d< t|dddgdddgg� t �d��dd�|d< t|dddgdddgg� t �d�d d d� �ddd�|d< t|dd dgdddgg� d d � }tt||t �d ��ddd�� d S ) Nr{ r& r .r r, r. rL r[ c S s || d< d S )N.r' )rI r^ r' r' rD �assignp s z;TestAssignment.test_assignment_broadcasting.<locals>.assign� )r/ rH r} r r r1 )rJ rI r� r' r' rD � test_assignment_broadcasting[ s "z+TestAssignment.test_assignment_broadcastingc sH G dd� d�}t �d�� � fdd�}tttf||� � tt|dg� d S )Nc @ s e Zd ZdS )z0TestAssignment.test_assignment_errors.<locals>.CN)r` ra rb r' r' r' rD r( w s r( r, c s | � d< d S )Nr r' )�v)rI r' rD r� { s z5TestAssignment.test_assignment_errors.<locals>.assign)r/ r2 r �AttributeErrorr� r1 )rJ r( r� r' )rI rD �test_assignment_errorsu s z%TestAssignment.test_assignment_errorsc sr ddl m� t� fdd��}t�dg�}t�d�}|d�� ||d< W d Q R X t|d d� t�t�d�g� d S ) Nr )�set_string_functionc 3 s2 �� fdd�dd� z dV W d�ddd� X dS )z% replace ndarray.__str__ temporarily c s � S )Nr' )r� )ru r' rD �<lambda>� � zLTestAssignment.test_unicode_assignment.<locals>.inject_str.<locals>.<lambda>F)�reprNr' )ru )r� )ru rD � inject_str� s z:TestAssignment.test_unicode_assignment.<locals>.inject_str�test�doneZbadu åäö)Znumpy.core.numericr� r r/ r� r )rJ r� Za1dZa0dr' )r� rD �test_unicode_assignment� s z&TestAssignment.test_unicode_assignmentc C sx t �dg�}t �dg�}G dd� dt�}tttj|dg � tttj|dg � tttj|d|� � tttj|d|� � d S )Nr� s donec @ s e Zd Zdd� Zdd� ZdS )z?TestAssignment.test_stringlike_empty_list.<locals>.bad_sequencec S s d S )Nr' )rJ r' r' rD � __getitem__� r� zKTestAssignment.test_stringlike_empty_list.<locals>.bad_sequence.__getitem__c S s t �d S )N)r� )rJ r' r' rD r+ � r� zGTestAssignment.test_stringlike_empty_list.<locals>.bad_sequence.__len__N)r` ra rb r� r+ r' r' r' rD � bad_sequence� s r� r )r/ r� r0 r r1 r7 �setitem)rJ �ur^ r� r' r' rD �test_stringlike_empty_list� s z)TestAssignment.test_stringlike_empty_listc C s� t �t �d�d�}t �t �d�d�}t �|g�}t|d |� ||d< t|d |� ||d< t|d |� |d |d< t|d |� |d |d< t|d |� t �t �|�g�}t|d |� d S )Nr r, r. )r ..)r/ � nextafter� longdoubler� r )rJ ZtinybZtinyaZtiny1dr� r' r' rD �test_longdouble_assignment� s z)TestAssignment.test_longdouble_assignmentN)r` ra rb r� r� r� r� r� r' r' r' rD r� Z s r� c @ s e Zd Zdd� Zdd� ZdS )�TestDtypedescrc C s< t �d�}t|t �t j�� t �d�}t|t �t j�� d S )Nr] r� )r/ r) r rl r )rJ �d1�d2r' r' rD �test_construction� s z TestDtypedescr.test_constructionc C s8 | � t�d�t�d�� | � t�dg�t�dg�� d S )Nz<i4z>i4)rI z<i4)rI z>i4)�assertNotEqualr/ r) )rJ r' r' rD �test_byteorders� s z TestDtypedescr.test_byteordersN)r` ra rb r� r� r' r' r' rD r� � s r� c @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dS )� TestZeroRankc C s t �d�t �dt�f| _d S )Nr r� )r/ r� r0 r� )rJ r' r' rD rK � s zTestZeroRank.setUpc C sV | j \}}| �|d d� | �|d d� | �|d j|k� | �|d j|k� d S )N.r r� )r� � assertEqualr� r� )rJ rI r^ r' r' rD �test_ellipsis_subscript� s z$TestZeroRank.test_ellipsis_subscriptc C s^ | j \}}| �|d d� | �|d d� | �t|d �|jjk� | �t|d �tk� d S )Nr' r r� )r� r� r� �typer) r� )rJ rI r^ r' r' rD �test_empty_subscript� s z!TestZeroRank.test_empty_subscriptc C sV | j \}}| �tdd� |� | �tdd� |� | �tdd� |� | �tdd� |� d S )Nc S s | d S )Nr r' )r� r' r' rD r� � r� z5TestZeroRank.test_invalid_subscript.<locals>.<lambda>c S s | d S )Nr r' )r� r' r' rD r� � r� c S s | t �g t� S )N)r/ r� r� )r� r' r' rD r� � r� c S s | t �g t� S )N)r/ r� r� )r� r' r' rD r� � r� )r� rP � IndexError)rJ rI r^ r' r' rD �test_invalid_subscript� s z#TestZeroRank.test_invalid_subscriptc C s: | j \}}d|d< | �|d� d|d< | �|�� d� d S )N�* .� )r� r� �item)rJ rI r^ r' r' rD �"test_ellipsis_subscript_assignment� s z/TestZeroRank.test_ellipsis_subscript_assignmentc C s: | j \}}d|d< | �|d� d|d< | �|�� d� d S )Nr� r' r� )r� r� r� )rJ rI r^ r' r' rD � test_empty_subscript_assignment� s z,TestZeroRank.test_empty_subscript_assignmentc C sL | j \}}dd� }| �t||dd� | �t||dd� | �t||dd� d S )Nc S s || |< d S )Nr' )r� rv r� r' r' rD r� � s z>TestZeroRank.test_invalid_subscript_assignment.<locals>.assignr r� r� r' )r� rP r� r1 )rJ rI r^ r� r' r' rD �!test_invalid_subscript_assignment� s z.TestZeroRank.test_invalid_subscript_assignmentc C s� | j \}}| �|tj jd� | �|dtjf jd� | �|tjdf jd� | �|dtjf jd� | �|tjdtjf jd� | �|dtjtjf jd� | �|tjtjdf jd� | �|tjfd jd� d S )N)r, .)r, r, rG ) r, r, r, r, r, r, r, r, r, r, )r� r� r/ �newaxisr? )rJ rI r^ r' r' rD � test_newaxis s zTestZeroRank.test_newaxisc C sD | j \}}dd� }| �t||tjdf� | �t||tjfd � d S )Nc S s | | d S )Nr' )r� rv r' r' rD � subscript s z4TestZeroRank.test_invalid_newaxis.<locals>.subscriptr �2 )r� rP r� r/ r� )rJ rI r^ r� r' r' rD �test_invalid_newaxis s z!TestZeroRank.test_invalid_newaxisc C sL t �d�}d|d< | �|d d� t jd|d�}d|d< | �|d d� d S )Nr' rL )r� r{ )r/ r= r� )rJ r� r� r' r' rD �test_constructor s z TestZeroRank.test_constructorc C s$ t �d�}| �tt j|dg|� d S )Nr& r, )r/ r� rP r1 �add)rJ r� r' r' rD � test_output s zTestZeroRank.test_outputN)r` ra rb rK r� r� r� r� r� r� r� r� r� r� r' r' r' rD r� � s r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�TestScalarIndexingc C s t �ddg�d | _d S )Nr r, )r/ r� r� )rJ r' r' rD rK # s zTestScalarIndexing.setUpc C s, | j }| �|d d� | �|d jd� d S )N.r r' )r� r� r? )rJ rI r' r' rD r� & s z*TestScalarIndexing.test_ellipsis_subscriptc C s, | j }| �|d d� | �|d jd� d S )Nr' r )r� r� r? )rJ rI r' r' rD r� + s z'TestScalarIndexing.test_empty_subscriptc C s. | j }| �tdd� |� | �tdd� |� d S )Nc S s | d S )Nr r' )r� r' r' rD r� 2 r� z;TestScalarIndexing.test_invalid_subscript.<locals>.<lambda>c S s | t �g t� S )N)r/ r� r� )r� r' r' rD r� 3 r� )r� rP r� )rJ rI r' r' rD r� 0 s z)TestScalarIndexing.test_invalid_subscriptc C s$ | j }dd� }| �t||dd� d S )Nc S s || |< d S )Nr' )r� rv r� r' r' rD r� 8 s zDTestScalarIndexing.test_invalid_subscript_assignment.<locals>.assignr r� )r� rP r� )rJ rI r� r' r' rD r� 5 s z4TestScalarIndexing.test_invalid_subscript_assignmentc C s� | j }| �|tj jd� | �|dtjf jd� | �|tjdf jd� | �|dtjf jd� | �|tjdtjf jd� | �|dtjtjf jd� | �|tjtjdf jd� | �|tjfd jd� d S )N)r, .)r, r, rG ) r, r, r, r, r, r, r, r, r, r, )r� r� r/ r� r? )rJ rI r' r' rD r� = s z TestScalarIndexing.test_newaxisc C s@ | j }dd� }| �t||tjdf� | �t||tjfd � d S )Nc S s | | d S )Nr' )r� rv r' r' rD r� K s z:TestScalarIndexing.test_invalid_newaxis.<locals>.subscriptr r� )r� rP r� r/ r� )rJ rI r� r' r' rD r� H s z'TestScalarIndexing.test_invalid_newaxisc C sx t �d�}|dd � |d d�< t|ddddg� t �d�}|d d� |dd �< t|ddddg� t �d�}|d d d� |d d �< t|ddddg� t �d��dd�}|d d �d d d�f |d d d�d d �f< t|dddgdddgg� t �d��dd�}|d d �d d d�f |d d d�d d d�f< t|dddgdddgg� t �d�}|dd � |d d�< t|dddddg� t �d�}|d d� |dd �< t|dddddg� t �d�}|dd � |dd d�< t|dddddg� t �d�}|dd d� |dd �< t|dddddg� t �d�}|d dd� |dd d�< t|dddddg� t �d�}|dd d� |d dd�< t|dddddg� d S ) Nr[ r, r. r& r r r{ rL )r/ rH r r} )rJ rI r' r' rD �test_overlapping_assignmentQ sB (* z.TestScalarIndexing.test_overlapping_assignmentN) r` ra rb rK r� r� r� r� r� r� r� r' r' r' rD r� " s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej d d � �Z d d� Z dd� Z dd� Z dd� Zdd� Ze�ejd dk�dd� �Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%S )&� TestCreationc C s& G dd� dt �}| �ttj|� � d S )Nc @ s e Zd Zddd�ZdS )z+TestCreation.test_from_attribute.<locals>.xNc S s d S )Nr' )rJ r) r' r' rD � __array__� s z5TestCreation.test_from_attribute.<locals>.x.__array__)N)r` ra rb r� r' r' r' rD r� � s r� )r0 rP r1 r/ r� )rJ r� r' r' rD �test_from_attribute� s z TestCreation.test_from_attributec C s` t jd t jd }ddg}t jddgtd�}x,|D ]$}d| }tt j||d�||d� q4W d S )N� AllInteger�Float�123�{ )r) zString conversion for %s)rh )r/ � typecodesr� r� r )rJ �typesZnstr�resultr� �msgr' r' rD �test_from_string� s z TestCreation.test_from_stringc C s t jg dd�}t|jjd� d S )N�V)r) )r/ r� r r) �kind)rJ r� r' r' rD � test_void� s zTestCreation.test_voidc C sp t �d�jdkrd}nt �d�jdkr,d}nd S ttt j|t jd� ttt j|t jd� ttt j|t jd� d S )N�intpi���)i� i� l ���� )l 4s j l 4s j )r) ) r/ �iinforr r r1 r: ri r2 r� )rJ r? r' r' rD �test_too_big_error� s z TestCreation.test_too_big_errorc C s� t jd t jd }x�|D ]�}t jd|d�}tt �|�d� t|�� d� t|�� � t jddd�}tt �|�d� t|�� d� t|�� � t jddd�}tt �|�d� t|�� d� t|�� � t jdd d�}tt �|�d� qW d S ) Nr� �AllFloat)� )r) r r& z(2,4)i4Z4i4z(2,4)i4, (2,4)i4)r/ r r2 r � count_nonzero�sumr �any)rJ r �dtr� r' r' rD � test_zeros� s zTestCreation.test_zerosc C sD t jd t jd }x*|D ]"}t jd|d�}t|�� � ~qW d S )Nr� r )i �)r) )r/ r r2 r r )rJ r r r� r' r' rD �test_zeros_big� s zTestCreation.test_zeros_bigc C s2 t jdtd�}t|dgd � tt �|�d� d S )N)r )r) r r )r/ r2 r0 r r r )rJ r� r' r' rD �test_zeros_obj� s zTestCreation.test_zeros_objc C s( t jddtdfgd�}t|d d� d S )NrG �kr& )r) r )r/ r2 r0 r )rJ r� r' r' rD �test_zeros_obj_obj� s z TestCreation.test_zeros_obj_objc C s� xNt jd D ]@}|dkrq t jd|d�}tt �|�|� tt �|�j|j� q W t jddd�}tt �|�|� tt �|�j|j� t jddd�}tt �|�|� tt �|�j|j� t jddd�}tt �|�|� tt �|�j|j� t jddd�}tt �|�|� tt �|�j|j� t jdd d�}tt �|�|� tt �|�j|j� t jdd d�}tt �|�|� tt �|�j|j� t jdd d�}tt �|�|� tt �|�j|j� d S ) N�Allr )r r )r) �S5�U5z<i4z>i4z<M8[s]z>M8[s]zf4,f4)r/ r r2 r � zeros_liker r) )rJ �cr� r' r' rD �test_zeros_like_like_zeros� s6 z'TestCreation.test_zeros_like_like_zerosc C s0 x*t ddd�D ]}tj|dd�}t|� qW d S )NrL �d �U)r) )rp r/ r: r� )rJ rv r� r' r' rD �test_empty_unicode� s z TestCreation.test_empty_unicodec C s� t t�ddd g�jtj� t t�ddd dg�jtj� t t�dd dg�jtj� t t�dd gd �jtj� t t�ddgddgg�jtj� t t�ddgddgg�jtj� t t�dddgdddgdddgg�jtj� d S )Nr[ r& �P r r, y �?g �?)r r/ r� r) r0 �complex)rJ r' r' rD � test_sequence_non_homogenous� s z)TestCreation.test_sequence_non_homogenousr r c C s� t t�td�td�g�jtj� t t�td�dd g�jtj� t t�td�dd td�g�jtj� t t�dd td�g�jtj� d S )Nr[ r& r )r r/ r� �longr) r0 )rJ r' r' rD �test_sequence_long s &z TestCreation.test_sequence_longc C sf G dd� dt �}G dd� dt �}t�|� g�}t|jdk� t|jt�t �k� tttj|� g� dS )a) Should not segfault. Class Fail breaks the sequence protocol for new style classes, i.e., those derived from object. Class Map is a mapping type indicated by raising a ValueError. At some point we may raise a warning instead of an error in the Fail case. c @ s e Zd Zdd� Zdd� ZdS )z5TestCreation.test_non_sequence_sequence.<locals>.Failc S s dS )Nr, r' )rJ r' r' rD r+ s z=TestCreation.test_non_sequence_sequence.<locals>.Fail.__len__c S s t � �d S )N)r1 )rJ �indexr' r' rD r� s zATestCreation.test_non_sequence_sequence.<locals>.Fail.__getitem__N)r` ra rb r+ r� r' r' r' rD �Fail s r% c @ s e Zd Zdd� Zdd� ZdS )z4TestCreation.test_non_sequence_sequence.<locals>.Mapc S s dS )Nr, r' )rJ r' r' rD r+ s z<TestCreation.test_non_sequence_sequence.<locals>.Map.__len__c S s t � �d S )N)�KeyError)rJ r$ r' r' rD r� s z@TestCreation.test_non_sequence_sequence.<locals>.Map.__getitem__N)r` ra rb r+ r� r' r' r' rD �Map s r' )r, N)r0 r/ r� r r? r) r r1 )rJ r% r' rI r' r' rD �test_non_sequence_sequence s z'TestCreation.test_non_sequence_sequencec C s: G dd� d�}t �|� |� |� g�}t|jt �t�� d S )Nc @ s e Zd Zdd� Zdd� ZdS )z4TestCreation.test_no_len_object_type.<locals>.Point2c S s d S )Nr' )rJ r' r' rD �__init__( s z=TestCreation.test_no_len_object_type.<locals>.Point2.__init__c S s |dkr |S t � �d S )N)r r, )r� )rJ �indr' r' rD r� + s z@TestCreation.test_no_len_object_type.<locals>.Point2.__getitem__N)r` ra rb r) r� r' r' r' rD �Point2' s r+ )r/ r� r r) r0 )rJ r+ r� r' r' rD �test_no_len_object_type% s z$TestCreation.test_no_len_object_typec C s" G dd� d�}t ttj|� � d S )Nc @ s e Zd Zdd� Zdd� ZdS )z/TestCreation.test_false_len_sequence.<locals>.Cc S s t �d S )N)r� )rJ rv r' r' rD r� 6 s z;TestCreation.test_false_len_sequence.<locals>.C.__getitem__c S s dS )Nr� r' )rJ r' r' rD r+ 8 s z7TestCreation.test_false_len_sequence.<locals>.C.__len__N)r` ra rb r� r+ r' r' r' rD r( 5 s r( )r r1 r/ r� )rJ r( r' r' rD �test_false_len_sequence3 s z$TestCreation.test_false_len_sequencec C s6 G dd� dt �}|dddg�}ttt�|��d� d S )Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z0TestCreation.test_failed_len_sequence.<locals>.Ac S s || _ d S )N)�_data)rJ r- r' r' rD r) @ s z9TestCreation.test_failed_len_sequence.<locals>.A.__init__c S s t | �| j| �S )N)r� r. )rJ r� r' r' rD r� B s z<TestCreation.test_failed_len_sequence.<locals>.A.__getitem__c S s t | j�S )N)r� r. )rJ r' r' rD r+ D s z8TestCreation.test_failed_len_sequence.<locals>.A.__len__N)r` ra rb r) r� r+ r' r' r' rD �A? s r/ r, r& r )r0 r r� r/ r� )rJ r/ r� r' r' rD �test_failed_len_sequence= s z%TestCreation.test_failed_len_sequencec C sr t �d�}t �t j�j}xTdD ]L}t �|�}|j}t j|d|| f|d� tt t j|d|| d f|d� q W d S )Nr )r �S20r^ )r )r� r� r? r) r, ) r/ r2 r r rr r) r9 r= r r1 )rJ rB Z max_bytesr) r9 r' r' rD �test_array_too_bigK s z TestCreation.test_array_too_bigN)r` ra rb r� r r r r r �slowr r r r r r! �skipifr� r� r# r( r, r- r0 r2 r' r' r' rD r� � s" " r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�TestStructuredc C s� t jddgd�}t �d��dddd�|d< t|jd |d �d d dd�� |jd d �}t|d j |d j � t|jd j |j�� d j � d S ) N)r rL )rI )r] )r& r& )r) rz r rL r& rI r, r r� )r* ) r/ r2 rH r} r r� � transposer� r r? )rJ rI r^ r' r' rD �test_subarray_field_access\ s z)TestStructured.test_subarray_field_accessc C s, t jjdddgdddgddggfdddgdddgddggfgddt jfd gd �}|�� }t||kd d g� t||kd d g� d |d _t||kd d g� t||kd d g� xRtd�D ]F}|d j|d _d|d j|< t||kd d g� t||kd d g� q�W xdtd�D ]X}xPtd�D ]D}|�� }d|d j ||f< t||kd d g� t||kd d g� �qW �qW t j dgdggdgd �}t j dddgdgd �}t||kd d d gd d d gg� t||kd d d gd d d gg� t j dgdggdgd �}t j dddgdgd �}t||kd d d gd d d gg� t||kd d d gd d d gg� t j ddgfgddgfggdgd �}t j ddgfddgfddgfgdgd �}t||kd d d gd d d gg� t||kd d d gd d d gg� t j ddgfgddgfggdgdd�}t j ddgfddgfddgfgdgd �}t||kd d d gd d d gg� t||kd d d gd d d gg� t j dddgd �}t j dddgd �}t � � }|� td� t||kd � W d Q R X t j dddgd �}t j dddgd �}t � � }|� td� t||kd � W d Q R X t j d d gd d gd �}t j d d gd!d"gd �}t||kd d g� d S )#Nr, r& r rI r[ r^ r )rI )�f4r )r )r] )r& r& )r) TFr rL rG )r )r, )rI r� )rI r� )r, )rI r� )r& r� )r) r* )rI )r8 )r, r& )r^ �i1)rI )r8 )r& z elementwise == comparison failed)rI )r8 )r& r, )rL r� )rG r, )rI z>i8)r^ z<f8)rL �+ )rI z<i8)r^ z>f8)r/ �rec� fromrecordsr0 r� r r^ rp rI r r� r2 r# �filter� FutureWarning)rJ rI r^ rv �jr� r� �supr' r' rD �test_subarray_comparisonj sb 6 $( &( z'TestStructured.test_subarray_comparisonc C s� t jdgdgd�}tt j|jdgdd�� |�dg�}t||�� �� � t|d d |d d � t jd d gdd gd�}t jd d gddgd�}tt j|j|jdd�� t||kddg� tt j|j|jdd�� t t �� |j|jdd�}W d Q R X t||kddg� ddg}tt j|j|dd�� |j|dd�}t|t jd d g|d�kddg� ddg}tt j|j|dd�� |j|dd�}t|t jd d g|d�kddg� ddg}tt j|j|dd� � t t |j|dd� dd g}tt j|j|dd� � t t |j|dd� ddg}tt j|j|dd� � t t |j|dd� tt j|j|jdd� � t t |j|jdd� xNdD ]F}dg}tt j|j||d� � dd dg}tt j|j||d� � �qhW d S ) N)r, )rI z<i4)r) )rI z>i4�unsafe)�castingrI r )rL r� )rG r, )r^ z<f8)r� rL )r, rG )r^ z>f8�equivT)rI z<i8�safe)r^ z>f4� same_kind)rI z>i8)r^ z<f4)rI z>i2)r^ z<i2�no)rG rE rD rF )r r] ) r/ r� r �can_castr) �astyper �byteswap� newbyteorderr r> r r� )rJ rI r^ r r� rC r' r' rD � test_casting� sN zTestStructured.test_castingc C s@ t jg dddgd�}|ddg t �ddd g�}|d d g d S ) N)rI r� )r^ r� )r �O)r) rI r^ r )r/ rv )�Bz|OrN r/ )r/ r� r2 )rJ rI Zdat2r' r' rD � test_objview� s zTestStructured.test_objviewc C sv t �dg�}t �dd|fg�}t �d|�}t jddd�|d d < t jd|d�|d d< t|d d t jddd�� d S ) N)�elemr] rL )�fieldr] rG �structr, rG r] )r) r rQ )r/ r) r2 r� r )rJ Z struct_dtr r� r' r' rD � test_setfield� s z TestStructured.test_setfieldc sd t jddgd�}t �d�|d d< t|d d t �d�� t jddgd�� � fdd �}tt|� d S ) Nr, )r� rM )r) r r r� )r� rM rL c s t �d�� d d< d S )Nr r r� )r/ rH r' )r r' rD � testassign s z7TestStructured.test_setfield_object.<locals>.testassign)r/ r2 rH r r r1 )rJ r^ rT r' )r rD �test_setfield_object� s z#TestStructured.test_setfield_objectc C sX t �dtfdg�}t jd|d�}t|d ddddg� t|d jd� dd d d g|d< t|d ddddg� t|d ddddg� d |d |d dk< t|d ddddg� t|d ddddg� d |d< t|d ddddg� t|d ddddg� t jd|d jd�}t|jd� t|d |� tt jdddgd�d jd� tt jddd�jd� tt jddd�d jd� |d �d�}t|jd� t|ddgddgg� t|d d � j|j� t t t |�t t j d��|� t�� }t �||� |�d� t �|�}t|jd� t||� tdd��2}t �||� t �|�}t|jd� t||� W d Q R X d S )N�I)�S�S0r[ )r) rW r� r rI r^ r r� Zhellor/ )rI zS0,S0)r^ �u1r zS0,S0zS0,u1�f0)r& r& )r� z.npy)�suffix)r/ r) r� r2 r r9 r= r: r} r �evalr� �dictr� �io�BytesIO�save�seek�loadr" )rJ r r� r� Zxxr^ �yy�tmpr' r' rD �test_zero_width_string sH z%TestStructured.test_zero_width_stringc C s( t jddd�}|d }t|j|k� d S )Nr zi4,f4)r) r )r/ r2 r r� )rJ rI r^ r' r' rD �test_base_attrF s z TestStructured.test_base_attrN) r` ra rb r7 rA rL rO rS rU re rf r' r' r' rD r5 [ s E= 8r5 c @ sB e Zd Zdd� Zdd� Zdd� Zdd� Zejd d � �Z d d � Z d S )�TestBoolc C sz t �d�}t �d�}| �||k� t �d�}t �d�}| �||k� | �t �dg�d |k� | �t �d�d |k� d S )Nr Fr, Tr' )r/ �bool_r� r� )rJ �a0�b0�a1�b1r' r' rD �test_test_interningM s z TestBool.test_test_interningc C s� t jdt jd�}t|�� |j� t|d d d� �� |d d d� j� t|d d d� �� |d d d� j� t jdtd�}t|�� |j� t|d d d� �� |d d d� j� t|d d d� �� |d d d� j� d S )N�e )r) r& r� s� ��������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������)r/ r� r� r r rA � frombuffer)rJ r� r' r' rD �test_sumW s $$$zTestBool.test_sumc s� dd� t |�D �}x�t d| �D ]�� � fdd�|D �}tj|tjd�}t�|�}| �t�|�|� |�tj �}|d9 }| �t�|�|� |d9 }| �t�|�|� d||d k< | �t�|�|� q W d S ) Nc S s g | ] }d | �qS )r& r' )�.0rv r' r' rD � <listcomp>c s z0TestBool.check_count_nonzero.<locals>.<listcomp>r& c s g | ]}� |@ d k�qS )r r' )rq r� )rv r' rD rr e s )r) r r[ � r ) rp r/ r� r� �builtinsr r� r �viewr; )rJ �power�lengthZpowersr� rI r �avr' )rv rD �check_count_nonzerob s z TestBool.check_count_nonzeroc C s | � dd� d S )Nr� � )ry )rJ r' r' rD �test_count_nonzeroq s zTestBool.test_count_nonzeroc C s | � dd� d S )Nrz )ry )rJ r' r' rD �test_count_nonzero_allv s z TestBool.test_count_nonzero_allc C s� x�t d�D ]�}tjdtjd�|d d � }d|d |�< | �t�|�t�|�� �� tj dtjd�|d d � }d|d |�< | �t�|�t�|�� �� q W d S )Nr� )� )r) r, TF) rp r/ r2 r� r� r rt r �tolistr� )rJ �orI r' r' rD � test_count_nonzero_unaligned| s z%TestBool.test_count_nonzero_unalignedN) r` ra rb rm rp ry r{ r r3 r| r� r' r' r' rD rg L s rg c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIdJ� Z'dKdL� Z(dMdN� Z)dOdP� Z*dQdR� Z+dSdT� Z,dUdV� Z-dWdX� Z.dYdZ� Z/d[d\� Z0d]d^� Z1d_d`� Z2dadb� Z3dcdd� Z4deS )f� TestMethodsc C s� dddddgg}t �d��dd�}|jdd gdd �}t||� d d gddgg}|jdd dd dgd d �}t||� d gdgg}t �d��dd�}|jdd gd d �}t||� t �d��dd�}|�dd g�}t|d � d S ) NrL r{ r� rd r� rG r& r r, )�axisr )r/ rH r} �compressr )rJ r� r� �outr' r' rD � test_compress� s zTestMethods.test_compressc C s� dt jdtd� }dt jdtd� }dt jdtd� }dt jdtd� }t �dddg�}|�||f�}t|dddg� |�||f�}t|dddgdddgg� |�||f�}t|dddgdddgg� d S )Nr& )r )r) r )r& r r r, )r/ r� r� r� �chooser )rJ r� r� �x2�y2r* r/ r' r' rD � test_choose� s zTestMethods.test_choosec C s� dddddddg}ddddgddd d gddddgg}x�t jt jt jt jt jt jt jt jgD ]�}t � ||�}t � ||�}|d kr�| � t |j � | j t |j dd � qZt |j d d �d� t|j d d �t � ddddg|�� t|j dd �t � dddg|�� qZW d S )Nr, r& rG � r{ rL r[ r r� r� )�1r^ )r� r i g r� �$ �T � r. � ib iX )r/ rj rk rl rm �float32r � complex64� complex128r� rP �ArithmeticError�prodr r )rJ �baZba2�ctyperI �a2r' r' rD � test_prod� s " zTestMethods.test_prodc C s^ t �ddddddg�}|�d�}|�ddddddg�}t|ddddddddddg � |�d�}t|ddddddddddddg � |jddgdd �}t|dddgdddgdddgg� |jdddgdd �}t|ddddddgddddddgg� |jddd �}t|dddgdddgdddgdddgg� |jddd �}t|ddddddgddddddgg� d S ) Nr, r& r r[ rL r{ )r& r r )r� )r/ r� r} �repeatr )rJ �m�m_rectr/ r' r' rD � test_repeat� s. zTestMethods.test_repeatc C s t �dddgdddgddd gd d d gg�}ddddddgddd d d d gg}t|�dd�|� ddddgddddgd d d d gg}t|�dd�|� dd ddgddd d gdddd gg}t|jd dd�|� dddd gdddd gddd d gg}t|jjd dd�|� d S )Nr, r& r r[ rL r{ r� rd r� rG r� r� )r r[ r� )r* r( )r/ r� r r} r� )rJ r� r� r' r' rD � test_reshape� s * """zTestMethods.test_reshapec C s� dd� }|t �ddg�ddg� |t �d�d� |t �ddg�d d gd � |t �d d g�ddgd� |t �dg�dg� |t �dg�dgd � d S )Nc W sB t | j|� |� t�| �}| j|d|i�}t ||� t ||� d S )Nr� )r �roundr/ r )r� �expectedZ round_argsr� �resr' r' rD � check_round� s z+TestMethods.test_round.<locals>.check_roundg333333�?g �?r, r& gffffff(@g /@rG ry r. g�����L(@g��Q� /@y @ �?y @ @y )@ /@y $@ 4@)r/ r� )rJ r� r' r' rD � test_round� s zTestMethods.test_roundc C sx t �dgdgdggg�}t|�� dddg� t|jdd�dgdgdgg� tt|jdd� t|jdd�dddgg� d S )Nr, r& r )r )r� )r, )r& )r/ r� r �squeezer r1 )rJ rI r' r' rD � test_squeeze� s zTestMethods.test_squeezec sp t �ddgddgg�� t� �� ddgddgg� | �t� fdd�� | �t� fdd�� | �t� fdd�� d S ) Nr, r& r r[ c s � � d�S )Nr )r6 r' )rI r' rD r� r� z,TestMethods.test_transpose.<locals>.<lambda>c s � � dd�S )Nr )r6 r' )rI r' rD r� r� c s � � ddd�S )Nr r, r& )r6 r' )rI r' rD r� r� )r/ r� r r6 rP r1 )rJ r' )rI rD �test_transpose s zTestMethods.test_transposec s� d}t �t jddg�}t �|�}t||d d d� |� d}t jdt jd�}| jt jt jt jddddddg 7 _| jt jddt jt jddddg 7 _t �|�}t||d d d� |� t � d�}|d d d� � � }xTd D ]L}d | }|� � }|j|d � t|||� |� � }|j|d � t|||� q�W |d d }|d d }xVd D ]N}d | }|� � }|j|d � t|||� |� � }|j|d � t|||� �qFW |d }|d }xVd D ]N}d| }|� � }|j|d � t|||� |� � }|j|d � t|||� �q�W xbdD ]Z}xRt j d D ]D} t jdddg|| d�} | � � }|�� d� | �}t|| |� �qW �qW d� t �� fdd�t d�D ��}|d d d� � � }xVd D ]N}d| }|� � }|j|d � t|||� |� � }|j|d � t|||� �q�W d� t j� fdd�t d�D �t jd�}|d d d� � � }xVd D ]N}d| }|� � }|j|d � t|||� |� � }|j|d � t|||� �q,W t jdt jd�}tt d��|d d �< |d d d� }xVd D ]N}d | }|� � }|j|d � t|||� |� � }|j|d � t|||� �q�W t �d tfd tfg�} t jd d� t d�D �| d�}|d d d� }xVd D ]N}d | }|� � }|j|d � t|||� |� � }|j|d � t|||� �qNW t j ddd!d�}|d d d� }xVd D ]N}d"| }|� � }|j|d � t|||� |� � }|j|d � t|||� �q�W t j ddd#d�}|d d d� }xVd D ]N}d$| }|� � }|j|d � t|||� |� � }|j|d � t|||� �q:W t �d%d&gddgg�}t �ddgd%d&gg�}t �d&d%gddgg�}|� � } | jdd'� t| |d(� |� � } | jdd'� t| |d)� |� � } | �� t| |d*� t �g �}d+|_x:t |j |j�D ]&} d,� | �}tt j|| d'�||� �qLW d-}tt j|d d'�|�� |� G d.d/� d/t�} t j| � gd0 td�}x"d D ]}d1| }|j|d � �q�W d S )2Nz Test real sort order with nansr, r r. z!Test complex sort order with nansr� )r) rn )�qr� �hzscalar sort, kind=%s)r y �?z%complex sort, real part == 1, kind=%sz%complex sort, imag part == 1, kind=%sz<>�Complexy �? @y @ @y @ �?z$byte-swapped complex sort, dtype={0}�aaaaaaaac s g | ]}� t |� �qS r' )�chr)rq rv )ru r' rD rr O s z)TestMethods.test_sort.<locals>.<listcomp>zstring sort, kind=%sc s g | ]}� t |� �qS r' )r� )rq rv )ru r' rD rr \ s zunicode sort, kind=%s)rn )r� r� r� zobject sort, kind=%sr� rv c S s g | ] }||f�qS r' r' )rq rv r' r' rD rr v s z datetime64[D]zdatetime64 sort, kind=%sztimedelta64[D]ztimedelta64 sort, kind=%sr r& )r� ztest sort with axis=0ztest sort with axis=1ztest sort with default axis)r r& r, r z#test empty array sort with axis={0}z$test empty array sort with axis=Nonec @ s e Zd Zdd� ZdS )z#TestMethods.test_sort.<locals>.Boomc S s dS )NTr' )rJ �otherr' r' rD �__lt__� s z*TestMethods.test_sort.<locals>.Boom.__lt__N)r` ra rb r� r' r' r' rD �Boom� s r� r z%bogus comparison object sort, kind=%s)r/ r� �nan�sortr r2 r� �real�imagrH r� r �formatrp r r: r0 �listr) �floatr� r? r� �ravel)rJ r rI r^ r r �ai�bi� endianessr r� r� r� r� r' )ru rD � test_sort s && " zTestMethods.test_sortc C sp x.t d�D ]"}t�dd�}|d d d� �� q W t�dg�}x.t d�D ]"}t�d|�}|d d d� �� qFW d S )Nr[ i� ZV4r. )�valr] )r, )rp r/ r: r� r) )rJ rv r� r r' r' rD �test_void_sort� s zTestMethods.test_void_sortc s� t jdt�� dgtd�}x dD ]}tt|j|d� q W G dd� dt�� t �� fdd �td �D ��� d �}t j � |� x dD ]}tt|j|d� qzW d S ) Nr r, )r) )r� r� r� )r c @ s, e Zd Zdd� Ze Z Z Z Z ZZ dS )z,TestMethods.test_sort_raises.<locals>.Raiserc _ s t d��d S )NzSOMETHING ERRORED)r� )�args�kwargsr' r' rD �raises_anything� s z<TestMethods.test_sort_raises.<locals>.Raiser.raises_anythingN) r` ra rb r� �__eq__�__ne__r� �__gt__�__ge__�__le__r' r' r' rD �Raiser� s r� c s g | ]}� � |g�qS r' r' )rq �n)r� r' rD rr � s z0TestMethods.test_sort_raises.<locals>.<listcomp>rG r. ) r/ r� r% �nowr0 r r� r� rp r} �random�shuffle)rJ r� r r' )r� rD �test_sort_raises� s " z TestMethods.test_sort_raisesc C s~ t �d�}|�� }|}x>|jdkrT|jd }|d || ||< |d< |d d� }qW tt �|�|� t|t �|� |� d S )Ni@B r r& r� )r/ rH r� rA r r� �argsort)rJ r� �dor� �midr' r' rD �test_sort_degraded� s z TestMethods.test_sort_degradedc C s� dd� }dd� }t jddd�}||�� � ||�d�� ||�d�� ||�d �� t jddd�}||�� � ||�d�� ||�d�� ||�d �� d S ) Nc S s* t | jj� t | jj� t | jj � d S )N)r rN �fortranr� r� )r� r' r' rD �assert_fortran� s z-TestMethods.test_copy.<locals>.assert_fortranc S s, t | jj � t | jj � t | jj� d S )N)r rN r� r� r� )r� r' r' rD �assert_c� s z'TestMethods.test_copy.<locals>.assert_c)r& r& r� )r* r( r/ )r/ r: r� )rJ r� r� rI r' r' rD � test_copy� s zTestMethods.test_copyc C s� t �dddg�}t �dddg�}t �ddd g�}t jj|||gd d �}|jd gd � t|jt �dddg�� t|jt �dddg�� t|jt �d ddg�� |jdgd � t|jt �dddg�� t|jt �dddg�� t|jt �ddd g�� |jdgd � t|jt �dddg�� t|jt �dddg�� t|jt �ddd g�� t j dk�rZd}nd}dt d fd|fg}t jddddg|d�}|jdd � t|d ddd d g� t|t jddddg|d�� d S ) N� rf � �my�first�nameg������@g @g������@zid,word,number)�names�id)r* �word�number�littlez>i2z<i2�5Zcol2)rI r, )r^ rs )r r )r� i )r) r, r rs i ) r/ r� r; � fromarraysr� r r� r� r� r� � byteorderr )rJ �x1r� �x3r� ZstrtypeZmydtyper' r' rD �test_sort_order� s4 zTestMethods.test_sort_orderc sz t �d�}|d d d� �� }xDdD ]<}d| }t|�� j|d�||� t|�� j|d�||� q"W |d d }|d d }xDdD ]<}d| }t|�� j|d�||� t|�� j|d�||� q�W |d }|d }xDdD ]<}d| }t|�� j|d�||� t|�� j|d�||� q�W xfd D ]^}xVt jd D ]H}t jd d d g|| d�} d�|�}t| �� t jt| �t j d�|� �q,W �q W d� t �� fdd�t d�D ��}|d d d� �� }t �d�} | d d d� } xFdD ]>}d| }t|�� j|d�| |� t|�� j|d�| |� �q�W d� t j� fdd�t d�D �t j d�}|d d d� }t �d�} | d d d� } xFdD ]>}d| }t|�� j|d�| |� t|�� j|d�| |� �qbW t j dt j d�}tt d��|d d �< |d d d� }t �d�} | d d d� } xFdD ]>}d| }t|�� j|d�| |� t|�� j|d�| |� �q�W t �dtfdtfg�}t jdd� t d�D �|d�}|d d d� }t �d�} | d d d� } xFdD ]>}d| }t|�� j|d�| |� t|�� j|d�| |� �q�W t jd dd d�}|d d d� }t �d�} | d d d� } xFd D ]>}d | }t|�� j|d�| |� t|�� j|d�| |� �qW t jd dd d�}|d d d� }t �d�} | d d d� } xFd D ]>}d!| }t|�� j|d�| |� t|�� j|d�| |� �q�W t �d"d#gdd gg�}t �ddgd d gg�}t �dd gdd gg�} t|�� jd d$�|� t|�� jdd$�| � t|�� �� | � t �g �}d%|_xFt |j |j�D ]2} d&�| �}tt j|| d$�t j|t j d�|� �qtW d'}tt j|d d$�t j|�� t j d�|� t �d(�} t �d(�}t|jd)d�| � t jd(t jd�}t|jd)d�| � t �d*d� t d(�D ��}t|jd)d�| � t jd+d� t d(�D �t j d�}t|jd)d�| � d S ),Nrn r. )r� r� r� zscalar argsort, kind=%s)r y �?r, zcomplex argsort, kind=%sz<>r� y �? @y @ @y @ �?)r) z'byte-swapped complex argsort, dtype={0}r� c s g | ]}� t |� �qS r' )r� )rq rv )ru r' rD rr F s z,TestMethods.test_argsort.<locals>.<listcomp>zstring argsort, kind=%sc s g | ]}� t |� �qS r' )r� )rq rv )ru r' rD rr Q s zunicode argsort, kind=%s)rn zobject argsort, kind=%sr� rv c S s g | ] }||f�qS r' r' )rq rv r' r' rD rr g s z!structured array argsort, kind=%sr z datetime64[D])r� r� r� zdatetime64 argsort, kind=%sztimedelta64[D]z timedelta64 argsort, kind=%sr r& )r� )r r& r, r z&test empty array argsort with axis={0}z'test empty array argsort with axis=Noner r� c S s g | ]}d �qS )� aaaaaaaaar' )rq rv r' r' rD rr � s c S s g | ]}d �qS )r� r' )rq rv r' r' rD rr � s )r/ rH r� r r� r r� r� r� r rp r r: r0 r� r) r� r� r? r� r r� r2 r )rJ rI r^ r r r� r� r� r r� r� �rrr r� r' )ru rD � test_argsort s� $ " zTestMethods.test_argsortc C s8 t �d�}d�d�}tt|j|d� tt|j|d� d S )NrG s ä�UTF8)r )r/ rH �decoder r1 r� r� )rJ r� r r' r' rD �test_sort_unicode_kind� s z"TestMethods.test_sort_unicode_kindc C s� t �ddt jg�}d}|j|dd�}t|t �d�|� d}|j|dd�}t|t �dd �|� t jd t jd �}| jddddddt jt jt jg 7 _| j ddddt jt jddt jg 7 _ d }|j|dd�}t|t �d �|� d }|j|dd�}t|t �dd�|� d}t jddgdd �}|�t jddd ��}t|d|� d}t jddgdd �}|�t jddd ��}t|d|� t � d�}|�dddgd�}t|dddg� |�dddgd�}t|dddg� t � d�}|�dddgd�}t|dddg� |�dddgd�}t|dddg� t � d�}|�dddgd�}t|dddg� |�dddgd�}t|dddg� t �d�}t � |j |j d d�}|dd � �|j�}||d d �< |�|d�}t||� |�|d�}t||d � |�|d�}t||� |�|d�}t||d � t �d�}|�ddd gd�}t|ddd g� |�ddd gd�}t|dddg� d�t jd t jd t jd d f�}x�|D ]�}|d k�r�d }|d k�r�t jd|d �}t �d�}nt jdd|d �}t �d�}|�|d�}t||� |�|d�}t||d � �qrW d S ) Nr r, z*Test real searchsorted with nans, side='l'r� )�sider z*Test real searchsorted with nans, side='r'r� r[ r� )r) z-Test complex searchsorted with nans, side='l'z-Test complex searchsorted with nans, side='r'rG z.Test searchsorted with little endian, side='l'� z<i4z+Test searchsorted with big endian, side='l'z>i4r& r; rL r{ r� r� r �Datetimez?O�MzM8[D]�?)r/ r� r� � searchsortedr rH r2 r� r� r� r� r: r9 rA ru r) �joinr ) rJ rI r r^ rX � unalignedr r r� r' r' rD �test_searchsorted� s� && z TestMethods.test_searchsortedc s� t jddddddddd d d d d dgt jd�� t �t� ��}t� fdd�� D �|� t� fdd�� D �|d � t� fdd�|D �|� t� fdd�|D �|d � d S )Nz'P:\20x_dapi_cy3\20x_dapi_cy3_20100185_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100186_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100187_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100189_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100190_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100191_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100192_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100193_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100194_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100195_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100196_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100197_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100198_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100199_1)r) c s g | ]}� � |d ��qS )�left)r� )rq r� )rI r' rD rr s z9TestMethods.test_searchsorted_unicode.<locals>.<listcomp>c s g | ]}� � |d ��qS )�right)r� )rq r� )rI r' rD rr ! s r, c s g | ]}� � � | d ��qS )r� )r� )rq rv )rI r' rD rr " s c s g | ]}� � � | d ��qS )r� )r� )rq rv )rI r' rD rr # s )r/ r� r rH r� r )rJ r* r' )rI rD �test_searchsorted_unicode s( z%TestMethods.test_searchsorted_unicodec C s t �dddddg�}t �|�}ttt j|ddd� ttt j|dd gd� ttt j|dddddgd� ttt j|ddddddd gd� ttt j|ddddddgd� ttt j|dd ddddgd� ttt j|dddd ddgd� t j�d �}|�� }t � |�}t � ddd �}t |�|�|j||d�� t �dddddgd �}|�� }dddddg}dd dddg}t |j|d|d�|� d ddddg}t |j|d|d�|� t � d�}|� � }t j�|� |�� }t �|j|j d d�}|dd � �|j�}||d d �< |�|d|�}t ||� |�|d|�}t ||d � ||d d �< |�|d|�}t ||� |�|d|�}t ||d � d�t jd t jd t jd df�} x�| D ]�} | d k�r�d } | d k�r�t jddg| d �}t jddgt jd �}t �ddg�} nLt jdddddg| d �}t jdddddgt jd �}t jdddddgt jd �} |�|d|�}t || � |�|d|�}t || d � �q�W t �dddddg�}t jd t jd �} d | dd d�< dddddg| d d d�< | d d d� }t jdddddgt jd �} |�|d|�}t || � |�|d|�}t || d � d S )!NrL r& r, r r[ r )r, )r& r )�sorterg�������?r{ r. i, ry �( rz r r� )r� r� r r� rG r; r� r� r r� z?Or� zM8[D]r� )r) )rG )r/ r� r� r r� r� r1 r� �randr� �linspacer rH r� r� r: r9 rA ru r) r� r rj r ) rJ rI ru r^ r r� �keysrX r� r r r� Zsrtr' r' rD � test_searchsorted_with_sorter% s~ z)TestMethods.test_searchsorted_with_sorterc C s� G dd� dt j�}t �d��|�}t �dd��|�}t �d��|�}tt|�|d�|� � tt|�|d�|� � tt|�|d|�|� � tt|�|d|�|� � d S )Nc @ s e Zd ZdS )z4TestMethods.test_searchsorted_return_type.<locals>.AN)r` ra rb r' r' r' rD r/ y s r/ rL r, r r� r� )r/ r= rH ru r r� r� )rJ r/ rI r^ ru r' r' rD � test_searchsorted_return_typew s z)TestMethods.test_searchsorted_return_typec C sT t �d�}tt|jd� tt|jd� t jdtd�}tt|jd� tt|jd� d S )NrG i����)r) )r/ rH r r1 � argpartitionr0 )rJ r� �d_objr' r' rD � test_argpartition_out_of_range� s z*TestMethods.test_argpartition_out_of_rangec C sT t �d�}tt|jd� tt|jd� t jdtd�}tt|jd� tt|jd� d S )NrG i����)r) )r/ rH r r1 � partitionr0 )rJ r� r� r' r' rD �test_partition_out_of_range� s z'TestMethods.test_partition_out_of_rangec C s8 t �d�}tt|jd� t jdtd�}tt|jd� d S )NrG g "@)r) )r/ rH r r� r� r0 )rJ r� r� r' r' rD �test_argpartition_integer� s z%TestMethods.test_argpartition_integerc C s8 t �d�}tt|jd� t jdtd�}tt|jd� d S )NrG g "@)r) )r/ rH r r� r� r0 )rJ r� r� r' r' rD �test_partition_integer� s z"TestMethods.test_partition_integerc C sp t �g �}d|_x:t|j |j�D ]&}d�|�}tt j|d|d�||� q"W d}tt j|dd d�|�� |� d S )N)r r& r, r z(test empty array partition with axis={0}r )r� z)test empty array partition with axis=None) r/ r� r? rp r� r� r r� r� )rJ rI r� r r' r' rD �test_partition_empty_array� s z&TestMethods.test_partition_empty_arrayc C s� t �g �}d|_xFt|j |j�D ]2}d�|�}tt j|d|d�t j|t j d�|� q"W d}tt j|dd d�t j|� � t j d�|� d S )N)r r& r, r z+test empty array argpartition with axis={0}r )r� )r) z,test empty array argpartition with axis=None) r/ r� r? rp r� r� r r� r r r� )rJ rI r� r r' r' rD � test_argpartition_empty_array� s z)TestMethods.test_argpartition_empty_arrayc C s, t �d�}ttt j|ddd� ttt j|ddd� ttt j|ddd� tt|jdddd� tt|jdddd� � x�dD � ]�}t �g �}tt j|d|d�|� tt j|d|d�|� t � d�}tt j|d|d�d |� t|t j|d|d� t j|d|d�� t �d d d g�}|� � }t �t �d �|� t||� x�ddgddgddgfD ]�}t �|�}t � |�}tt j|d|d�d |d � tt j|d|d�d |d � t|t j|d|d� t j|d|d�� t|t j|d|d� t j|d|d�� x*t |j �D ] }||d � jd|d� �q�W t||� �qHW �xfd ddgddd gddd gdd dgdddgdddgdddgdddgfD �] }t �|�}t � |�}tt j|d|d�d |d � tt j|d|d�d |d � tt j|d|d�d |d � t|t j|d|d� t j|d|d�� t|t j|d|d� t j|d|d�� t|t j|d|d� t j|d|d�� x*t |j �D ] }||d � jd|d� �q^W t||� �qlW t � d�}tt j|d|d�|� t|t j|d|d� t j|d|d�� t �d�}| �t j|d |d�d d � | �t j|d |d�d d � t|t j|d |d� t j|d |d�� t|t j|d |d� t j|d |d�� t �d�d d d� }| �t j|d|d�d d� | �t j|d|d�d d� t|t j|d|d� t j|d|d�� t|t j|d|d� t j|d|d�� tt j|d|d�t j|d|d�� tt j|d|d�t j|d|d�� t|t j|d|d� t j|d|d�� t �d�}t �||j d �}|j d d } tt �|| �| | � t �d�}t �||j d d �}|j d d } tt �|| �| | � t � d�}d|d< tt �|d�d d� tt �|d�d d� t|t �|d� d d� t|t �|d� d d� t j|d< tt �|t �|d� d �� tt �t �|d�d �� t �d�d }t � t �d�d �}t j�|� x4t |j �D ]&}| �t j|||d�| || � �q�W t|t j|d|d� t j|d|d�� t|t j|d|d� t j|d|d�� x*t |j �D ] }||d � jd|d� �qXW t||� t �dddd dd d d d d d d d d d d d d d d d d d g�}dd d d g}tt j|||d�| d � t|t j|||d� | d � t �ddg�}|jd|d� tt|jd� tt j|jd dd!� ttt j|d� tt jt j|ddd!� tt|jd� tt j|jd dd!� ttt j|d� tt jt j|ddd!� t �d��d"�}|jdd|d� |jdd|d� t j|dd|d� t j|dd|d� t j|dd |d� t j|d d |d� |jdd|d� |jdd|d� t j|dd|d� t j|dd|d� t j|dd |d� t j|d d |d� tt|jddd!� tt|jd#dd!� tt|jdd d!� ttt j|d dd!� ttt j|d#d d!� tt|jddd!� tt|jd#dd!� ttt j|d dd!� ttt j|d#d d!� d$d%� t jt jt jgD �} �x�| D �]�\} } t} | j}t j| | d&�}t j�|� t � t j| | d&�d'�}t t jj|� t � |�}�xht |j �D �]X}t j|||d�}| �|| |� t |d |� || � t || ||d d � � | ||t j|||d� � t j||d|d�}| |d d �|f t j|g|j d | d&�� ||d d �d |�f j!|d d �|f k�"� d(||d d �|f |d d �d |�f j!f d)� ||d d �|d d �f j!|d d �|f k�"� d*||d d �|f |d d �|d d �f j!f d)� | ||t �|j d �d d �d f t j||d|d�f � t j||d|d�}| ||d d �f t j|g|j d | d&�� ||d |�d d �f ||d d �f k�"� d(|||d d �f |d |�d d �f f d)� |||d d �d d �f ||d d �f k�"� d*|||d d �f |d d �|d d �f f d)� | ||t j||d|d�t �|j d �d d d �f f � |� � }|j||d� t|t j|||d�� |� � }|j|d|d� t|t j||d|d�� |� � }|j|d|d� t|t j||d|d�� � q�W � qVW qvW d S )+NrG r& r, )r Znonsenser )r� r )� introselectr� � rL r� r r� �1 �/ r. r{ re i�����) i�� i@B iAB r[ )r& r. r� r� � ry )r r r� r� )r� )r& rL r� c S s g | ]}d D ] }||f�q qS ))r� re r' )rq r ru r' r' rD rr ^ s z.TestMethods.test_partition.<locals>.<listcomp>)r) )r[ r, z %d: %r <= %r)r z %d: %r < %r)#r/ rH r r� r� r1 r� r� r r� r� r� rp rA r� �rollr r� r �isnanr� r� � AxisErrorr} rl r� r� r� �tile�mapr6 r r? r� �all)rJ r� r �kthZokthr� r� rv r� r� �tdr ru Zaae�atr� �d0�pZdcr' r' rD �test_partition� s` $* & & ,*0.4 ,(.,2"zTestMethods.test_partitionc C sx d}xnt �|�D ]`}t|||� || d| d� t||d � || k�� d|||d � || f d� |d }qW d S )Nr zkth %d)rh z kth %d, %r not greater equal %d)r r, )r/ r� r r r )rJ r� r �prevr r' r' rD �assert_partitioned� s z TestMethods.assert_partitionedc C s� t �d�}d}tt|j|� tt|j|� t �d��d�}tt|j|dd� tt|j|dd� ttt j||dd� ttt j||d d� t �dd d dg�}t �|d �}| �|d � | �|t �|d � d � t |t �|d �� t ||t �|d � � t �d�}t j � |� |�t |j �� t t �d�|� t j � |� t t �d�||�t |j �� � t �d�}t j � |� t �ddd dg�}t j � |� t �||�}| �||� |t �||� }| �||� t j � |� t t �||�|� t |t �||� |� t �d�d d d� }| �t �|dgd �dg� | �t �|dgd ddg �dgd ddg � | �|t �|dgd � dg� | �|t �|dgd ddg � dgd ddg � t �d�}t j � |� t �t �d�d�}tt j j |� t �|�}d}t j||dd�}|t �|jd �d d �d f |j|dd�f }t ||� x0t |jd �D ] }| �||d d �f |� �q>W t j||dd�}|t j||dd�t �|jd �d d d �f f }t ||� x0t |jd �D ] }| �|d d �|f |� �q�W d S )Nrz )r r, r& i� �� rG )r& rL r )r� r, r r[ r& )r r )r� r. rd r� ry r. rL r{ r r� )r[ r, )r, r{ r� r. )r/ rH r r1 r� r� r} r� r r r� r� rp rA r r r6 r? ) rJ r� r r r� r� r �parv r' r' rD �test_partition_iterative� sr z$TestMethods.test_partition_iterativec C sR t jdddgdddgd�}t j|dd gd �}tt j|t|j�dd gd �|� t|t j|t|j�dd gd � |� xZt|j�D ]L}tt j||dd gd �| || � t|t j||dd gd � | || � q|W t �d d d dg�}t �|�}tt �|t|j��|� xLt|j�D ]>}tt �||�| || � t|t �||� | || � �q W d S )N)�Galahadg333333�?�& )�Arthurg������?r )�Lancelotgffffff�?r )r� z|S10)�heightz<f8)�agez<i4)r) r! r )r* r r Zzebrar ) r/ r� r� r r� rp rA r� r )rJ r� r� r r' r' rD �test_partition_cdtype� s* z!TestMethods.test_partition_cdtypec C s< t �d�}d�d�}tt|jd|d� tt|jd|d� d S )NrG s är� r& )r )r/ rH r� r r1 r� r� )rJ r� r r' r' rD �test_partition_unicode_kind� s z'TestMethods.test_partition_unicode_kindc C s� x�t dd�D ]�}x�t d|d �D ]z}t�|�}tj�|� |tj�dd� }tj�|j�}d|||d g}t�|�| }tt� ||�| |d||f d� q W q W d S )NrG r� r, r& r zdata: %r kth: %r)rh ) rp r/ rH r� r� �randintrA r� r r� )rJ r? rv r� �idxr r� r' r' rD �test_partition_fuzz� s z TestMethods.test_partition_fuzzc C s8 ddddddg}t �|d�}| �t �|�| dg� d S )Nr{ r� r r& r� r r, )r/ r� r r� )rJ r� r r' r' rD �test_argpartition_gh5524 s z$TestMethods.test_argpartition_gh5524c C s$ t �dddgdddggt j�}t �ddgddggddgddgggt j�}t �ddddddgt j�}t �ddddddgt j�}t �ddddddddgt j�}t �ddddddddgt j�}t|�� |� t|�d �|� t|�d �|j�� � t|�� |� t|�d �|� t|�d �|j�� � d S ) Nr, r& r r[ rL r{ r� rd r� )r/ r� rl r �flattenr� )rJ Zx0r� Zy0Zy0f�y1Zy1fr' r' rD � test_flatten s * zTestMethods.test_flattenc C sr t �ddgddgg�}t �ddgddgg�}t �ddgddgg�}t �d��dd�}t �dd d d gd d d dgd d ddgd dddgg�}t �ddddddgdddddd gddd d d d gddd d!d"d#gddd d"d$d%gdd d d#d%d&gg�}x�t jt jt jt jgD ]�}|�|�}t t � ||�|� t t � |j |�|� t t � ||j �|� t t � |j |j �|� t t � |j � � |�|� t t � ||j � � �|� t t � |j � � |j � � �|� q�W x�t jt jt jt jgD ]j}|�|�}|�|�} t t � | | �|� t t � | j | �|� t t � | | j �|� t t � | j | j �|� �q�W �xFt jt jt jt jgD �],}|�|�} t t � | d d d'�d d �f | j �t � | d d d'�d d �f � � | j � � �� t t � | d d �d d d'�f | j �t � | d d �d d d'�f � � | j � � �� t t � | | d d d'�d d �f j �t � | | d d d'�d d �f j � � �� t t � | | d d �d d d'�f j �t � | | d d �d d d'�f j � � �� t t � | d | j d d( �d d �f | d d d(�d d �f j �t � | d | j d d( �d d �f � � | d d d(�d d �f j � � �� t t � | d d d(�d d �f | d | j d d( �d d �f j �t � | d d d(�d d �f � � | d | j d d( �d d �f j � � �� �qFW xdt jt jt jt jgD ]L}|�|�} |�|�} |�|�} t t � | | j �| � t t � | j | �| � �q�W t t � ||�|� |�� t t � t � ||�|�|� |�� |�� t �|�}|� ||� t |t � ||�� t �|�}|j ||d)� t |t � ||�� d S )*Nr, r r� i����r� r[ r{ �7 � �� iE i� i� i' i� i i� i� i i@ id i� i� iD il i� i� i� i� i� i� i i� i$ iT iX i� i� r. r& )r^ r� )r/ r� rH r} r� r r� r� rI r �dotr� r� r? r ) rJ rI r^ r r� Zddt�dtd�etZeafZebfZedfZeddtfZedtdfr' r' rD �test_dot s� " * * & &8D8J & zTestMethods.test_dotc C s6 d}t jddd�}ttt j||� ttt j||� d S )Ng �?)r, r, zi,i)r) )r/ r� r r� r. )rJ r r/ r' r' rD �test_dot_type_mismatchx s z"TestMethods.test_dot_type_mismatchc C s� t j�d� dd� t jd D �}x�|D ]�}t j�dd��|�}td|d�}t j�dd�|d< t �||�}t j|||d �}t||t |�d � t t t j|||d d d � d � t t t j|||j d � q&W d S ) Nr, c S s g | ]}|d krt �|��qS )ZUSVM)r/ r) )rq �coder' r' rD rr � s z8TestMethods.test_dot_out_mem_overlap.<locals>.<listcomp>r r )r r )r) .)r� )rh r& ) r/ r� �seedr r� rI rE r. r r� r r1 r� )rJ �dtypesr) rI r^ r� r� r' r' rD �test_dot_out_mem_overlap s z$TestMethods.test_dot_out_mem_overlapc C sn t �d��d�}t|�� dddg� t|�d�dddg� t|�d�dddg� t|�d �d d g� t �d ��d �}t|�� ddgddgg� t|�d�ddgddgg� t|�d�dgdgg� t|�d �d gdgg� tt|jddd� t|�ddd�ddgd dgg� t|�ddd�ddgddgg� t|jdddd�dgdgg� t|�ddd�ddgd dgg� d S )Nr� )r r[ r rL rG r, r{ r� r. r[ r� rd )r& r& r& r� r& r )�axis1�axis2)rC r7 r8 )r/ rH r} r �diagonalr r1 )rJ rI r^ r' r' rD � test_diagonal� s zTestMethods.test_diagonalc C s� t �d��� }t|jj � t|jj � t �t �d��}t|jj � t|jj � t �t �d��}t|jj � t|jj � d S )Nr )r/ �eyer9 r rN rO rW �diag)rJ rI r' r' rD � test_diagonal_view_notwriteable� s z+TestMethods.test_diagonal_view_notwriteablec C sT t �d�}tr tt�|�dk � xtd�D ] }|�� q*W trPtt�|�dk � d S )N)r r r� r )r/ r2 r r r� � getrefcountrp r9 )rJ rI rv r' r' rD �test_diagonal_memleak� s z!TestMethods.test_diagonal_memleakc C sV t jg t jd�}t �d�}x td�D ]}t j|||d� q$W trRtt�|�dk � d S )N)r) g @r )r� r� ) r/ r� r rp r. r r r� r> )rJ rI r� �_r' r' rD �test_size_zero_memleak� s z"TestMethods.test_size_zero_memleakc C s t �d��d�}t|�� d� t|�d�d� t|�d�d� t|�d�d� t �d ��d �}t|�� d d g� t|�d�d d g� t|�d�d d g� t|�d�ddg� t|�ddd�d d g� t|�ddd �ddg� t|�ddd �d dg� t|jddd d�dd g� d S )Nr� )r r[ r r r, r} r. r rd )r& r& r& r{ r& r r[ rL r� r� )rC r7 r8 )r/ rH r} r �trace)rJ rI r^ r' r' rD � test_trace� s zTestMethods.test_tracec C sB G dd� dt j�}t �d��d��|�}|�� }t||�s>t�d S )Nc @ s e Zd ZdS )z0TestMethods.test_trace_subclass.<locals>.MyArrayN)r` ra rb r' r' r' rD �MyArray� s rD rd )r& r& r& )r/ r= rH r} ru rB r� �AssertionError)rJ rD r^ r� r' r' rD �test_trace_subclass� s z TestMethods.test_trace_subclassc C s� t jd }t jd }x�|| d D ]�}t jddddddg|d�}t jd |d�}|�dddgdddg� t||� t jd |d�}|�dddgdddg� t||�d d�� q"W x�d D ]�}t jd dd dd dg|d�}t jd |d�}|�dddgdgd � t||� t jd |d�}|�dddgdgd � t||�d d�� q�W t �d �}d |j_t t |jdddgdddg� dd dg}t t t j|dd gd� d S )Nr� r rM r r, r rL )r) r{ )r& r r& r� FT) r/ r r� r2 �putr r} rN rO r r1 r� )rJ �icodes�fcodesr r� rI Z bad_arrayr' r' rD �test_put� s. zTestMethods.test_putc C s� t �ddgddgg�}t|�� ddddg� t|�� jj � t|�d�ddddg� t|jdd�ddddg� t|jdd�ddddg� t|jdd�ddddg� t|jdd�jj � t|jd d�ddddg� t|jd d�jj � t|�� |�d �� t jddgddggdd�}t|�� ddddg� t|jdd�ddddg� t|jd d�ddddg� t|jdd�jj � t|jd d�jj � t|�� |�d �� t|jdd�|jd dd�� t �ddgddgg�d d d �d d �f }t|�� ddddg� t|jdd�ddddg� t|jdd�ddddg� t|jdd�ddddg� t|jd d�ddddg� t|jd d�jj� t �d �d d d� }t|�d �jj� t|�d�jj� t|�d�jj� t �d �d d d� }|�dddd�� d d �}t |j �}d|d< ||_ t|jd d�jj� t|�d �t �ddd�� t �d�}|�dddd�� d d �}t |j �}d|d< ||_ tt � |jd d�|�� t|jd d�t �d�� t �d�d d d � �dd�}t|jdd�jj� t|jd d�jj� t|�d�ddddg� t|�d �ddddg� t �dgg�}d|_ t � d�j dk�rftt � |�d �|�� t|�d �j |jjf� x@dD ]8}t �d�}t|�|�dg� tt � |�|�|�� �qlW t �d�d d d� �dddd�}|dd d d�f }t|�d �ddddd dddg� t|�d�ddddd dddg� t|�d�ddddd dddg� t|�d�dd ddddddg� |d d d�df }t|�d �dddddd dd g� t|�d�dddddd dd g� t|�d�dddddd dd g� t|�d�dddddd dd g� d S ) Nr r, r& r r� r( )r* r/ �Kr. rG re r� r� r rd r[ )r� i� )rd )r( r� r/ rK rf .r� ry r� � r{ r� )r/ r� r r� r rN rW r} rH �swapaxesr� r� �may_share_memoryr� r) r9 )rJ rI r� r* r^ r' r' rD � test_ravel s� ( zTestMethods.test_ravelc C s� G dd� dt j�}t �d��|�}tt|�d�|�� tt|�d�|�� tt|�d�|�� tt|�d�|�� t �d�d d d� �|�}tt|�d�|�� tt|�d�|�� tt|�d�|�� tt|�d�|�� d S ) Nc @ s e Zd ZdS )z6TestMethods.test_ravel_subclass.<locals>.ArraySubclassN)r` ra rb r' r' r' rD � ArraySubclassk s rP rG r( r� r/ rK r& )r/ r= rH ru r r� r� )rJ rP rI r' r' rD �test_ravel_subclassj s z TestMethods.test_ravel_subclassc C s� t �d��dddd��� }t �|j�}t|jd � |�� }tt |j dd� tt |j dd� tt |j dd� tt |j dd� �x8t d d�D �](}�x t d d�D �]}�xt ||f�D ]�\}}|� ||�}t |j�} |j| | |< |j| | |< t|j| t|||f�� d d � |jD �\} } } } d d � |jD �\}}}}t||| || || || f ||| || || || f t|||f�� t|jd t|||f�� |dkr�|}q�W q�W q�W d S ) Nr� r, r& r r[ �OWNDATA���r ���c S s g | ] }|d �qS )r, r' )rq �dimr' r' rD rr � s z-TestMethods.test_swapaxes.<locals>.<listcomp>c S s g | ] }|d �qS )r, r' )rq rU r' r' rD rr � s )r/ rH r} r� �indicesr? r rN r r1 rM rp � enumerater� r r� )rJ rI r% r^ rv r? r �srcr r? �i0r9 �i2Zi3Zj0�j1�j2Zj3r' r' rD � test_swapaxesz s0 zTestMethods.test_swapaxesc s� t �dddg�� � �� }t� j|j� t� j|j � t|� �� � t|t �� �� t �dddgd�� � �� }t� j|j� t� j|j � t|� �� � t|t �� �� t �dddg�� � �� }t� |� t|� �� � t|t �� �� t �dd d g�� � �� }t� |� t|� �� � t|t �� �� t �dddd gt�� � �� }t|d d � � D �� t|� �� � t|t �� �� t �ddd d gt�� tt � fdd�� tt � fdd�� d S )Ny �? �y �? �?y 7@ 7@r� r, r& r g �?g @g @c S s g | ] }|� � �qS r' )� conjugate)rq r r' r' rD rr � s z.TestMethods.test_conjugate.<locals>.<listcomp>r� c s � � � S )N)�conjr' )rI r' rD r� � r� z,TestMethods.test_conjugate.<locals>.<lambda>c s � � � S )N)r^ r' )rI r' rD r� � r� ) r/ r� r_ r r� r� r^ r0 r r� )rJ Zacr' )rI rD �test_conjugate� s< zTestMethods.test_conjugatec C s� ddddddddd d d d d dddg}x�|D ]�}t jd|d�}t jdg|d�}t jdggggg|d�}d�|�}t|�}t|||� t|�}t|||� t|�} t| ||� q*W d S )Nr9 rZ r] �i8rY �u2�u4�u8r� r� �gr� �D�Gr� rM r� )r) z dtype: {0})r/ r� r� r r ) rJ r5 r rI r^ r r �ap�bp�cpr' r' rD �test__complex__� s zTestMethods.test__complex__c C s� ddddddddd d d d d dddg}x,|D ]$}t jdddg|d�}ttt|� q*W t �ddg�}t jd|d�}ttt|� t jddg|d�}ttt|� t �d�}ttt|� t �dgd�}ttt|� d S )Nr9 rZ r] ra rY rb rc rd r� r� re r� rf rg r� rM r, r& r )r) )rI r� )r^ r9 )g �?r )g����Mb`?r� z1+1jr )r/ r� r r� r r) )rJ r5 r rI r^ r r� r� r' r' rD � test__complex__should_not_work� s z*TestMethods.test__complex__should_not_workN)5r` ra rb 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* r1 r2 r6 r: r= r? rA rC rF rJ rO rQ r] r` rk rl r' r' r' rD r� � sj 3 " ]R QD [ ([ %r� c @ s e Zd Zdd� ZdS )�TestCequenceMethodsc C s8 t dt�d��dd�k� t dt�d��dd�k� d S )Ng @g 0@r[ g 4@)r r/ rH r} )rJ r' r' rD �test_array_contains� s z'TestCequenceMethods.test_array_containsN)r` ra rb rn r' r' r' rD rm � s rm c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )� TestBinopc C s t t�dg�t�ddg� ddg� t�ddg�d d d� }t ||t�ddg� ddg� t�dg�}t�dg�}|| }|| }|| }|| }t||� t|d� ||d | | || }t||� t|d� t�dg�}t�dg�}|| | }t|d� t|d� t|d� d S ) Ng �?g �?g @r& g �?rL r g������ @)r r/ r� r r )rJ r� rI r^ r r' r' rD � test_inplace� s, zTestBinop.test_inplacec s� t jdtft jdtft jdtft jdtft jdtft jdtft jdtft j dt ft j dt ft j dt ft j dt ft jdt ft jdt fd� �G dd� dt�� � fdd��dd � �d d � �d d � �dd� }d�����fdd� }d � �fdd� }||t�ddd� ||tdd�ddd� ||tdd�ddd� ||t jdd�ddddd� ||td|d�ddd� ||t jd|d�ddd� ||td d�ddd� ||t jd d�ddddd� d S ) NTF) r� �subr8 �truediv�floordiv�mod�divmod�pow�lshift�rshift�and�xor�orc @ s e Zd ZdS )z7TestBinop.test_ufunc_binop_interaction.<locals>.CoercedN)r` ra rb r' r' r' rD �Coerced+ s r| c s � �d S )Nr' )rJ )r| r' rD � array_impl. s z:TestBinop.test_ufunc_binop_interaction.<locals>.array_implc S s dS )N�forwardr' )rJ r� r' r' rD �op_impl1 s z7TestBinop.test_ufunc_binop_interaction.<locals>.op_implc S s dS )N�reverser' )rJ r� r' r' rD �rop_impl4 s z8TestBinop.test_ufunc_binop_interaction.<locals>.rop_implc S s dS )Nzin-placer' )rJ r� r' r' rD �iop_impl7 s z8TestBinop.test_ufunc_binop_interaction.<locals>.iop_implc _ s d||||fS )N�__array_ufunc__r' )rJ �ufunc�methodr� r� r' r' rD �array_ufunc_impl: s z@TestBinop.test_ufunc_binop_interaction.<locals>.array_ufunc_impl�__main__c s� d� i}|dk r||d< x6�D ].}�|d� |�< �|d� |�< �|d� |�< q W |dk r`||d< | ||d�}td |�}t|tj�r�t�d d ��|�S |� S d S ) Nr� F�__array_priority__z__{0}__z__r{0}__z__i{0}__r� )r� �class_namespacer` z(type('MyType', (base,), class_namespace)r rL )r� r\ � issubclassr/ r= rH ru )r� �array_priority� array_ufuncZalleged_moduler� �opZeval_namespaceZMyType)r} r� r �opsr� r' rD �make_obj@ s" z8TestBinop.test_ufunc_binop_interaction.<locals>.make_objc s< �x4�� � D �]&\� \}}}d� |||f }tjdd|d�g} |rT| �| d d � �x�| D �]�} t| d�� ��} � fdd�} |r�t| | �t|� nl|r�t| | �d d |� nRt| tj �r�t | �j tj j kr�| | | ��} t | j | j k|� ntt�f| | |d � t| d �� ��}|�rD|| �} t| d d |d � t| d ||d � nVt| tj ��r�t | �j tj j k�r�| || ��} t | j | j k|� ntt�f|| |d � |�r�t| tj ��r�t| d �� ��}|�r�t| | �t|d � n�|�r<|| �} t| d d |� t| d ||� t t | d d �tk|� t | d d d | k|� nJt| tj ��rrt | �j tj j k�rrt || �| k|� ntt�f|| |d � tt� d �}|d k�r�tt� d d �}|d k�r�tt� �}t|| | �d|� t| tj ��s|�r�t|| | �d|� n |�rt|| | �d d |� |r\t|| | �d d |� q\W q W d S )Nz-op: %s, ufunc: %s, has_inplace: %s, dtype: %sr rL )r) r z__{0}__c s& � dkr t t| t�� | d S | S d S )Nru r )r r� �tuple)r )r� r' rD � first_out_argc s zLTestBinop.test_ufunc_binop_interaction.<locals>.check.<locals>.first_out_argr� )rh z__r{0}__r, z__i{0}__r. r� r@ r~ r� )�itemsr/ rH �append�getattrr� r �NotImplementedr� r= r� r� r � __class__r r� r� r7 rt )�objZbinop_override_expectedZufunc_override_expectedZinplace_override_expected� check_scalarr� Z has_inplacer) rh Z check_objsr� Z arr_methodr� r� Z arr_rmethodZ arr_imethodZop_fn)r| r� )r� rD �checkX s� z5TestBinop.test_ufunc_binop_interaction.<locals>.checki �)r� r, )r� )r� r� )r� )FFr� )T)r/ r� r� �subtract�multiply� true_divide� floor_divide� remainderru rv r� � left_shift� right_shift� bitwise_and� bitwise_xor� bitwise_orr� r0 r= )rJ r� r� r� r' )r| r} r� r r� r� rD � test_ufunc_binop_interaction sD ] z&TestBinop.test_ufunc_binop_interactionc C s� G dd� dt �}|� }t�|dg�}td|ko4d|k� tj|dgdd�}td|koZd|k� t|d d� tj|dgdd�}td|ko�d|k� t|d d� d S ) Nc @ s e Zd Zdd� ZdS )zDTestBinop.test_ufunc_override_normalize_signature.<locals>.SomeClassc _ s |S )Nr' )rJ r� r� �inputs�kwr' r' rD r� � s zTTestBinop.test_ufunc_override_normalize_signature.<locals>.SomeClass.__array_ufunc__N)r` ra rb r� r' r' r' rD � SomeClass� s r� r, �sig� signaturezii->i)r� )r� )r0 r/ r� r r )rJ r� rI r� r' r' rD �'test_ufunc_override_normalize_signature� s z1TestBinop.test_ufunc_override_normalize_signaturec C sN G dd� dt �}|� }t�d�}tt�|�d� tt�||�d� ttj||d�d� ttj||fd�d� tt�||�d� ttj||d�d� ttj||fd�d� tt�||�d� tt�|d |�d� tt�|||�d� ttj||d fd�d� ttj|||fd�d� ttj|d |fd�d� ttj|||fd�d� ttj|||fd�d� tjdd ��:}t�d d t � ttj||d�d� t |d j t k� W d Q R X t t tj||fd� tt�||�d� tt�||�d � tt�|||�d� tt�|||�d � ttj|||d�d� ttj|||fd�d� ttj|||d�d� d S ) Nc @ s e Zd Zdd� ZdS )z4TestBinop.test_array_ufunc_index.<locals>.CheckIndexc _ sN x t |�D ]\}}|| kr |S q W x&t |d �D ]\}}|| kr0|fS q0W d S )Nr� )rW )rJ r� r� r� r� rv rI r? r' r' rD r� � s zDTestBinop.test_array_ufunc_index.<locals>.CheckIndex.__array_ufunc__N)r` ra rb r� r' r' r' rD � CheckIndex� s r� g @r )r )r� )r, T)�record�alwaysr� r, )r0 r/ rH r �sin�modf�warnings�catch_warnings�filterwarnings�DeprecationWarningr �categoryr r1 r� )rJ r� rI �dummy�wr' r' rD �test_array_ufunc_index� s< z TestBinop.test_array_ufunc_indexc C s� G dd� dt j�}t �dg��|�}t �dg�}t �dg�}t �|||� t|d d� tt||�� d|d< t j|||d� t|d d� tt||�� d S )Nc @ s e Zd Zdd� ZdS )z-TestBinop.test_out_override.<locals>.OutClassc _ s> d|kr:|� � }|�d� t||�}|||�|d d d< d S )Nr� r .)r� �popr� )rJ r� r� r� r� Ztmp_kw�funcr' r' rD r� s z=TestBinop.test_out_override.<locals>.OutClass.__array_ufunc__N)r` ra rb r� r' r' r' rD �OutClass s r� r rL r{ r� )r� )r/ r= r� ru r� r r r� )rJ r� r/ rN r( r' r' rD �test_out_override s zTestBinop.test_out_overridec C s� G dd� dt j�}t jdt jd��|�}t|d d� tt�� |d W d Q R X tt�� |d W d Q R X tt�� |d W d Q R X tt�� |d W d Q R X tt�� |d W d Q R X d S ) Nc @ s e Zd Zdd� ZdS )z:TestBinop.test_pow_override_with_errors.<locals>.PowerOnlyc _ s |t jk rt�dS )NzPOWER!)r/ rv �NotImplementedError)rJ r� r� r� r� r' r' rD r� ! s zJTestBinop.test_pow_override_with_errors.<locals>.PowerOnly.__array_ufunc__N)r` ra rb r� r' r' r' rD � PowerOnly s r� g @)r) g @zPOWER!g �?r r, r. r& )r/ r= r� r ru r r r� )rJ r� rI r' r' rD � test_pow_override_with_errors s z'TestBinop.test_pow_override_with_errorsN) r` ra rb rp r� r� r� r� r� r' r' r' rD ro � s % 80ro c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�TestTemporaryElidec C sF ddl m} t�d�}||�\}}|| t||� t||| � d S )Nr )� incref_elidei�� )r� r� r/ r� r )rJ r� r� �origr� r' r' rD �test_extension_incref_elide7 s z.TestTemporaryElide.test_extension_incref_elidec C sV ddl m} ddddt�d�g}||�}t|d t�d�� t||d |d � d S )Nr )�incref_elide_lr, i�� r[ )r� r� r/ r� r )rJ r� r� r� r' r' rD �!test_extension_incref_elide_stackF s z4TestTemporaryElide.test_extension_incref_elide_stackc C s| t jdt jd�}t|| dd jt �d�� || d }t|jt �d�� t �|| d�}t|jt �d�� || d }t|jt �d�� || d }t|jt �t j�� t jdt jd�}t|| |�t j� jt �d�� |�t j�}t|| | j|j� t jdt j d�}t|| | j|j� xPt j t j t j fD ]<}t jd|d�}t |d �}t|jt �d |jd �� �q8W d S ) Ni@ )r) r& �� rM r� g @i�� zf%d)r/ r� rn r r) r� r� rI r r� r� r� � clongdouble�absr9 )rJ r� r� r� r� r r r' r' rD �test_temporary_with_castU s( " z+TestTemporaryElide.test_temporary_with_castc C sD t jdtd�}t jdt jd�}d| | }t|d� t|jd� d S )N)i� r, )r) i� r, )i� i� )r/ r� r� r� r r? )rJ r� r^ r� r' r' rD �test_elide_broadcastv s z'TestTemporaryElide.test_elide_broadcastc C s$ t �� }tt||@ �t jk� d S )N)r/ rh r r� )rJ rI r' r' rD �test_elide_scalar� s z$TestTemporaryElide.test_elide_scalarc C s t jdt jd�}|jd d S )Ni�� )r) r& )r/ r: r r� )rJ rI r' r' rD �test_elide_scalar_readonly� s z-TestTemporaryElide.test_elide_scalar_readonlyc C s, t �t �t �d�d�j�d }t|d� d S )Nr, i�� g r )r/ �asarray� broadcast_tor2 �flatr )rJ r� r' r' rD �test_elide_readonly� s z&TestTemporaryElide.test_elide_readonlyc C s2 t �d�d d d� }|j�� d }~t|d� d S )Ni r& r, )r/ r� r� r� r )rJ rI r^ r' r' rD �test_elide_updateifcopy� s z*TestTemporaryElide.test_elide_updateifcopyN) r` ra rb r� r� r� r� r� r� r� r� r' r' r' rD r� 4 s ! r� c @ s e Zd Zdd� ZdS )�TestCAPIc C sP ddl m} t|d�� t|d�� t|dd �� t|d�� t|d�� d S ) Nr )�IsPythonScalars foobarr, r& r g @rI )r� r� r )rJ r� r' r' rD �test_IsPythonScalar� s z TestCAPI.test_IsPythonScalarN)r` ra rb r� r' r' r' rD r� � s r� c @ s e Zd Zdd� ZdS )�TestSubscriptingc C sd t �dddg�}| �t|d t j�� tjd dk rH| �t|d t�� | �t|d �t j k� d S )Nr, r& r r )r .) r/ r� r� r� r� r� r� r� r� r= )rJ r� r' r' rD �test_test_zero_rank� s z$TestSubscripting.test_test_zero_rankN)r` ra rb r� r' r' r' rD r� � s r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )� TestPicklingc C s� dd l }t�ddgddgddgg�}|t�|�tjdgdtdffd tfd tfgd �g}x(|D ] }t||�|� � �d | d � q\W d S )Nr r& r� r� r rd )�xxxr, g @rI r^ r )r) z%r)rh ) �pickler/ r� r6 r� r� r� r �loads�dumps)rJ r� rR ZDATArI r' r' rD �test_roundtrip� s zTestPickling.test_roundtripc C s* t jd dkr tj|dd�S t�|�S d S )Nr r �latin1)�encoding)r� r� r/ r� )rJ r� r' r' rD �_loads� s zTestPickling._loadsc C s4 d}t jddddgt jd�}| �|�}t||� d S )Ns} �cnumpy.core._internal _reconstruct qcnumpy ndarray qK �Ub�Rq(K�cnumpy dtype qUi1K K�Rq(U|NNJ����J����tb�Utb.r, r& r r[ )r) )r/ r� ri r� r )rJ ru rI r r' r' rD �test_version0_int8� s z TestPickling.test_version0_int8c C s4 d}t jddddgt jd�}| �|�}t||� d S )Ns� �cnumpy.core._internal _reconstruct qcnumpy ndarray qK �Ub�Rq(K�cnumpy dtype qUf4K K�Rq(U<NNJ����J����tb�U �? @ @@ �@tb.g �?g @g @g @)r) )r/ r� r� r� r )rJ ru rI r r' r' rD �test_version0_float32� s z"TestPickling.test_version0_float32c C s2 d}t �ddiddig�}| �|�}t||� d S )Ns� �cnumpy.core._internal _reconstruct qcnumpy ndarray qK �Ub�Rq(K�cnumpy dtype qUO8K K�Rq(U|NNJ����J����tb�]q(}qUaKs}qUbKsetb.rI r, r^ r& )r/ r� r� r )rJ ru rI r r' r' rD �test_version0_object� s z!TestPickling.test_version0_objectc C s4 d}t jddddgt jd�}| �|�}t||� d S )Ns� �cnumpy.core._internal _reconstruct qcnumpy ndarray qK �Ub�Rq(KK�cnumpy dtype qUi1K K�Rq(KU|NNJ����J����tb�Utb.r, r& r r[ )r) )r/ r� ri r� r )rJ ru rI r r' r' rD �test_version1_int8� s z TestPickling.test_version1_int8c C s4 d}t jddddgt jd�}| �|�}t||� d S )Ns� �cnumpy.core._internal _reconstruct qcnumpy ndarray qK �Ub�Rq(KK�cnumpy dtype qUf4K K�Rq(KU<NNJ����J����tb�U �? @ @@ �@tb.g �?g @g @g @)r) )r/ r� r� r� r )rJ ru rI r r' r' rD �test_version1_float32� s z"TestPickling.test_version1_float32c C s2 d}t �ddiddig�}| �|�}t||� d S )Ns� �cnumpy.core._internal _reconstruct qcnumpy ndarray qK �Ub�Rq(KK�cnumpy dtype qUO8K K�Rq(KU|NNJ����J����tb�]q(}qUaKs}qUbKsetb.rI r, r^ r& )r/ r� r� r )rJ ru rI r r' r' rD �test_version1_object� s z!TestPickling.test_version1_objectc C s0 d}t jdgddgd�}| �|�}t||� d S )Ns� cnumpy.core.multiarray _reconstruct p0 (cnumpy ndarray p1 (I0 tp2 S'b' p3 tp4 Rp5 (I1 (I1 tp6 cnumpy dtype p7 (S'V6' p8 I0 I1 tp9 Rp10 (I3 S'|' p11 N(S'a' p12 g3 tp13 (dp14 g12 (g7 (S'V4' p15 I0 I1 tp16 Rp17 (I3 S'|' p18 (g7 (S'i1' p19 I0 I1 tp20 Rp21 (I3 S'|' p22 NNNI-1 I-1 I0 tp23 b(I2 I2 tp24 tp25 NNI4 I1 I0 tp26 bI0 tp27 sg3 (g7 (S'V2' p28 I0 I1 tp29 Rp30 (I3 S'|' p31 (g21 I2 tp32 NNI2 I1 I0 tp33 bI4 tp34 sI6 I1 I0 tp35 bI00 S'\x01\x01\x01\x01\x01\x02' p36 tp37 b.)r, )r, r& )rI r9 )r& r& )r^ r9 r& )r) )r/ r� r� r )rJ ru rI r r' r' rD �test_subarray_int_shape� s z$TestPickling.test_subarray_int_shapeN) r` ra rb r� r� r� r� r� r� r� r� r� r' r' r' rD r� � s r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestFancyIndexingc C sl t �d�}d|d d �dgf< t|t �dgg�� t �d�}d|d d �d d �dgf< t|t �dggg�� d S )N)r, r, g @r )r, r, r, )r/ r� r r� )rJ r� r' r' rD � test_list� s zTestFancyIndexing.test_listc C sh t �d�}d|d d �df< t|t �dgg�� t �d�}d|d d �d d �df< t|t �dggg�� d S )N)r, r, g @)r )r, r, r, )r/ r� r r� )rJ r� r' r' rD � test_tuple� s z TestFancyIndexing.test_tuplec C s@ t �ddddg�}t �ddddgt�}t|| t �dg�� d S )Nr, r& r r[ r )r/ r� r� r )rJ r� r� r' r' rD � test_mask s zTestFancyIndexing.test_maskc C s� t �ddddgddddgg�}t �d dgt�}t �d dd d gdd d d ggt�}t �d dd d gd d d d ggt�}t|| t �ddddgg�� t|| t �ddg�� t|| t �dg�� d S ) Nr, r& r r[ rL r{ r� rd r )r/ r� r� r )rJ r� r� �m2�m3r' r' rD � test_mask2 s z TestFancyIndexing.test_mask2c C sJ t �ddddg�}t �ddddgt�}d||< t|t �ddddg�� d S )Nr, r& r r[ r rL )r/ r� r� r )rJ r� r� r' r' rD �test_assign_mask s z"TestFancyIndexing.test_assign_maskc C s t �ddddgddddgg�}t �d dgt�}t �d dd d gdd d d ggt�}t �d dd d gd d d d ggt�}|�� }d ||< t|t �ddddgd d d d gg�� |�� }d ||< t|t �dd ddgd dddgg�� |�� }d ||< t|t �dd ddgddddgg�� d S ) Nr, r& r r[ rL r{ r� rd r rG )r/ r� r� r� r )rJ Zxorigr� r� r� r� r' r' rD �test_assign_mask2 s $$z#TestFancyIndexing.test_assign_mask2N) r` ra rb r� r� r� r� r� r� r' r' r' rD r� � s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestStringComparec s� t �dddg�� t �dddg��t� �k� �fdd�dD �� t� �k� �fdd�dD �� t� �k� �fd d�dD �� t� �k� �fd d�dD �� t� �k � �fd d�dD �� t� �k� �fd d�dD �� d S ) N�This�is�example�wasc s g | ]}� | �| k�qS r' r' )rq rv )�g1�g2r' rD rr ) s z1TestStringCompare.test_string.<locals>.<listcomp>)r r, r& c s g | ]}� | �| k�qS r' r' )rq rv )r� r� r' rD rr * s c s g | ]}� | �| k�qS r' r' )rq rv )r� r� r' rD rr + s c s g | ]}� | �| k�qS r' r' )rq rv )r� r� r' rD rr , s c s g | ]}� | �| k �qS r' r' )rq rv )r� r� r' rD rr - s c s g | ]}� | �| k�qS r' r' )rq rv )r� r� r' rD rr . s )r/ r� r )rJ r' )r� r� rD � test_string& s z TestStringCompare.test_stringc s� t �ddddg�}d� t|� k� fdd�|D �� t|� k� fdd�|D �� t|� k � fdd�|D �� t|� k� fd d�|D �� t|� k� fd d�|D �� t|� k� fd d�|D �� d S ) N�spamZspaZspammerzand eggsc s g | ] }|� k�qS r' r' )rq r� )r� r' rD rr 3 s z0TestStringCompare.test_mixed.<locals>.<listcomp>c s g | ] }|� k�qS r' r' )rq r� )r� r' rD rr 4 s c s g | ] }|� k �qS r' r' )rq r� )r� r' rD rr 5 s c s g | ] }|� k�qS r' r' )rq r� )r� r' rD rr 6 s c s g | ] }|� k�qS r' r' )rq r� )r� r' rD rr 7 s c s g | ] }|� k�qS r' r' )rq r� )r� r' rD rr 8 s )r/ r� r )rJ r� r' )r� rD � test_mixed0 s z TestStringCompare.test_mixedc s� t �dddg�� t �dddg��t� �k� �fdd�dD �� t� �k� �fdd�dD �� t� �k� �fd d�dD �� t� �k� �fd d�dD �� t� �k � �fd d�dD �� t� �k� �fd d�dD �� d S ) Nr� r� r� r� c s g | ]}� | �| k�qS r' r' )rq rv )r� r� r' rD rr = s z2TestStringCompare.test_unicode.<locals>.<listcomp>)r r, r& c s g | ]}� | �| k�qS r' r' )rq rv )r� r� r' rD rr > s c s g | ]}� | �| k�qS r' r' )rq rv )r� r� r' rD rr ? s c s g | ]}� | �| k�qS r' r' )rq rv )r� r� r' rD rr @ s c s g | ]}� | �| k �qS r' r' )rq rv )r� r� r' rD rr A s c s g | ]}� | �| k�qS r' r' )rq rv )r� r� r' rD rr B s )r/ r� r )rJ r' )r� r� rD � test_unicode: s z TestStringCompare.test_unicodeN)r` ra rb r� r� r� r' r' r' rD r� % s r� c ! @ s� e Zd Zddddejgdfdddejdgdfejddddgdfejdejddgdfddddedej�gdfddddeejd�gdfdddeejd�dgdfdddedej�dgdfedej�ddddgdfeejej�ddddgdfeejd�eejd�eejd�gdfeejej�eejd�eejd�gdfeejd�eejd�eejej�gdfedd�edd�edd�gdfedd�edd�edd�gdfedd�edd�edd�gdfe�d�e�d�e�d�e�d �e�d �e�d �gd fe�d �e�d�e�d�e�d�e�d�e�d�gdfe�d�e�d�e�d�e�d�e�d�e�d�gdfe�d�e�d�e�d�e�d�e�d�e�d�gdfe�dd�e�dd�e�dd�e�dd�gdfe�dd�gd dfed d d �edd d �ed d d �gdfedd!d �ed"d d �ed d d �gdfed"d#d �ed"d d �ed"d!d �gdfd$d$d$d$d%gdfd$d$d$d%d$gdfd%d$d$d$d$gdfd%d$d%d$d$gdfg Z d&d'� Z d(d)� Z d*d+� Z d,d-� Z d.d/� Zd0d1� Zd2S )3� TestArgmaxr r, r& r r[ z1923-04-14T12:43:12z1994-06-21T14:43:15z2001-10-15T04:10:32z1995-11-25T16:02:16z2005-01-04T03:14:12z2041-12-03T14:05:03rL z1935-09-14T04:40:11z1949-10-12T12:32:11z2010-01-03T05:14:12z2015-11-20T12:20:59z1932-09-23T10:10:13z2014-10-10T03:50:30�NaTz2015-09-23T10:10:13z1932-10-10T03:50:30z2059-03-14T12:43:12z1996-09-21T14:43:15z2022-12-25T16:02:16z1963-10-04T03:14:12z2013-05-08T18:15:23ru r� )�days�seconds�# r. � r: rG r� FTc C sx t j�ddd�}xbt|j�D ]T}|�|�}|�|�}tt|j��}|�|� t t � ||j |j |f|�� � k�� q W d S )Nr r, )r[ rL r{ r� rd ) r/ r� �normalrp r� rr �argmaxr� �remover r r� r6 )rJ rI rv �amaxZaargmax�axesr' r' rD �test_all� s zTestArgmax.test_allc C sR xL| j D ]B\}}tt�|�|d| d� t|t�|� t�|�d| d� qW d S )Nz%r)rh )�nan_arrr r/ r rr )rJ r� �posr' r' rD �test_combinations� s z TestArgmax.test_combinationsc C s� t �d�}t jdt jd�}tt|jd|� t jdt jd�}tt|jd|� t jdt jd�}tt|jd|� t jdt jd�}|jd|d� t||�d�� d S ) N)rG rL r� )r) r. )r& rL )r, rG rG )r� )r/ r� r� r r1 r r )rJ rI r� r' r' rD �test_output_shape� s z TestArgmax.test_output_shapec C s( t jddd�}d|d< t|�� d� d S )Ni� z<U9)r) �asi6 )r/ r2 r r )rJ r� r' r' rD �test_argmax_unicode� s z TestArgmax.test_argmax_unicodec C s� t jjdd�}t jdtd�}t jdtd�}t|�d|�t �|d|�� t||� t jdtd�}t jdtd�}t|j|dd�t j||dd�� t||� d S ) N)r& r )rA r& )r) r, r r )r� r� )r/ r� r r2 r� r r )rJ rI �out1�out2r' r' rD �test_np_vs_ndarray� s z TestArgmax.test_np_vs_ndarrayc C s` t jddd�}t�|jjd|j� t|�� d� d|d< t|�� d� d|d< t|�� d� d S ) Nr[ rM )r) r rG r r� r, )r/ r: �ctypes�memsetr- r� r r )rJ rI r' r' rD � test_object_argmax_with_NULLs� s z(TestArgmax.test_object_argmax_with_NULLsN)r` ra rb r/ r� r � datetime64� timedelta64r$ r r r r r r r r' r' r' rD r� E sz $&& r� c ! @ s� e Zd Zddddejgdfdddejdgdfejddddgdfejdejddgdfddddedej�gdfddddeejd�gdfdddeejd�dgdfdddedej�dgdfedej�ddddgdfeejej�ddddgdfeejd�eejd�eejd�gdfeejej�eejd�eejd�gdfeejd�eejd�eejej�gdfedd�edd�edd�gdfedd�edd�edd�gdfedd�edd�edd�gdfe�d�e�d�e�d�e�d �e�d �e�d �gdfe�d �e�d �e�d�e�d�e�d�e�d�gdfe�d�e�d�e�d�e�d�e�d�e�d�gdfe�d�e�d�e�d�e�d�e�d�e�d�gdfe�dd�e�dd�e�dd�e�dd�gdfe�dd�gd dfeddd�edd d�ed d d�gdfedd d�ed dd�eddd�gdfed d!d�ed dd�ed d d�gdfd"d"d"d"d#gdfd"d"d"d#d"gdfd#d"d"d"d"gdfd#d"d#d"d"gdfg Z d$d%� Z d&d'� Z d(d)� Z d*d+� Z d,d-� Zd.d/� Zd0d1� Zd2S )3� TestArgminr r, r& r r[ z1923-04-14T12:43:12z1994-06-21T14:43:15z2001-10-15T04:10:32z1995-11-25T16:02:16z2005-01-04T03:14:12z2041-12-03T14:05:03z1935-09-14T04:40:11z1949-10-12T12:32:11z2010-01-03T05:14:12z2014-11-20T12:20:59z2015-09-23T10:10:13z1932-10-10T03:50:30rL r� z2059-03-14T12:43:12z1996-09-21T14:43:15z2022-12-25T16:02:16z1963-10-04T03:14:12z2013-05-08T18:15:23ru r� )r� r� r� r. r� r: rG r� TFc C sx t j�ddd�}xbt|j�D ]T}|�|�}|�|�}tt|j��}|�|� t t � ||j |j |f|�� � k�� q W d S )Nr r, )r[ rL r{ r� rd ) r/ r� r rp r� �min�argminr� r r r r� r6 )rJ rI rv �aminZaargminr r' r' rD r s zTestArgmin.test_allc C sR xL| j D ]B\}}tt�|�|d| d� t|t�|� t�|�d| d� qW d S )Nz%r)rh )r r r/ r r )rJ r� r r' r' rD r s z TestArgmin.test_combinationsc C s� t jdddgt jd�}tt �|�d� t jdddgt jd�}tt �|�d� t jdddgt jd�}tt �|�d� t jdd d gt jd�}tt �|�d� d S ) Nr, i����i����)r) i ���i���i �i �l���� l�������� )r/ r� ri r r rj rl rn )rJ rI r' r' rD � test_minimum_signed_integers s z'TestArgmin.test_minimum_signed_integersc C s� t �d�}t jdt jd�}tt|jd|� t jdt jd�}tt|jd|� t jdt jd�}tt|jd|� t jdt jd�}|jd|d� t||�d�� d S ) N)rG rL r� )r) r. )r& rL )r, rG rG )r� )r/ r� r� r r1 r r )rJ rI r� r' r' rD r # s z TestArgmin.test_output_shapec C s( t jddd�}d|d< t|�� d� d S )Ni� z<U9)r) �0iq )r/ r� r r )rJ r� r' r' rD �test_argmin_unicode5 s z TestArgmin.test_argmin_unicodec C s� t jjdd�}t jdtd�}t jdtd�}t|�d|�t �|d|�� t||� t jdtd�}t jdtd�}t|j|dd�t j||dd�� t||� d S ) N)r& r )rA r& )r) r, r r )r� r� )r/ r� r r2 r� r� r r )rJ rI r r r' r' rD r : s z TestArgmin.test_np_vs_ndarrayc C s` t jddd�}t�|jjd|j� t|�� d� d|d< t|�� d� d|d< t|�� d� d S ) Nr[ rM )r) r r� r rG r, )r/ r: r r r- r� r r )rJ rI r' r' rD � test_object_argmin_with_NULLsJ s z(TestArgmin.test_object_argmin_with_NULLsN)r` ra rb r/ r� r r r r$ r r r r r r r r r' r' r' rD r � s| $&& r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestMinMaxc C sx t tjtjdd� t tjtjdd� ttjddd�d� ttjddd�d� ttjdd d�d� ttjdd d�d� d S )Nr, r )r� )r r/ r r r r )rJ r' r' rD � test_scalarW s zTestMinMax.test_scalarc C s8 t tjtjdddgd� ttjdddggdd�d� d S )Nr, r& r i� )r� )r r/ r r r )rJ r' r' rD � test_axis` s zTestMinMax.test_axisc C s� x�dD ]�}t �d��|�}d|d< tt �|�|d � tt �|�|d � d|d< tt �|�|d � tt �|�|d � |�d� tt �|�|d � tt �|�|d � qW d S )N)zm8[s]zm8[Y]rG r� r r r� r, )r/ rH rI r r r r> )rJ r) rI r' r' rD � test_datetimed s zTestMinMax.test_datetimeN)r` ra rb r r r r' r' r' rD r U s r c @ s e Zd Zdd� ZdS )� TestNewaxisc C s< t �dddg�}d|d d �t jf }t|�� d| � d S )Nr g��������g�������?� )r/ r� r� r r� )rJ �skr� r' r' rD � test_basict s zTestNewaxis.test_basicN)r` ra rb r# r' r' r' rD r s s r c @ s> e Zd Zdd� Zddd�Zdd� Zd d � Zd d � Zd d� ZdS )�TestClipc C s( t t�||k�� t t�||k�� d S )N)r r/ r )rJ r� ZcminZcmaxr' r' rD � _check_range{ s zTestClip._check_rangeFNc C s� |d kr |}|d kr|}x�t j| D ]�}tjdkr<ddg} nddg} x�| D ]z} t �|��| �} t j�d�| �| �} |r�| �||| � n| �||�} d} | jjdkr�d} t | jj| � | � | ||� qJW q$W | S )Nr� �=�>�<i� �|) r/ �sctypesr� r� r) rK r� rI �clipr r% ) rJ Z type_groupZ array_maxZclip_minZclip_max�inplace� expected_minZ expected_maxr� Z byte_ordersr� r) r� r' r' rD � _clip_type s( zTestClip._clip_typec C s� x�dD ]~}| j dddd|d� | j dddd|d� | j ddd d |d� | j dddd|d� | j d ddd|d� | j d dd d |dd � qW d S )N)FTr� i g������)�g����� Y@)r, r r� i����g Y@�uintr )r, r- )r. )rJ r, r' r' rD r# � s zTestClip.test_basicc C s: t jddgdddgd�}|d �dd �}| �|dd � d S ) N)rS g @g @)g @g @g @)r� z<f8)r� z<f8)�zz<f8)r) r� g333333ӿg �?)r/ r� r+ r% )rJ r; r� r' r' rD �test_record_array� s zTestClip.test_record_arrayc C sv t �ddddddddg�}|�d�}tt �|dk�� |jdd �}tt �|dk�� |jdd �}tt �|dk�� d S ) Nr r, r& r r[ rL r{ r� )r )rr )r/ r� r+ r r )rJ r� r� r' r' rD �test_max_or_min� s zTestClip.test_max_or_minc C sN t �dt jdddt jg�}|�dd�}t �dt jdddt jg�}t||� d S ) Ng �g �?g @g �?r. r, g �g �?)r/ r� r� r+ r )rJ Z input_arrr r� r' r' rD �test_nan� s zTestClip.test_nan)FNN) r` ra rb r% r. r# r1 r2 r3 r' r' r' rD r$ z s r$ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestCompressc C sx dddddgg}t �d��dd�}t jdd g|dd �}t||� d d gddgg}t jdd dd dg|d d �}t||� d S ) NrL r{ r� rd r� rG r& r r, )r� r )r/ rH r} r� r )rJ r� r� r� r' r' rD r � s zTestCompress.test_axisc C s@ dgdgg}t �d��dd�}t jddg|dd�}t||� d S )Nr, r{ rG r& rL r )r� )r/ rH r} r� r )rJ r� r� r� r' r' rD � test_truncate� s zTestCompress.test_truncatec C s0 t �d��dd�}t �ddg|�}t|d� d S )NrG r& rL r r, )r/ rH r} r� r )rJ r� r� r' r' rD r* � s zTestCompress.test_flattenN)r` ra rb r r5 r* r' r' r' rD r4 � s r4 c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )� TestPutmaskc C s0 t �|||� t|| ||�� t|j|� d S )N)r/ �putmaskr r) )rJ r� r� �maskr� r' r' rD � tst_basic� s zTestPutmask.tst_basicc c s~ t ttjtg}tj�d�d }|dk }xRdD ]J}xDtj�� D ]6}x0|D ](}||krF| j|� � � |�|||fV qFW q<W q,W d S )Ni� r r� )i����r r ) �bytesr r/ �voidr0 r� r* �valuesr9 r� rI )rJ �unchecked_typesr� r8 r� r r� r' r' rD � test_ip_types� s zTestPutmask.test_ip_typesc C s$ t ttjt�dddg�dgd� d S )Nr, r& r TrL )r r1 r/ r7 r� )rJ r' r' rD �test_mask_size� s zTestPutmask.test_mask_sizec C s: t �dddg|�}t �|dddgd� t|dddg� d S )Nr, r& r TFr. )r/ r� r7 r )rJ r) r� r' r' rD � tst_byteorder� s zTestPutmask.tst_byteorderc c s xdD ]}| j |fV qW d S )N)z>i4z<i4)r@ )rJ r) r' r' rD �test_ip_byteorder� s z TestPutmask.test_ip_byteorderc C s� t jddgdddgd�}t �|d dd gd � t|d d d g� t|d d dg� t|d ddg� t �|d dd gd� t|d d d g� t|d ddg� t|d ddg� d S )N)rS g @g @)g @g @g @)r� z<f8)r� z>f8)r0 z<f8)r) r� TFrG rL r� r& r[ r0 r r� )r/ r� r7 r )rJ r; r' r' rD r1 � s z TestPutmask.test_record_arrayN) r` ra rb r9 r> r? r@ rA r1 r' r' r' rD r6 � s r6 c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�TestTakec C s* t t|jd ��}t|j|dd�|� d S )Nr )r� )r� rp r? r �take)rJ r� r* r' r' rD r9 s zTestTake.tst_basicc c sh t ttjtg}tj�d�d }d|_x>tj�� D ]0}x*|D ]"}||kr:| j |� � � |�fV q:W q0W d S )Nr� r )r& r r[ ) r: r r/ r; r0 r� r? r* r< r9 r� rI )rJ r= r� r r� r' r' rD r> s zTestTake.test_ip_typesc C sd t j�d�d }d|_tt|jdddgdd� tt|jdgdd� t|jd gdd�d |d � d S ) Nr� r )r& r r[ r r, r& )r� r� r. )r/ r� r? r r� rC r )rJ r� r' r' rD � test_raise s zTestTake.test_raisec C sZ t j�d�d }d|_t|jdgddd�d |d � t|jdgddd�d |d � d S ) Nr� r )r& r r[ r. r r+ )r� �moder& r, )r/ r� r? r rC )rJ r� r' r' rD � test_clip s zTestTake.test_clipc C sz t j�d�d }d|_t|jdgddd�d |d � t|jd gddd�d |d � t|jd gddd�d |d � d S ) Nr� r )r& r r[ r. r �wrap)r� rE r, r& r )r/ r� r? r rC )rJ r� r' r' rD � test_wrap# s zTestTake.test_wrapc C s2 t �dddg|�}t|�dddg�dddg� d S )Nr, r& r r )r/ r� r rC )rJ r) r� r' r' rD r@ * s zTestTake.tst_byteorderc c s xdD ]}| j |fV qW d S )N)z>i4z<i4)r@ )rJ r) r' r' rD rA . s zTestTake.test_ip_byteorderc C sD t jddgdddgd�}|�dg�}t|d d ko<|d d k� d S ) N)rS g @g @)g @g @g @)r� z<f8)r� z>f8)r0 z<f8)r) r, r� g @r� g @)r/ r� rC r )rJ r; Zrec1r' r' rD r1 2 s zTestTake.test_record_arrayN) r` ra rb r9 r> rD rF rH r@ rA r1 r' r' r' rD rB s rB c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestLexsortc C s� ddddddg}ddddddg}t �||f�}t �ddddddg�}t||� t �||f�}t �|�}t||� t|d | t �|d �� d S )Nr, r& r rL r r[ r{ )r/ �lexsortr� r �vstackr� )rJ rI r^ r% � expected_idxr� r' r' rD r# ; s zTestLexsort.test_basicc C s� t jdddgdd�}t jdddgdd�}t �||f�}t �dddg�}t||� t jdddgdd�}t jdddgdd�}t �||f�}t �dddg�}t||� d S )Nr z datetime64[D])r) r& r, ztimedelta64[D])r/ r� rJ r )rJ rI r^ r% rL r' r' rD r H s zTestLexsort.test_datetimec C s� t j�dd�}t j�ddddddgd�}x:||fD ].}t �|�d �f�}t j|d d �}t||� q2W x�||f||ffD ]|\}}t �||f�}t|t �|�d �|f�� t|t �||�d �f�� t j|d d �t j|d d � }}t|t �||f�� qvW d S )NrG i� �abc�xyZwzZefghiZqwstr� rM � mergesort)r r0 )r) )r/ r� �choicerJ rI r� r r� )rJ rI r^ r� r� r� r� r% r' r' rD � test_objectU s zTestLexsort.test_objectc C s. t �ddd��dd�}tt jt j|dd� d S )Ng g �?�~ r� r r& )r� )r/ r� r} r r rJ )rJ r� r' r' rD �test_invalid_axise s z TestLexsort.test_invalid_axisN)r` ra rb r# r rQ rS r' r' r' rD rI : s rI c @ s> e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Ze j d2d3� �Z d4d5� Z d6d7� Z d8d9� Z!d:d;� Z"d<d=� Z#d>d?� Z$d@dA� Z%dBdC� Z&dDdE� Z'dFdG� Z(dHdI� Z)dJdK� Z*dLS )M�TestIOz.Test tofile, fromfile, tobytes, and fromstringc C sx d}t jj}||�||��t j�d | _t jt jt j t jg| jdd d �df< | jj| _t� � | _ tj | j d�| _ d S )N)r& r[ r y �?r r, )�dir) r/ r� rI r r� r� �infr) �tempfile�mkdtemp�tempdir�mktemp�filename)rJ r? r� r' r' rD rK l s & z TestIO.setUpc C s t �| j� d S )N)�shutil�rmtreerY )rJ r' r' rD �tearDownu s zTestIO.tearDownc s> t �� � tttj� tjd� t�d�}tt� fdd�|� d S )Nr r� c s | � � �S )N)�tofile)r� )r^ r' rD r� ~ r� z$TestIO.test_nofile.<locals>.<lambda>)r^ r_ r �IOErrorr/ �fromfiler; r� )rJ r� r' )r^ rD � test_nofilex s zTestIO.test_nofilec C s8 t jddddgt jd�}t jddt jd�}t||� d S )NTF)r) z 1 0 -2.3 0.0� )�sepr) )r/ r� rh � fromstringr )rJ r� r� r' r' rD �test_bool_fromstring� s zTestIO.test_bool_fromstringc C s4 t jdt jdd�}t jddgt jd�}t||� d S )Nz#9923372036854775807 104783749223640rc )r) rd l �� *�M l �M<3} )r) )r/ re ro r� r )rJ r� r� r' r' rD �test_uint64_fromstring� s z TestIO.test_uint64_fromstringc C s4 t jdt jdd�}t jddgt jd�}t||� d S )Nz -25041670086757 104783749223640rc )r) rd l����e �o[l �M<3} )r) )r/ re rn r� r )rJ r� r� r' r' rD �test_int64_fromstring� s z TestIO.test_int64_fromstringc C s4 t | jd�}|�� t�| j�}t|jdkd� d S )Nr� r zArray not empty)�openr[ �closer/ ra r rA )rJ r� r� r' r' rD �test_empty_files_binary� s z TestIO.test_empty_files_binaryc C s8 t | jd�}|�� tj| jdd�}t|jdkd� d S )Nr� rc )rd r zArray not empty)ri r[ rj r/ ra r rA )rJ r� r� r' r' rD �test_empty_files_text� s z TestIO.test_empty_files_textc C sV t | jd�}| j�|� |�� t | jd�}tj|| jd�}|�� t|| jj � d S )N�wb�rb)r) ) ri r[ r� r_ rj r/ ra r) r r� )rJ r� r� r' r' rD �test_roundtrip_file� s zTestIO.test_roundtrip_filec C s2 | j �| j� tj| j| jd�}t|| j j� d S )N)r) )r� r_ r[ r/ ra r) r r� )rJ r� r' r' rD �test_roundtrip_filename� s z TestIO.test_roundtrip_filenamec C sZ | j �� }tj|| jd�}t|| j j� | j �d�}tj|| jd�}t|| j �d�� d S )N)r) r� )r� �tobytesr/ re r) r r� r( )rJ ru r� r' r' rD �test_roundtrip_binary_str� s z TestIO.test_roundtrip_binary_strc C sf | j j�� }d�tt|��}tj|dd�}t�|� }t || || � t || || dd� d S )N�@)rd rL )�decimal) r� r� r� r� r r� r/ re �isfiniter r )rJ r� ru r� Znan_maskr' r' rD �test_roundtrip_str� s zTestIO.test_roundtrip_strc C s8 | j j�� }d�tt|��}tj|dd�}t||� d S )Nrs )rd ) r� r� r� r� r r� r/ re r )rJ r� ru r� r' r' rD �test_roundtrip_repr� s zTestIO.test_roundtrip_reprc C sZ | j �| j� dd� }tj| jddd��(}||_||_| jtt j || j d� W d Q R X d S )Nc _ s t d��d S )NzCan not tell or seek)r` )r� r� r' r' rD �fail� s z-TestIO.test_unseekable_fromfile.<locals>.failrn r )� buffering)r) ) r� r_ r[ r^ ri ra �tellrP r` r/ ra r) )rJ rx r� r' r' rD �test_unseekable_fromfile� s z TestIO.test_unseekable_fromfilec C sN | j �| j� tj| jddd��$}tj|| jd�}t|| j j � W d Q R X d S )Nrn r )ry )r) ) r� r_ r[ r^ ri r/ ra r) r r� )rJ r� r� r' r' rD � test_io_open_unbuffered_fromfile� s z'TestIO.test_io_open_unbuffered_fromfilec C s t �d�}|�| j� ttj�| j�|j� t |t � | j�� t | jd��6}|� |j� |�|� ttj�| j�|jd � W d Q R X t | jd�� � t | jd��}|�|� W d Q R X t |t � | j�� t | jd��}|�|� W d Q R X ttj�| j�|jd � d S )Ni @ zr+br& r� �ab)r/ r2 r_ r[ r �os�path�getsizer� r ra ri ra rj )rJ r� r� r' r' rD �test_largish_file� s $zTestIO.test_largish_filec C sN | j �| j� tj| jddd��}tj|| jd�}W d Q R X t|| j j � d S )Nrn r. )ry )r) ) r� r_ r[ r^ ri r/ ra r) r r� )rJ r� r� r' r' rD � test_io_open_buffered_fromfile� s z%TestIO.test_io_open_buffered_fromfilec C s� t jd t jt jd g}x�|D ]�}t| jd�}|�|d � |�d� |�� x^dD ]V}d||f }t| j|�}|�d� tj |tj dd� |� � }|�� t |d |d � qTW q W d S ) Nrd rm r, � )rn zr+bz%d %sr& )r) �countrG )rh ) r^ �DEFAULT_BUFFER_SIZEri r[ ra �writerj �readr/ ra r rz r )rJ �sizesrA r� rE rh r r' r' rD �!test_file_position_after_fromfile� s z(TestIO.test_file_position_after_fromfilec C s� t jd t jt jd g}x�|D ]�}d|f }t| jd�}|�|d � |�d� |�d� |�d� tjdgtjd �� |� |� � }|� � t |d |d � t| jd �}|� d � |�dd� tjdgtjd �� |� |� � }|� � t |d|d � q W d S )Nrd z%drm r, r� rG s 12r )r) ry )rh zr+br& )r^ r� ri r[ ra r� r/ r� r r_ rz rj r r� )rJ r� rA rh r� r r' r' rD � test_file_position_after_tofile s* z&TestIO.test_file_position_after_tofilec K sT t j|f|�}t||� t| jd�}|�|� |�� t j| jf|�}t||� d S )Nrm )r/ re r ri r[ r� rj ra )rJ ru r� r� r� r� r' r' rD � _check_from$ s zTestIO._check_fromc C s0 | j dtjtjtjtjtjtjtjgdd� d S )Ns3 nan +nan -nan NaN nan(foo) +NaN(BAR) -NAN(q_u_u_x_)rc )rd )r� r/ r� )rJ r' r' rD r3 . s zTestIO.test_nanc C s6 | j dtjtjtj tjtj tjtj gdd� d S )Ns. inf +inf -inf infinity -Infinity iNfInItY -inFrc )rd )r� r/ rV )rJ r' r' rD �test_inf4 s $zTestIO.test_infc C s | j ddddddgdd� d S ) Ns' 1.234 -1.234 .3 .3e55 -123133.1231e+133gX9��v��?gX9��v��g333333�?g�. LKR?KgY��]x��rc )rd )r� )rJ r' r' rD � test_numbers: s zTestIO.test_numbersc C s" | j dt�ddddg�dd� d S )Ns �? @ @@ �@r, r& r r[ z<f4)r) )r� r/ r� )rJ r' r' rD � test_binary> s zTestIO.test_binaryc C s� t jdkrdS y�d}tjdtjd�}t|�}t�� }|j}t� |||j �� |� |� d� tj |tjd�}|�� tt|�|k� t|d|� |k�� � t|| d� |k�� � W n ttfk r� Y nX dS )z�Test workarounds for 32-bit limited fwrite, fseek, and ftell calls in windows. These normally would hang doing something like this. See http://projects.scipy.org/numpy/ticket/1660�win32Nl rd )r) r )r� �platformr/ rH ri r� rW �NamedTemporaryFile�filer r� r_ ra ra rj r r � MemoryErrorr1 )rJ Z fourgbplusZ testbytesr� Zfliker� rI r' r' rD �test_big_binaryC s" zTestIO.test_big_binaryc C s | j dddddgdd� d S )Ns 1,2,3,4g �?g @g @g @�,)rd )r� )rJ r' r' rD r� \ s zTestIO.test_stringc C sP | j dddddgddd� | j ddddgd dd� | j dddddgd dd� d S ) Ns 1,2,3,4g �?g @g @g @r[ r� )r� rd r r. )r� )rJ r' r' rD �test_counted_string_ s zTestIO.test_counted_stringc C s | j dddddgtdd� d S )Ns 1 2 3 4 r, r& r r[ rc )r) rd )r� r� )rJ r' r' rD �test_string_with_wsd s zTestIO.test_string_with_wsc C s | j ddddgdtdd� d S )Ns 1 2 3 4 r, r& r rc )r� r) rd )r� r� )rJ r' r' rD �test_counted_string_with_wsg s z"TestIO.test_counted_string_with_wsc C s6 | j dddddgdd� | j dddddgtdd � d S ) Ns 1 , 2 , 3 , 4g �?g @g @g @r� )rd s 1,2,3,4)r) rd )r� r� )rJ r' r' rD � test_asciik s zTestIO.test_asciic C s | j dddgdd� d S )Ns 1.234 1,234gX9��v��?g �?rc )rd )r� )rJ r' r' rD �test_malformedo s zTestIO.test_malformedc C s | j dddddgdd� d S )Ns 1_x_3_x_4_x_5r, r r[ rL Z_x_)rd )r� )rJ r' r' rD � test_long_sepr s zTestIO.test_long_sepc C s0 t jddddgt jd�}| jd|dt jd� d S ) Nr, r& r r[ )r) s 1,2,3,4r� )rd r) )r/ r� r� r� )rJ r� r' r' rD � test_dtypeu s zTestIO.test_dtypec C sj t jddddgt jd�}d}t| jd�}|�|� |�� t j| jdt jd�}t|j dk� t ||� d S ) NTF)r) s 1,0,-2.3,0rm r� )rd r) r� ) r/ r� rh ri r[ r� rj ra r r) r )rJ r� ru r� r� r' r' rD �test_dtype_booly s zTestIO.test_dtype_boolc C s| t jddddgtd�}t| jd�}|j|dd� |�� t| jd �}|�� }|�� t �d d � |�d�D ��}t ||� d S ) Ng)\��(�?r& g�G�z @r[ )r) r� r� )rd r� c S s g | ] }t |��qS r' )r� )rq r r' r' rD rr � s z*TestIO.test_tofile_sep.<locals>.<listcomp>) r/ r� r� ri r[ r_ rj r� �splitr )rJ r� r� ru r� r' r' rD �test_tofile_sep� s zTestIO.test_tofile_sepc C sd t jddddgtd�}t| jd�}|j|ddd � |�� t| jd �}|�� }|�� t|d � d S ) Ng)\��(�?r& g�G�z @r[ )r) r� r� z%.2f)rd r� r� z1.51,2.00,3.51,4.00) r/ r� r� ri r[ r_ rj r� r )rJ r� r� ru r' r' rD �test_tofile_format� s zTestIO.test_tofile_formatc C sd t | j�� t | j�� t | j�� t | j�� t | j�� t | j�� t | j�� t | j�� d S )N) r r� r3 r� r� r� r� r� r� )rJ r' r' rD � test_locale� s zTestIO.test_localeN)+r` ra rb �__doc__rK r^ rb rf rg rh rk rl ro rp rr rv rw r{ r| r� r� r� r� r� r3 r� r� r� r r3 r� r� r� r� r� r� r� r� r� r� r� r� r� r' r' r' rD rT i sL rT c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestFromBufferc C s t tj|f|�|� d S )N)r r/ ro )rJ r� r� r� r' r' rD r9 � s zTestFromBuffer.tst_basicc c sl xfdD ]^}xXt ttjgD ]H}t�|��|�}tj�d�d �|�}|�� }| j ||j d|ifV qW qW d S )N)r( r' )r[ r� rL r) ) r� r� r/ r r) rK r� rI rq r9 r� )rJ r� r) r r� rB r' r' rD � test_ip_basic� s z TestFromBuffer.test_ip_basicc c s | j dt�g �i fV d S )Nr� )r9 r/ r� )rJ r' r' rD � test_empty� s zTestFromBuffer.test_emptyN)r` ra rb r9 r� r� r' r' r' rD r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestFlatc C sh t �d�}|�dd�}d|_d|j_|| _|d d d�d d d�f | _|| _|d d d�d d d�f | _ d S )Ng 4@r[ rL )r[ rL Fr& ) r/ rH r} r? rN rO rI r^ ri rj )rJ ri rI r' r' rD rK � s zTestFlat.setUpc C sN d}yd| j jd< W n tk r, d}Y nX t|� t| j jd dk� d S )NFg Y@r� Tg (@)rI r� r1 r )rJ � testpassedr' r' rD �test_contiguous� s zTestFlat.test_contiguousc C sN d}yd| j jd< W n tk r, d}Y nX t|� t| j jd dk� d S )NFg Y@r[ Tg (@)r^ r� r1 r )rJ r� r' r' rD �test_discontiguous� s zTestFlat.test_discontiguousc C s� | j j�� }| jj�� }| jj�� }| jj�� }t|jjdk� t|jjdk� t|jjdk� t|jjdk� t|jj dk� t|jj dk� t|jj dk� t|jj dk� t|j | jk� d S )NFT) rI r� r� r^ ri rj r rN rO rY r� )rJ r r� r� r� r' r' rD �test___array__� s zTestFlat.test___array__N)r` ra rb rK r� r� r� r' r' r' rD r� � s r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )� TestResizec C s� t �dddgdddgdddgg�}tr6|jddd� n |�d� t|jd d� t �dddgdddgdddgg�j� t|dd � jd� d S )Nr, r )rL rL F)�refcheckr� )r/ r� r �resizer r� )rJ r� r' r' rD r# � s " &zTestResize.test_basicc C s< t �dddgdddgdddgg�}|}| �t|jd� ~d S )Nr, r )rL r, )r/ r� rP r1 r� )rJ r� r� r' r' rD �test_check_reference� s "z TestResize.test_check_referencec C sH t �d�}tr |jddd� n |�d� t|t �d�dd d �f � d S )Nr F)r� r )r/ r; r r� r )rJ r� r' r' rD �test_int_shape� s zTestResize.test_int_shapec C s@ t �d�}|�d � t|t �d�� |�� t|t �d�� d S )Nr )r/ r; r� r )rJ r� r' r' rD �test_none_shape� s zTestResize.test_none_shapec C sn xht d�D ]\}t�d�}|�d� t|jd� t|jd� t�d�}|�d� t|jd� t|jd� q W d S )NrG )r, r' r, )rp r/ r: r� r r? rA )rJ rv r� r' r' rD � test_0d_shape s zTestResize.test_0d_shapec C s` | � tt�d�jd� | � tt�d�jd� | j tt�d�jdd� | j tt�d�jdd� d S )Nr �hir. r, )r* )r� )rP r� r/ r; r� r1 )rJ r' r' rD �test_invalid_arguments s z!TestResize.test_invalid_argumentsc C sB t �d�}tr"|jddddd� n|�ddd� t|jdk� d S )Nr r& r, F)r� )r r& r, )r/ r; r r� r r? )rJ r� r' r' rD �test_freeform_shape s z TestResize.test_freeform_shapec C s\ t �d�}tr"|jddddd� n|�ddd� t|d t �d�� t|d t �d�� d S )Nr r& F)r� r r, )r r )r/ r; r r� r r2 )rJ r� r' r' rD �test_zeros_appended s z TestResize.test_zeros_appendedc C sp t jddtdfgd�}tr*|jddd� n |�d� t|jd� t|d d d � d � t|d d d � d � d S ) NrG r r& )r) r F)r� )r rS r r, )r/ r� r0 r r� r r? r )rJ rI r' r' rD � test_obj_obj' s zTestResize.test_obj_objN) r` ra rb r# r� r� r� r� r� r� r� r� r' r' r' rD r� � s r� c @ st e Zd Zdd� Zdd� Zejd dkr8dd� Zd d � Znd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� ZdS )� TestRecordc C s4 t �dtfdtfg�}ddg|_t|jddg� d S )Nr� rv r r� )r/ r) r� r� r� r )rJ r r' r' rD �test_field_rename4 s z TestRecord.test_field_renamec C s dd� }t t|� d S )Nc S s t �dddg�} d S )N)r/ r� )rN r� )r/ r) )r) r' r' rD � test_assign: s zCTestRecord.test_multiple_field_name_occurrence.<locals>.test_assign)r r1 )rJ r� r' r' rD �#test_multiple_field_name_occurrence9 s z.TestRecord.test_multiple_field_name_occurrencer r c C s� t ttjdtfg� t ttjdtfg� t�dtfg�}t t|jd� tjdddg|d�}t t|jd� |d }t t|jd� d S ) N� a)r^ r� )r� r^ )r, )r& )r )r) r ) r r� r/ r) r� r1 r� r� r� )rJ r r� r� r' r' rD �test_bytes_fieldsA s z TestRecord.test_bytes_fieldsc C s dd� }t t|� d S )Nc S s t �dddg�} d S )N)u ₹r� )rN r� )r/ r) )r r' r' rD �test_assign_unicodeQ s zHTestRecord.test_multiple_field_name_unicode.<locals>.test_assign_unicode)r r1 )rJ r� r' r' rD � test_multiple_field_name_unicodeP s z+TestRecord.test_multiple_field_name_unicodec C sh d}t �|dftfg�}|| |d t jdddg|d�}|| |d |d }|| |d d S )Nr^ rI )r, )r& )r )r) r )r/ r) r� r� )rJ �titler r� r� r' r' rD �test_unicode_field_titlesZ s z$TestRecord.test_unicode_field_titlesc C s4 d}t ttj|tfg� t ttjd|ftfg� d S )Nr^ rI )r r� r/ r) r� )rJ r� r' r' rD �test_unicode_field_namesg s z#TestRecord.test_unicode_field_namesc C s� t jdddddgfgd�}tjd dk}|rxtf}tt|jd d � tt|jd � tt|d jd d � tt|d jd � ntt f}�x�|D �]�}|� � }|d �}d ||< t || d � |d �}tt |j|d � tt |j|� d|d |< t || d� tt |d j|d � tt |d j|� |d�}|d�} d || | < t || | d � tt || j|d � tt || j|� |d�} d|| < t � ��} | �td� | �td� t |d dg d �� d� t |dd g d �� d� t |d dg d �� d� t |d dg d �d��� d� t |dd g d �d��� d� dddgfg} t |d dg d �| ��� d� W d Q R X q�W |�sxtd��n tt |jdd � tt |jd� d S )N)r, )r� r] )�f2r] �f3)�sf1r] )r) r r s f1r, r� s sf1z not at allr& r� r� z&Assignment between structured arrays.*z Numpy has detected that you .*)r& r )r r& )r& )r, )r] r& )r� r] zGnon ascii unicode field indexing skipped; raises segfault on python 2.xu Ϡ)r/ r2 r� r� r� r r� � __setitem__r� r r� r r1 r# r= r> r~ ru r! ) rJ rI Zis_py3�funcsr� r^ �fn1ZfnnZfn3Zsfn1Zfn2r@ Z view_dtyper' r' rD �test_field_namesm sd zTestRecord.test_field_namesc C sl dd� }t jdddddgfgd�}d |d d < d |d d < d|d d < t jdddddgfgd�}d |d d < d |d d < d|d d < t||d d g jd d�tg� t||� |d d g }|�� }t||d jd d�tg� t|d d d� t||d jd d�g � t jddd�}t||dd g jd�tg� t jd ddgd�}t jd ddgd�}t||jd|�tg� d S )Nc _ s: t jdd��}t �d� | ||� W d Q R X dd� |D �S )NT)r� r� c S s g | ] }|j �qS r' )r� )rq r� r' r' rD rr � s zUTestRecord.test_field_names_deprecation.<locals>.collect_warnings.<locals>.<listcomp>)r� r� � simplefilter)r� r� r� �logr' r' rD �collect_warnings� s zATestRecord.test_field_names_deprecation.<locals>.collect_warnings)r, )r� r] )r� r] r� )r� r] )r) r, r� r r& r� )r )rG ry rG r zi8,i8,i8rZ zi8,i8)rI r] )r^ r] r' )r/ r2 r r� r> ru )rJ r� rI r^ �subsetZ subset_viewr r' r' rD � test_field_names_deprecation� s8 z'TestRecord.test_field_names_deprecationc C s� t jddgdd�}d|j_t jddgddgd�}d|j_t jddgdd�}d|j_| �t|d �t|d �k� | �t|d �t|d �k� | �t|d �t|d �k� | �t|d �t|d �ko�|d |d k� d S ) N)r, r& zi1,i2)r) F)r r[ )Znum1r9 )Znum2rZ r r, )r/ r� rN rO r� rq )rJ rI r^ r r' r' rD �test_record_hash� s zTestRecord.test_record_hashc C s( t jddgdd�}| �tt|d � d S )N)r, r& zi1,i2)r) r )r/ r� rP r� rq )rJ rI r' r' rD �test_record_no_hash� s z TestRecord.test_record_no_hashc C s@ t jdgg g g dd�d� t jdddddgg g g dd�d� d S )Nr' r� )r� �formats�offsetsr9 )r) )r/ r� )rJ r' r' rD � test_empty_structure_creation� s z(TestRecord.test_empty_structure_creationN)r` ra rb r� r� r� r� r� r� r� r� r� r� r� r� r� r' r' r' rD r� 3 s ?0 r� c @ s e Zd Zdd� ZdS )�TestViewc C sd t jddgdt jfdt jfdt jfdt jfgd�}|jdd�}|�d�}t||� t|d d g� d S ) N)r, r& r r[ )rL r{ r� rd r� re r^ rI )r) z<i4ii)r/ r� ri ru r )rJ r� r� r0 r' r' rD r# � s zTestView.test_basicN)r` ra rb r# r' r' r' rD r� � s r� c K s | j f |�S )N)�mean)rI r� r' r' rD �_mean s r� c K s | j f |�S )N)�var)rI r� r' r' rD �_var s r� c K s | j f |�S )N)�std)rI r� r' r' rD �_std s r� c @ s� e Zd ZeeegZdd� Zdd� Zdd� Z dd� Z d d � Z d d � Z d d� Z dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd S ) � TestStatsc C s^ t j�td�� t j�d�| _| jd| j | _t �dd� | jjD ��| _| j� dd�| _d S )Nr )r[ rL y �?c S s g | ]}t t|���qS r' )r r� )rq r� r' r' rD rr s z#TestStats.setUp.<locals>.<listcomp>r[ rL ) r/ r� r4 rp �rmat�cmatr� r� �omatr} )rJ r' r' rD rK s zTestStats.setUpc C sV xPt �d�dddfD ]:}tt �|g�d� tt �|g�d� tt �|g�d� qW d S )Ng �?r, y �? g )r/ �float16r r� r� r� )rJ r� r' r' rD �test_python_type s zTestStats.test_python_typec C s� t �d�}xv| jD ]l}x<dD ]4}|||dd�}t|j|jk� t|j| dk� q W x(dD ] }|||dd�}t|jdk� qZW qW d S )Nr )r r, T)r� �keepdimsr, )N)r, r, )r/ r; r� r r� r? )rJ �matr� r� r� r' r' rD � test_keepdims s zTestStats.test_keepdimsc C s� t �d�}xF| jD ]<}t �d�}||dd�}||d|d�}t||� t||� qW t �d�}tt||d|d� t �d�}tt||d|d� d S )Nr r, )r� )r� r� r& )r& r& )r/ r; r� r2 r r: r r1 )rJ r� r� r� r� r� r' r' rD �test_out( s zTestStats.test_outc C s� t jd }t jd }xh| jD ]^}t �td�gd gd �}|jj}||dd�jj}t||k� t||d d��}t|tk� q W xh| jD ]^}xX|D ]P}t jd|d�}t j }||dd�jj}t||k� ||d d�jj}t||k� q�W q�W xjt gD ]`}xZ|D ]R}t jd|d�}|jj}||dd�jj}t||k� ||d d�jj}t||k� q�W q�W xrt t gD ]f}x^|D ]V}t jd|d�}|j jj}||dd�jj}t||k� ||d d�jj}t||k� �qhW �q^W d S )Nr� r r, r )r� )r) )r/ r r� r� r r) r� r r; r r� r� r� r� )rJ rH rI r� r� r� r� r r' r' rD �test_dtype_from_input5 sB z TestStats.test_dtype_from_inputc C sz t �d�}xj| jD ]`}xZt jd D ]L}t �|�j}||d|d�jj}t||k� ||d |d�jj}t||k� q"W qW d S )Nr r r, )r� r) )r/ r; r� r r) r� r )rJ r� r� r r� r� r' r' rD �test_dtype_from_dtypef s z TestStats.test_dtype_from_dtypec C s� xVt gD ]L}xFtd�D ]:}| jjd }|| jdd�| }|| jd|d�|| }qW qW xvtgD ]l}xftd�D ]Z}| jjd }|| jdd�t�|� }|| jd|d�t�|| � }t||� t||� qnW q`W d S )Nr r, )r� )r� �ddof)r� rp r� r? r� r/ �sqrtr )rJ r� r� rU r� r� r' r' rD � test_ddof� s zTestStats.test_ddofc C s� | j jd }x�ttgD ]�}x~t||d �D ]l}tjdd��V}t�d� || j d|d�}t|dk � � � tt |�dk� tt |d j t �� W d Q R X q*W qW d S )Nr, r& T)r� r� )r� r� r )r� r? r� r� rp r� r� r� r r r� r� r� �RuntimeWarning)rJ rU r� r� r� r� r' r' rD �test_ddof_too_big� s zTestStats.test_ddof_too_bigc C s� t �d�}x�| jD ]�}xldD ]d}tjdd��N}t�d� tt �|||d���� � tt |�dk� tt |d j t �� W d Q R X q W xFdD ]>}tjdd��(}t�d� t |||d�t �g �� W d Q R X q�W qW d S ) N)r r )r NT)r� r� )r� r )r, )r/ r2 r� r� r� r� r r r r� r� r� r� r )rJ r/ r� r� r� r' r' rD r� � s " zTestStats.test_emptyc C s� x�| j | j| jgD ]z}x8dD ]0}|j|d�}t||d�|j| }t||� q W x:dD ]2}|j|d�}t||d�t�|j� }t||� qVW qW d S )N)r r, )r� )N) r� r� r� r r� r? r r/ r� )rJ r� r� r� r� r' r' rD �test_mean_values� s zTestStats.test_mean_valuesc C s t tjddd��dkst�d S )Ni�� r� )r) r, )r� r/ r� rE )rJ r' r' rD �test_mean_float16� s zTestStats.test_mean_float16c C sr xl| j | j| jgD ]X}xRdD ]J}t||�� |d�}t||d�}|||�� }t||d�}t||� q W qW d S )N)r r, N)r� )r� r� r� r� r_ r^ r� r )rJ r� r� Zmsqrr� r� r� r' r' rD �test_var_values� s zTestStats.test_var_valuesc C sT xN| j | j| jgD ]:}x4dD ],}t�t||d��}t||d�}t||� q W qW d S )N)r r, N)r� )r� r� r� r/ r� r� r� r )rJ r� r� r� r� r' r' rD �test_std_values� s zTestStats.test_std_valuesc C s� G dd� dt j�}|ddddgddd d ggd �}|�d�}t|j|jk� |�d�}t|j|jk� |�d�}t|j|jk� d S ) Nc @ s e Zd Zdd� Zdd� ZdS )z*TestStats.test_subclass.<locals>.TestArrayc S s t �|�}|�| �}||_|S )N)r/ r� ru �info)�clsr- r� r r' r' rD �__new__� s z2TestStats.test_subclass.<locals>.TestArray.__new__c S s t |dd�| _d S )Nr� r� )r� r� )rJ r� r' r' rD �__array_finalize__� s z=TestStats.test_subclass.<locals>.TestArray.__array_finalize__N)r` ra rb r� r� r' r' r' rD � TestArray� s r� r, r& r r[ rL r{ r� rd Zjubba)r/ r= r� r r� r� r� )rJ r� �datr� r' r' rD � test_subclass� s zTestStats.test_subclassN)r` ra rb r� r� r� r� rK r� r� r� r� r� r� r� r� r� r� r� r� r� r' r' r' rD r� s 1 r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestVdotc C s t jd t jd }t jd }t �d�}xF|d D ]:}|�|�}t �||�}tt �|�� tt �||�d� q2W t �d�d }xF|d D ]:}|�|�}t �||�}tt �|�� tt �||�d� q�W t jdt jd�}t �||�}tt �|�� tt �||�d� d S ) Nr r� r� r rM y �?)r) T) r/ r r; rI �vdotr �isscalarr r� )rJ Z dt_numericZ dt_complexrI r r^ r� r' r' rD r# � s$ zTestVdot.test_basicc C sz t jddgddggdd�}t jddgddggdd�}t �||�}tt �||�|� tt �||�|� tt �||�|� d S )Nr, r& r r[ r( )r* r� )r/ r� r� r )rJ rI r^ r� r' r' rD �test_vdot_array_order� s z TestVdot.test_vdot_array_orderc C s8 �x0dD �]&}t �|ddf�}t �|ddf�}t �|�|d d �ddf< t �|�d |d d �ddf< |d }|d }tt �||�t �|�� |�� �� tt �||�� �t �|�� |�� �� tt �|�� |�t �|�� |�� �� tt �|�d�|�t �|�� |�� �� tt �||�d��t �|�� |�� �� qW d S )N)r& i� r& r r, ).r r� )r/ r2 rH r r� r( r� )rJ rA rI r^ r' r' rD �test_vdot_uncontiguous s" z TestVdot.test_vdot_uncontiguousN)r` ra rb r# r� r� r' r' r' rD r� � s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+S ),�TestDotc C sb t j�d� t j�dd�| _t j�dd�| _t j�d�| _t j�dd�| _t j�d�| _d| _ d S )Nr� r[ r& r, r� ) r/ r� r4 r� r/ rl �b2�b3�b4�N)rJ r' r' rD rK s z TestDot.setUpc C s@ | j }t�|�� |�}t�ddgddgg�}t||| jd� d S )Ng�h�5�?g��V��?g1�-�?)rt )r/ r/ r. r6 r� r r� )rJ r/ r� r� r' r' rD �test_dotmatmat! s zTestDot.test_dotmatmatc C sH | j | j }}t�||�}t�dgdgdgdgg�}t||| jd� d S )NgcLU5���?g �� �?g� )XF�?g��с��?)rt )r/ rl r/ r. r� r r� )rJ r/ rl r� r� r' r' rD �test_dotmatvec( s zTestDot.test_dotmatvecc C s@ | j | j }}t�||�}t�ddddg�}t||| jd� d S )Ng�Ho��?g�Ċ�"�?g-P����?g���Nz��?)rt )r/ r� r/ r. r� r r� )rJ r/ r� r� r� r' r' rD �test_dotmatvec2/ s zTestDot.test_dotmatvec2c C s< | j | j }}t�||�}t�ddg�}t||| jd� d S )Ng�ة�[��?g����?)rt )r/ r� r/ r. r� r r� )rJ r/ r� r� r� r' r' rD �test_dotvecmat5 s zTestDot.test_dotvecmatc C sF | j | j }}t�||�� �}t�ddddgg�}t||| jd� d S )Ng���oc��?gTb�i\�?g�2�r��?g��b��?)rt )r� r/ r/ r. r6 r� r r� )rJ r� r/ r� r� r' r' rD �test_dotvecmat2; s zTestDot.test_dotvecmat2c C s@ | j | j }}t�|�� |�}t�ddg�}t||| jd� d S )Ng�ة�[��?g����?)rt )r/ r� r/ r. r6 r� r r� )rJ r/ r� r� r� r' r' rD �test_dotvecmat3A s zTestDot.test_dotvecmat3c C sD | j | j }}t�||�}t�ddgddgg�}t||| jd� d S )Ngc�Q0���?g�X��O��?g�5쵨h�?g 5����?)rt )rl r� r/ r. r� r r� )rJ rl r� r� r� r' r' rD �test_dotvecvecouterG s zTestDot.test_dotvecvecouterc C s< | j | j }}t�||�}t�dgg�}t||| jd� d S )Ng�".!��?)rt )rl r� r/ r. r� r r� )rJ rl r� r� r� r' r' rD �test_dotvecvecinnerM s zTestDot.test_dotvecvecinnerc C s@ t �d�}dg}t �||�}t �dddg�}t||| jd� d S )N)r r, g333333@)rt )r/ r� r. r� r r� )rJ rl r� r� r� r' r' rD �test_dotcolumnvect1S s zTestDot.test_dotcolumnvect1c C sD t �d��� }dg}t �||�}t �dddg�}t||| jd� d S )N)r r, g������@)rt )r/ r� r6 r. r� r r� )rJ rl r� r� r� r' r' rD �test_dotcolumnvect2Z s zTestDot.test_dotcolumnvect2c C s\ t j�d� t j�dd�}t j�dd�}t �||�}t �ddddgg�}t||| jd� d S ) Nr r, r[ g�#:�\�?gN�p���?g�wr*a�?g��p<�e?)rt )r/ r� r4 r� r. r� r r� )rJ rl r� r� r� r' r' rD �test_dotvecscalara s zTestDot.test_dotvecscalarc C sb t j�d� t j�dd�}t j�dd�}t �||�}t �dgdgdgdgg�}t||| jd� d S ) Nr r[ r, g��p<�e?gpVȅ��U?gX;'Ii`?g3���Tp?)rt )r/ r� r4 r� r. r� r r� )rJ rl r� r� r� r' r' rD �test_dotvecscalar2i s zTestDot.test_dotvecscalar2c C s� dddg}dddddddddg }xlt |t�||��D ]V\}\}}t�|�}t�|�}t�||�}t�|�} t|j| jk� t|| | j d� q4W d S )Nr' )r, )r, r, )rt ) �zip� itertools�productr/ r2 r. r r? r r� ) rJ �dims�doutrU Zdim1Zdim2rl r� r� r� r' r' rD r q s zTestDot.test_allc s� G � fdd�dt �� t�ddgddgg�}t�|�}t�� ddg�� ddg�g�}t�� ddg�� ddg�g�}t�||�t�||� }t|d j|d j� t|d j|d j� d S )Nc sB e Zd Zd dd�Z� fdd�Z� fdd�Z� fdd �Zd d � ZdS ) z#TestDot.test_vecobject.<locals>.VecNc S s |d kr g }t �|�| _d S )N)r/ r� )rJ �sequencer' r' rD r) ~ s z,TestDot.test_vecobject.<locals>.Vec.__init__c s � � }| j |j |_ |S )N)r� )rJ r� r� )�Vecr' rD �__add__� s z+TestDot.test_vecobject.<locals>.Vec.__add__c s � � }| j |j |_ |S )N)r� )rJ r� r� )r r' rD �__sub__� s z+TestDot.test_vecobject.<locals>.Vec.__sub__c s � | j �� �}| j |9 _ |S )N)r� r� )rJ r� r� )r r' rD �__mul__� s z+TestDot.test_vecobject.<locals>.Vec.__mul__c S s | | S )Nr' )rJ r� r' r' rD �__rmul__� s z,TestDot.test_vecobject.<locals>.Vec.__rmul__)N)r` ra rb r) r r r r r' )r r' rD r } s r g �?g @g r r, )r0 r/ r6 r� r� r. r )rJ Z U_non_contZU_contr� r2 Z zeros_testr' )r rD �test_vecobject| s zTestDot.test_vecobjectc C sr ddl m} tjddgddggtd�}tjddgddggtd�}tjddgddggtd�}|||�}t||� d S ) Nr )r. r, r& r r[ )r) r� )�numpy.core.multiarrayr. r/ r� r� r )rJ r. rI r^ r r� r' r' rD �test_dot_2args� s zTestDot.test_dot_2argsc C s� ddl m} tj�d� tj�d�}tj�d�}t�d�}x td�D ]}||||� qDW trlt t � |�d� |||d d �}t ||� t |||||d �k� |d d �df �� }|d d �df �� }|||�}t |||||�k� t ||� d S ) Nr )r. � )i re )re rf )i rf r� r& )r� )r r. r/ r� r4 � random_sampler: rp r r r� r> r r r� )rJ r. r� r� r� rv Zr2r' r' rD �test_dot_3args� s" zTestDot.test_dot_3argsc C s8 ddl m} tj�d� tj�d�}tj�d�}t�d�}tt||||� t�d�}tt||||� t�d�}tt||||� t�d �}tt||||� tt||||j � t�d �}tt||||d d �d d d �f � tt||||d d �d d �f � tjd tj d�}tt||||� tjd t d�}tt||||� d S )Nr )r. r )i re )re rf )i r )i )rf )rf i )i rg r& rf )i rf )r) ) r r. r/ r� r4 r r: r r1 r� r� r� )rJ r. r� r� r� r' r' rD �test_dot_3args_errors� s( " z TestDot.test_dot_3args_errorsc C sz t jddgddggdd�}t jddgddggdd�}t �||�}tt �||�|� tt �||�|� tt �||�|� d S )Nr, r& r r[ r( )r* r� )r/ r� r. r )rJ rI r^ r� r' r' rD �test_dot_array_order� s z TestDot.test_dot_array_orderc C sN t jddgtd�}t jddggtd�}tt �|d�|� tt �d|�|� d S )Nr, r& )r) r r{ )r/ �matrixr0 r r. )rJ r� �desiredr' r' rD �%test_dot_scalar_and_matrix_of_objects� s z-TestDot.test_dot_scalar_and_matrix_of_objectsc s ddd�� d� fdd� }dd� }� dd t j�}� d d t j�}t �||� t�d d d d�}�x�|D �]�\}}}}t j�||�} t j�|�} t �| | �} || |t j|d�} || |t j�} || | | � | d d d� }t �|| �} | d d d� }||| | � |d d �d d d�f }| d d d� }t �||�} |d d �d d d�f }| d d d� }|||| � |dk�rtt|jdd| f� nt|jd| df� t|jd� |||t j�}|||| � | d d �d d d�f }t �||�} | d d �d d d�f }|||| � |||| � q`W d S )Nr( c S s� |d�}t �| �}t j||j | t jd�}|jd d }x"t|�D ]}|| | dkrDP qDW |||||j � j|d�}|j| |d�S )Nr )r) r- )r* ) r/ r� r2 r� r; r< rp ru r} ) r? r@ r) r* r� r� rd �addressrC r' r' rD � aligned_array� s zBTestDot.test_accelerate_framework_sgemv_fix.<locals>.aligned_arrayc s( � | j |||�}| d d � |d d �< |S )N)r? )r� r@ r) r* rX )r# r' rD � as_aligned� s z?TestDot.test_accelerate_framework_sgemv_fix.<locals>.as_alignedc S s t t�| |�|ddd� d S )Ng�h㈵��>gH�����z>)�rtol�atol)r r/ r. )r/ �Xr r' r' rD �assert_dot_close� s zETestDot.test_accelerate_framework_sgemv_fix.<locals>.assert_dot_closer r )r r )r rf )i' )�� �Y )r( r� )r* r& r� rd )rd )r( )r( ) r/ r� r. r r r� r� r r� )rJ r$ r( r� ru Ztestdatar@ r� Za_orderZA_dZX_dr ZA_fZX_fZA_d_2ZA_f_2ZA_d_22ZX_d_2ZA_f_22ZX_f_2ZX_f_2cZA_d_12ZA_f_12r' )r# rD �#test_accelerate_framework_sgemv_fix� sF z+TestDot.test_accelerate_framework_sgemv_fixN)r` ra rb rK r r r r r r r r r r r r r r r r r r r! r+ r' r' r' rD r� s* ! r� c @ sL e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dS )� MatmulCommonzpCommon tests for '@' operator and numpy.matmul. Do not derive from TestCase to avoid nose running it. z?bhilqBHILQefdgFDGc C sf ddddddddd g }xJt �| j|�D ]8\}\}}tj||d �}tj||d �}tt| j||� q&W d S ) N))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 r1 �matmul)rJ r r �dm1�dm2rI r^ r' r' rD �test_exceptions7 s z MatmulCommon.test_exceptionsc C s� dddg}xTt �| j|�D ]B\}\}}tj||d�}tj||d�}| �||�}t|jdk� qW xJ| jD ]@}tjd|d�}tjd|d�}| �||�}tt�|�jdk� qhW d S )N))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� ) rJ r r r. r/ rI r^ r� r r' r' rD � test_shapesI s zMatmulCommon.test_shapesc C s� t �d�}t �d�}x~| jD ]t}|�|�}|�|�}x6||f||f||fgD ] }| j|� }t|j|k� qLW | �||�}tt|�t �|�jk� q W d S )N)r, r, )r, )r/ r� r rI r- r r) r� )rJ r� �vecr r� r� �argr� r' r' rD �test_result_types] s z MatmulCommon.test_result_typesc C sv t �ddg�}d}x6| jdd � D ]$}|�|�}| �||�}t||� q"W t jddgdd�}| �||�}t|d� d S )Nr, r& rL Tr� )r) )r/ r� r rI r- r )rJ r2 r� r �v1r� r' r' rD �test_vector_vector_valuesk s z&MatmulCommon.test_vector_vector_valuesc C sF t �ddg�}t �ddgddgg�}t j|gd dd�}t �ddg�}t j|gd dd�}x`| jdd � D ]N}|�|�}|�|�}|�|�} | �||�} t| |� | �|| �} t| |� qjW t �d d g�}t �d d gd d gg�}t j|gd dd�}t �d d g�}t j|gd dd�}| �||�} t| |� | �||�} t| |� d S ) Nr, r& r r[ r )r� r� rG TF)r/ r� �stackr rI r- r ) rJ r2 �mat1�mat2�tgt1�tgt2r r� �m1r� r� r' r' rD �test_vector_matrix_valuesx s, z&MatmulCommon.test_vector_matrix_valuesc C sF t �ddg�}t �ddgddgg�}t j|gd dd�}t �ddg�}t j|gd dd�}x`| jdd � D ]N}|�|�}|�|�}|�|�} | �||�} t| |� | �| |�} t| |� qjW t �d d g�}t �d d gd d gg�}t j|gd dd�}t �d d g�}t j|gd dd�}| �||�} t| |� | �||�} t| |� d S ) Nr, r& r r[ r )r� rL r� TF)r/ r� r7 r rI r- r ) rJ r2 r8 r9 r: r; r r� r< r� r� r' r' rD �test_matrix_vector_values� s, z&MatmulCommon.test_matrix_vector_valuesc C s� t �ddgddgg�}t �ddgddgg�}t j||gdd�}t j||gdd�}t �ddgd d gg�}t �ddgddgg�}t �ddgdd gg�}t j||gdd�}t j||fdd�} t j||fdd�} x�| jdd � D ]�} |�| �} |�| �} |�| �}|�| �}| �| | �}t||� | �| | �}t||� | �|| �}t|| � | �| |�}t|| � | �||�}t||� q�W t jddgddggt jd �} t jddgddggt jd �} t j| | gdd�}t j| | gdd�}| }| }t jddgddggt jd �}t j||gdd�}t j||fdd�} t j||fdd�} | �| | �}t||� | �| | �}t||� | �|| �}t|| � | �| |�}t|| � | �||�}t||� d S ) Nr, r& r r[ r )r� r� rG r r r{ )r) )r/ r� r7 r rI r- r rh )rJ r8 r9 Zmat12Zmat21Ztgt11Ztgt12Ztgt21Ztgt12_21Ztgt11_12Ztgt11_21r r< r� Zm12Zm21r� r' r' rD �test_matrix_matrix_values� sZ z&MatmulCommon.test_matrix_matrix_valuesN) r` ra rb r� r r0 r1 r4 r6 r= r> r? r' r' r' rD r, - s r, c @ s e Zd ZejZdd� ZdS )� TestMatmulc C s� t jdt jd�}t jdt jd�}t jddt jd�}d}t jdt jd�}| �|||� t|||d� d}t jdt jd�}| j|||d� t|||d� d}t jdt jd�}tt | j|||d� d S ) N)r& r& )r) r& zout positional argument)rh zout keyword argument)r� z out argument with illegal cast) r/ r� r� �fullr2 r- r rl r r� )rJ rI r^ r� r r� r' r' rD � test_out_arg� s zTestMatmul.test_out_argN)r` ra rb r/ r- rB r' r' r' rD r@ � s r@ )r rL c @ s" e Zd ZddlZejZdd� ZdS )�TestMatmulOperatorr Nc C sH G dd� dt �}|� }t�d�}t| �||�d� t| �||�d� d S )Nc @ s e Zd ZdZdd� Zdd� ZdS )z:TestMatmulOperator.test_array_priority_override.<locals>.Ai� c S s dS )Nr/ r' )rJ r� r' r' rD � __matmul__& s zETestMatmulOperator.test_array_priority_override.<locals>.A.__matmul__c S s dS )Nr/ r' )rJ r� r' r' rD � __rmatmul__) s zFTestMatmulOperator.test_array_priority_override.<locals>.A.__rmatmul__N)r` ra rb r� rD rE r' r' r' rD r/ # s r/ r& )r0 r/ r� r r- )rJ r/ rI r^ r' r' rD � test_array_priority_override! s z/TestMatmulOperator.test_array_priority_override)r` ra rb r7 r- rF r' r' r' rD rC s rC c C s\ t �d�} t �d�}tt| j|� dd l}tt|j| |� ttd�}tt|dt � t � � d S )Nr r �execza @= b) r/ r; r r� � __imatmul__r7 �imatmulr� rt �globalsrM )rI r^ r7 Zexec_r' r' rD �test_matmul_inplace1 s rK c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )� TestInnerc C s6 d}t jddd�}ttt j||� ttt j||� d S )Ng �?)r, r, zi,i)r) )r/ r� r r� �inner)rJ r r/ r' r' rD �test_inner_type_mismatchB s z"TestInner.test_inner_type_mismatchc C s� xzt jd t jd d D ]^}t jd|d�d }t jddg|d�}t jdd g|d�}tt �||�|� tt �||�|� qW d S ) Nr� r r� r )r) r' r, r& r{ )r/ r r� r rM )rJ r �scar2 r r' r' rD � test_inner_scalar_and_vectorI s z&TestInner.test_inner_scalar_and_vectorc C s� x�t jd t jd d D ]n}t jd|d�d }t jddgdd gg|d�}t jdd gd d gg|d�}tt �||�|� tt �||�|� qW d S ) Nr� r r� r )r) r' r, r& r[ r{ r� r� )r/ r r� r r rM )rJ r rO r� r r' r' rD � test_inner_scalar_and_matrixQ s z&TestInner.test_inner_scalar_and_matrixc C sN t jddgtd�}t jddggtd�}tt �|d�|� tt �d|�|� d S )Nr, r& )r) r r{ )r/ r r0 r rM )rJ r� r r' r' rD �'test_inner_scalar_and_matrix_of_objectsY s z1TestInner.test_inner_scalar_and_matrix_of_objectsc C s. t jdt jd�}t �||�}t|ddd� d S )N)r, r )r? r) r r� )rt )r/ r2 r rM r )rJ rI r r' r' rD � test_vecself` s zTestInner.test_vecselfc C s~ �xvt jd t jd d D �]X}t jddgddgg|d�}t jddgddgg|d�}t jddg|d�}t jdd g|d�}tt �|j|�|� tt �||j�|� tt �||�|� tt �||�|� t jd d gd d gg|d�}tt �||�|� t jddgddgg|d�}tt �||�|� tt �||�� �|� t �d��|�}|d d d� }t jd |d�� � }tt �||�|� q W d S )Nr� r r� r, r& r r[ )r) r{ r� rG r r rL r� � r. ) r/ r r� r rM r� r� rH rI r� )rJ r r/ rN r( r rI r^ r' r' rD �,test_inner_product_with_various_contiguitiesh s$ "z6TestInner.test_inner_product_with_various_contiguitiesc C s �xt jd t jd d D ]�}t �d��ddd��|�}t �dd��ddd��|�}t jd d d gd d dggdddgdddggdddgdddgggdd d gd d d ggd!d"d#gd$d%d&ggd'd(d)gd*d+d,gggg|d-�}tt �||�|� tt �||��ddd.d/�|� q W d S )0Nr� r r� r� r& r r[ �0 � � �� �� � i i6 i� i� i> i� i� i� if i� i� i. i� if i> i i� i� i� i� i i. iF i^ iv i� i� iF i� i� iN )r) r r, ) r/ r rH r} rI r� r rM r6 )rJ r rI r^ r r' r' rD �test_3d_tensor� s& zTestInner.test_3d_tensorN) r` ra rb rN rP rQ rR rS rU r\ r' r' r' rD rL @ s rL c @ s e Zd Zdd� Zdd� ZdS )�TestSummarizationc C s6 t �d�}d}tt|�|k� d}tt|�|k� d S )Ni� z$[ 0 1 2 ..., 998 999 1000]z0array([ 0, 1, 2, ..., 998, 999, 1000]))r/ rH r r� r� )rJ r/ �strA�reprAr' r' rD �test_1d� s zTestSummarization.test_1dc C s> t �d��dd�}d}tt|�|k� d}tt|�|k� d S )Ni� r& i� zL[[ 0 1 2 ..., 498 499 500] [ 501 502 503 ..., 999 1000 1001]]zdarray([[ 0, 1, 2, ..., 498, 499, 500], [ 501, 502, 503, ..., 999, 1000, 1001]]))r/ rH r} r r� r� )rJ r/ r^ r_ r' r' rD �test_2d� s zTestSummarization.test_2dN)r` ra rb r` ra r' r' r' rD r] � s r] c @ s e Zd Zdd� Zdd� ZdS )�TestAlenc C s� t �dddg�}| �t �|�d� t �dddgdddgg�}| �t �|�d� dddg}| �t �|�d� dddgdddgg}| �t �|�d� d S )Nr, r& r r[ rL r� )r/ r� r� �alen)rJ r� r' r' rD r# � s zTestAlen.test_basicc C s | � t�d�d� d S )NrL r, )r� r/ rc )rJ r' r' rD �test_singleton� s zTestAlen.test_singletonN)r` ra rb r# rd r' r' r' rD rb � s rb c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestChoosec C s` dt jdtd� | _dt jdtd� | _dt jdtd� | _dt jdtd� | _dddg| _d S )Nr& )r )r) r )r& r r r, )r/ r� r� r� r� r� r� r* )rJ r' r' rD rK � s zTestChoose.setUpc C s* t �| j| j| jf�}t|dddg� d S )Nr& r )r/ r� r* r� r� r )rJ r/ r' r' rD r# � s zTestChoose.test_basicc C s4 t �| j| j| jf�}t|dddgdddgg� d S )Nr& r )r/ r� r* r� r� r )rJ r/ r' r' rD �test_broadcast1� s zTestChoose.test_broadcast1c C s4 t �| j| j| jf�}t|dddgdddgg� d S )Nr& r )r/ r� r* r� r� r )rJ r/ r' r' rD �test_broadcast2� s zTestChoose.test_broadcast2N)r` ra rb rK r# rf rg r' r' r' rD re � s re c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TestRepeatc C s* t �ddddddg�| _| j�d�| _d S )Nr, r& r r[ rL r{ )r& r )r/ r� r� r} r� )rJ r' r' rD rK � s zTestRepeat.setUpc C s< t �| jddddddg�}t|ddddddddddg � d S )Nr, r r& r[ rL r{ )r/ r� r� r )rJ r/ r' r' rD r# � s zTestRepeat.test_basicc C s4 t �| jd�}t|ddddddddddddg � d S )Nr& r, r r[ rL r{ )r/ r� r� r )rJ r/ r' r' rD rf � s zTestRepeat.test_broadcast1c C sz t j| jddgdd�}t|dddgdddgdddgg� t j| jdddgdd�}t|ddddddgddddddgg� d S ) Nr& r, r )r� r r[ rL r{ )r/ r� r� r )rJ r/ r' r' rD �test_axis_spec� s zTestRepeat.test_axis_specc C sx t j| jddd�}t|dddgdddgdddgdddgg� t j| jddd�}t|ddddddgddddddgg� d S ) Nr& r )r� r, r r[ rL r{ )r/ r� r� r )rJ r/ r' r' rD rg � s zTestRepeat.test_broadcast2N)r` ra rb rK r# rf ri rg r' r' r' rD rh � s rh r, r[ )�zeror| �constant�circular�mirrorc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd S ) �TestNeighborhoodIterc C s� t jddgddgg|d�}t jdddgdddgg|d�t jdddgdddgg|d�t jdddgdddgg|d�t jdddgdddgg|d�g}t|ddddg|d td �}t||� t jdddgdddgg|d�t jdddgdddgg|d�t jdddgdddgg|d�t jdddgdddgg|d�g}t|ddddg|d td �}t||� t jd d d gd ddgg|d�t jd d d gddd gg|d�t jd ddgd ddgg|d�t jddd gddd gg|d�g}t|ddddgd td �}t||� d S ) Nr r, r& r )r) r. rj r| r[ rk )r/ r� r � NEIGH_MODEr )rJ r r� r� r� r' r' rD �_test_simple2d s, z#TestNeighborhoodIter._test_simple2dc C s | � tj� d S )N)rp r/ r� )rJ r' r' rD � test_simple2d s z"TestNeighborhoodIter.test_simple2dc C s | � t� d S )N)rp r )rJ r' r' rD �test_simple2d_object! s z)TestNeighborhoodIter.test_simple2d_objectc C s� t jddgddgg|d�}t jdddgdddgg|d�t jdddgdddgg|d�t jdddgdddgg|d�t jdddgdddgg|d�g}t|ddddg|d td �}t||� d S )Nr r, r& r )r) r. rm )r/ r� r ro r )rJ r r� r� r� r' r' rD �_test_mirror2d$ s z#TestNeighborhoodIter._test_mirror2dc C s | � tj� d S )N)rs r/ r� )rJ r' r' rD � test_mirror2d. s z"TestNeighborhoodIter.test_mirror2dc C s | � t� d S )N)rs r )rJ r' r' rD �test_mirror2d_object1 s z)TestNeighborhoodIter.test_mirror2d_objectc C s t �ddd��|�}dddgdddgdddgdddgdddgg}t|ddg|d td �}t||� dddgdddgdddgdddgdddgg}t|ddg|d td �}t||� |d ddgdddgdddgdddgdd|d gg}t|ddg|d td �}t||� d S ) Nr, rL r r& r r[ r. rj r| rk )r/ r� rI r ro r )rJ r r� r� r� r' r' rD � _test_simple5 s , , 4z!TestNeighborhoodIter._test_simplec C s | � tj� d S )N)rv r/ r� )rJ r' r' rD �test_simple_floatD s z&TestNeighborhoodIter.test_simple_floatc C s | � t� d S )N)rv r )rJ r' r' rD �test_simple_objectG s z'TestNeighborhoodIter.test_simple_objectc s� t �ddd��� �}t jdddddgdddddgdddddgdddddgdddddgg� d�}t|ddg|d td �}| �� fd d �|D �� t||� d S ) Nr, rL r& r r[ )r) r� rm c s g | ]}|j � k�qS r' )r) )rq rv )r r' rD rr P s z5TestNeighborhoodIter._test_mirror.<locals>.<listcomp>)r/ r� rI r� r ro r� r )rJ r r� r� r� r' )r rD � _test_mirrorK s ("z!TestNeighborhoodIter._test_mirrorc C s | � tj� d S )N)ry r/ r� )rJ r' r' rD � test_mirrorS s z TestNeighborhoodIter.test_mirrorc C s | � t� d S )N)ry r )rJ r' r' rD �test_mirror_objectV s z'TestNeighborhoodIter.test_mirror_objectc C s� t �ddd��|�}t jdddddgdddddgdddddgdddddgdddddgg|d�}t|ddg|d td �}t||� d S ) Nr, rL r[ r& r )r) r� r rl )r/ r� rI r� r ro r )rJ r r� r� r� r' r' rD �_test_circularZ s ("z#TestNeighborhoodIter._test_circularc C s | � tj� d S )N)r| r/ r� )rJ r' r' rD � test_circulara s z"TestNeighborhoodIter.test_circularc C s | � t� d S )N)r| r )rJ r' r' rD �test_circular_objectd s z)TestNeighborhoodIter.test_circular_objectN)r` ra rb rp rq rr rs rt ru rv rw rx ry rz r{ r| r} r~ r' r' r' rD rn s rn c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestStackedNeighborhoodIterc C s6 t j}t jdddg|d�}t jdg|d�t jdg|d�t jdg|d�t jdg|d�t jdg|d�t jdg|d�t jdg|d�g}t|ddgtd ddgtd �}t||� t jdddg|d�t jdddg|d�t jdddg|d�t jdddg|d�t jdddg|d�g}t|d dgtd d dgtd �}t||� d S ) Nr, r& r )r) r r� r[ rj r. r| )r/ r r� r ro r )rJ r r� r� r� r' r' rD �test_simple_constj s( z-TestStackedNeighborhoodIter.test_simple_constc C s� t j}t jdddg|d�}t jdddg|d�t jdddg|d�t jdddg|d�t jdddg|d�t jdddg|d�g}t|ddgtd ddgtd �}t||� t jdddg|d�}t jdddg|d�t jdddg|d�t jdddg|d�t jdddg|d�t jdddg|d�g}t|ddgtd d dgtd �}t||� t jdddg|d�}t jdddg|d�t jdddg|d�t jdddg|d�t jdddg|d�t jdddg|d�g}t|ddgtd ddgtd �}t||� t jdddg|d�}t jdddddg|d�t jdddddg|d�t jdddddg|d�t jdddddg|d�t jdddddg|d�g}t|ddgtd d dgtd �}t||� d S ) Nr, r& r )r) r r. rm rj r� )r/ r r� r ro r )rJ r r� r� r� r' r' rD �test_simple_mirror� sJ z.TestStackedNeighborhoodIter.test_simple_mirrorc C s� t j}t jdddg|d�}t jdddg|d�t jdddg|d�t jdddg|d�t jdddg|d�t jdddg|d�g}t|ddgtd ddgtd �}t||� t jdddg|d�}t jdddg|d�t jdddg|d�t jdddg|d�t jdddg|d�t jdddg|d�g}t|ddgtd d dgtd �}t||� t jdddg|d�}t jdddg|d�t jdddg|d�t jdddg|d�t jdddg|d�t jdddg|d�g}t|ddgtd ddgtd �}t||� t jdddg|d�}t jdddddg|d�t jdddddg|d�t jdddddg|d�t jdddddg|d�t jdddddg|d�g}t|ddgtd d dgtd �}t||� d S ) Nr, r& r )r) r r. rl rj r� )r/ r r� r ro r )rJ r r� r� r� r' r' rD �test_simple_circular� sJ z0TestStackedNeighborhoodIter.test_simple_circularc C s t j}t jdddg|d�}t jddddg|d�g}t|ddgtd ddgtd �}t||� t jdddg|d�}t jddddg|d�g}t|ddgtd ddgtd �}t||� t jdddg|d�}t jddddg|d�g}t|ddgtd ddgtd �}t||� d S ) Nr, r& r )r) r rj r. rm rl )r/ r r� r ro r )rJ r r� r� r� r' r' rD �test_simple_strict_within� s z5TestStackedNeighborhoodIter.test_simple_strict_withinN)r` ra rb r� r� r� r� r' r' r' rD r h s 00r c @ s e Zd Zdd� ZdS )� TestWarningsc C sf t �ddg�}t �ddg�}t�� �8 t�dt j� tt j|jtd �|� t |ddg� W d Q R X d S )Nr, r& y �? �y �? @�error) r/ r� r� r� r� �ComplexWarningr r� �slicer )rJ r� r� r' r' rD �test_complex_warning s z!TestWarnings.test_complex_warningN)r` ra rb r� r' r' r' rD r� � s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestMinScalarTypec C s" t �d�}t �d�}t||� d S )Nrs r; )r/ �min_scalar_typer) r )rJ r �wantedr' r' rD �test_usigned_shortshort s z)TestMinScalarType.test_usigned_shortshortc C s" t �d�}t �d�}t||� d S )Ni�� rk )r/ r� r) r )rJ r r� r' r' rD �test_usigned_short s z$TestMinScalarType.test_usigned_shortc C s" t �d�}t �d�}t||� d S )Nl �� rm )r/ r� r) r )rJ r r� r' r' rD �test_usigned_int s z"TestMinScalarType.test_usigned_intc C s" t �d�}t �d�}t||� d S )Nl ���� ro )r/ r� r) r )rJ r r� r' r' rD �test_usigned_longlong s z'TestMinScalarType.test_usigned_longlongc C s" t �d�}t �d�}t||� d S )Nl rM )r/ r� r) r )rJ r r� r' r' rD rQ s z TestMinScalarType.test_objectN)r` ra rb r� r� r� r� rQ r' r' r' rD r� s r� )�_dtype_from_pep3118c @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dS )�TestPEP3118Dtypec C s, t �|�}t|�}t||d||f d� d S )Nzspec %r != dtype %r)rh )r/ r) r� r )rJ �specr� r �actualr' r' rD �_check* s zTestPEP3118Dtype._checkc C s| t �d�j}xjtd�D ]^}|dkr(d}nd| }| �d| dd|d|| fd �� | �d | ddd| fd �� qW d S ) Nrv rd r r� zb%dxirs )r9 r r, )rZ r� r& )r/ r) r3 rp r� )rJ r@ r? ru r' r' rD �test_native_padding0 s z$TestPEP3118Dtype.test_native_paddingc C s< | � ddddidfdfi� | � ddddidfdfi� d S ) Nzx3T{xi}rZ )rv r[ )r r[ z^x3T{xi})rv r, r, )r� )rJ r' r' rD �test_native_padding_2< s z&TestPEP3118Dtype.test_native_padding_2c sx t �d�j� t �d�j}� fdd�}tdgdgd�}| �dtf d||d �i|��� | �d tf d||d �i|��� | �d tf d||d �i|��� | �d tf d||d �i|��� | �dtf d||d �i|��� | �dtf d|d i|��� | �dtf d|d i|��� | �dtf d|d i|��� | �dtf d|d i|��� | �dtf d|d i|��� d S )Nrv c s � d| d � S )Nr, r' )r� )r@ r' rD rX G s z7TestPEP3118Dtype.test_trailing_padding.<locals>.alignedrZ )r� r� �ixr9 r, Zixxr& Zixxxr Zixxxxr[ Zi7xr� z^ixz^ixxz^ixxxz^ixxxxz^i7x)r/ r) r3 r9 r] r� )rJ rA rX r� r' )r@ rD �test_trailing_paddingA s """"" z&TestPEP3118Dtype.test_trailing_paddingc C s` t jdddt �d�fdgdd�}| �d|� t �ddd d d dt jddd�fg�}| �d |� d S ) N)rI r^ )r^ rv rq zb,i)r rv T)r@ z(T{b:a:xxxi:b:T{b:f0:=i:f1:}:sub:xxxi:c:})r r^ )r� r^ )r� r^ z-T{b:a:=i:b:b:c:b:d:b:e:T{b:f0:xxxi:f1:}:sub:})r/ r) r� )rJ r r' r' rD �test_native_padding_3X s z&TestPEP3118Dtype.test_native_padding_3c C s& t jddddgdd�}| �d|� d S )N)rI r^ )r^ rv )r r^ )r )r� rv T)r@ zT{b:a:xxxi:b:3b:c:xi:d:})r/ r) r� )rJ r r' r' rD �%test_padding_with_array_inside_structd s z6TestPEP3118Dtype.test_padding_with_array_inside_structc C s | � dddidfdd�� d S )Nz@T{^i}xirZ )rv r r )rv rL )rZ r� )r� )rJ r' r' rD � test_byteorder_inside_structk s z-TestPEP3118Dtype.test_byteorder_inside_structc sR t �d�j� t �d�j}� fdd�}| �dtdgdgdg||d �d�d f� d S ) Nrv c s � d| d � S )Nr, r' )r� )r@ r' rD rX v s z4TestPEP3118Dtype.test_intra_padding.<locals>.alignedz(3)T{ix}rZ r r, )r� r� r� r9 )r )r/ r) r3 r9 r� r] )rJ rA rX r' )r@ rD �test_intra_paddingq s z#TestPEP3118Dtype.test_intra_paddingc C s4 t �d�}| �d|� t �ddg�}| �d|� d S )Nr )rZ �S1)r[ )r� �S4Z4c4s)r/ r) r� )rJ r r' r' rD �test_char_vs_string� s z$TestPEP3118Dtype.test_char_vs_stringc C s$ | � dddg� | � dddg� d S )Nz (0)I:a:f:b:)rI rV )r )r^ r� z (0)I:b:f:a:)r^ rV )r )rI r� )r� )rJ r' r' rD �test_field_order� s z!TestPEP3118Dtype.test_field_orderc C s> | � dddg� | � dddg� | � dd� | � ddg� d S )N�ii)rZ rv )r� rv zii:f0:rv zi:f0:)r� )rJ r' r' rD �test_unnamed_fields� s z$TestPEP3118Dtype.test_unnamed_fieldsN)r` ra rb r� r� r� r� r� r� r� r� r� r� r� r' r' r' rD r� ) s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!S )"�TestNewBufferProtocolc C s� t �|�}t|�}t �|�}t �|�}t|jj � t|jj� t|j|j� t|j |j � t ||� t|j|j� t|j |j � t ||� d S )N) r/ r� � memoryviewr� r rN rW r r) r? r )rJ r� r� r� r� r' r' rD �_check_roundtrip� s z&TestNewBufferProtocol._check_roundtripc C s> t jdddddgdd�}| �|� t jddgddggt jd�}| �|� t jdt jd�d d �d d d �f }| �|� d d d d dddddddt jfdt jfdt jfdt j fdt j fdt j fddd d d t j fg}t jd g|d�}| �|� t jddgddggfd t d!ffgd�}| �|� t jdddgd"d�}| �|� t jdddgd#d�}| �|� t jdddgd$d�}| �|� t jdddgd%d�}| �|� t jdddgd&d�}| �|� tjd'k�r�t jdddgd(d�}tt| j|� t jdddgd)d�}| �|� n@t jdddgd(d�}| �|� t jdddgd)d�}tt| j|� d S )*Nr, r& r r[ rL r] )r) )r r r r )rI r^ )r^ r� )r rv )r� r� )�dxr� )r� rN )r� �H)re rV )r� �L)�hx�Qrv r? r r� �jx�kx)r� r� )r� �U4)r� �V3)r r� r )r, r, r, r, r, r, r, r, r, r, r, r, r, r, r, r, s aaaa�bbbbs xxxTg �?rI )r& r& z>i2z<i2z>i4z<i4z>qr� z>gz<g)r/ r� r� r r2 r� �single�doubler� �csingle�cdoubler� �halfr� r� r� r r1 )rJ r� r r' r' rD r� � sh " & z$TestNewBufferProtocol.test_roundtripc C sT dddddddt d�t d �d g }tj|d d �}| �|� tj|d d �}| �|� d S )Ng �?g �g ��@g ?g p>g g �z+infz-infg T�?z>e)r) z<e)r� r/ r� r� )rJ Z half_listr� r' r' rD �test_roundtrip_half� s z)TestNewBufferProtocol.test_roundtrip_halfc C s� x�t j�� D ]�}t �|�}|jdkr&q |jdkr2q t jd|d�}| �|� |jdkr |�d�}t jd|d�}| �|� |�d�}t jd|d�}| �|� q W d S )N�Mmr r[ )r) ZqQgGr( r' )r/ �typeDictr< r) r� r2 r� rK )rJ �typr) r� r r' r' rD �test_roundtrip_single_types� s z1TestNewBufferProtocol.test_roundtrip_single_typesc C s | � d� d S )Nr )r� )rJ r' r' rD �test_roundtrip_scalar s z+TestNewBufferProtocol.test_roundtrip_scalarc C sl t jdddddgdd�}t|�}t|jd� t|jd� t|jd� t|jd � t|jt � t|j d� d S ) Nr, r& r r[ rL rv )r) )rL )r[ ) r/ r� r� r r� r? r� r� � suboffsets�EMPTYr9 )rJ r� r� r' r' rD �test_export_simple_1d s z+TestNewBufferProtocol.test_export_simple_1dc C sp t jddgddggt jd�}t|�}t|jd� t|jd� t|jd� t|jd� t|j t � t|j d � d S ) Nr, r& r r[ )r) r� )r& r& )re rd rd ) r/ r� r r� r r� r? r� r� r� r� r9 )rJ r� r� r' r' rD �test_export_simple_nd" s z+TestNewBufferProtocol.test_export_simple_ndc C sv t jdt jd�d d �dd d �f }t|�}t|jd� t|jd� t|jd� t|jd� t|j t � t|j d� d S ) N)r r r )r) r r� )r r r& )r� r[ r[ ) r/ r2 r� r� r r� r? r� r� r� r� r9 )rJ r� r� r' r' rD �test_export_discontiguous, s " z/TestNewBufferProtocol.test_export_discontiguousc C s ddddddddd d d t jfd t jfd t jfdt jfdt jfdt jfdddddt jfg}t jdg|d�}t |�}t |j d� t |j d� t |j t� tdd� |D ��}t �d �jd kr�t |jd � n t |jd � t �d�jd t �t j�jks�t |j|f� t |j|� d S )!N)rI r^ )r^ r� )r rv )r� r� )r� r� )r� rN )r� r� )re rV )r� r� )r� r� rv 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, s aaaar� s Tg �?)r) )r, r, c S s g | ]\}}t �|�j�qS r' )r/ r) r9 )rq rI r^ r' r' rD rr V s z<TestNewBufferProtocol.test_export_record.<locals>.<listcomp>r� r[ zjT{b:a:=h:b:i:c:l:d:q:dx:B:e:@H:f:=I:g:L:h:Q:hx:f:i:d:j:^g:k:=Zf:ix:Zd:jx:^Zg:kx:4s:l:=4w:m:3x:n:?:o:@e:p:}zjT{b:a:=h:b:i:c:q:d:q:dx:B:e:@H:f:=I:g:Q:h:Q:hx:f:i:d:j:^g:k:=Zf:ix:Zd:jx:^Zg:kx:4s:l:=4w:m:3x:n:?:o:@e:p:}r )r/ r� r� r� r� r� r� r� r� r� r r? r� r� r� r r) r9 r� r� r� r r rr )rJ r r� r� �szr' r' rD �test_export_record6 sF z(TestNewBufferProtocol.test_export_recordc C sr t jddgddggfdgd�}t|�}t|jd� t|jt� t|jd� t|jt� t|j t� t|j d � d S ) Nr, r& r r[ )rI )rv )r& r& )r) z T{(2,2)i:a:}r re ) r/ r� r� r r� r? r� r� r� r� r9 )rJ r� r� r' r' rD �test_export_subarray` s z*TestNewBufferProtocol.test_export_subarrayc C s� t jdddgdd�}t|�}tjdkr4t|jd� n t|jd� t jdddgdd�}t|�}tjdkrtt|jd� n t|jd� d S ) Nr, r& r z>i)r) r� rv z<i)r/ r� r� r� r� r r� )rJ r� r� r' r' rD �test_export_endianj s z(TestNewBufferProtocol.test_export_endianc C s" t ttt�d�d d d� d� d S )NrL r& )ZSIMPLE)r r1 r r/ rH )rJ r' r' rD �test_export_flagsy s z'TestNewBufferProtocol.test_export_flagsc C s: x4t d�D ](}tjddgdt|fid�}| �|� q W d S )Nrd )r, )r& rZ )r) )rp r/ r� r� r� )rJ r? r� r' r' rD � test_padding} s z"TestNewBufferProtocol.test_paddingc C sP t rt�tjj�}t�d�}t|�}t�|�}t rJt�tjj�}t ||� ~d S )Nr[ ) r r� r> r/ �core� _internalr2 r� r� r )rJ Zcount_1rI r^ r Zcount_2r' r' rD �test_reference_leak� s z)TestNewBufferProtocol.test_reference_leakc C s� t jdddt �d�fdgdd�}t j|jt jd��|�}| �|� t jddd d gdd�}t j|jt jd��|�}| �|� t �ddd d d dt jddd�fg�}t j|jt jd��|�}| �|� d S )N)rI r^ )r^ rv rq zb,i)r rv T)r@ )r) )r r^ )r )r� rv )r r^ )r� r^ )r� r^ )r/ r) rH r9 ri ru r� )rJ �dt1r� �dt2r� �dt3r� r' r' rD �test_padded_struct_array� s z.TestNewBufferProtocol.test_padded_struct_arrayc C s� t jddd�}t jddd�jjr&d|_tt|�jdk� t�� }|� |j � |j }tt|�jd k� t �d �}|jjr�t |d g�\}}t|d d k� t jddd�}t |dg�\}}t|d d k� d S )N)r, rG rG ra )r) )rG r, r( )r* )r. r rd )i r rd )rd r i )r, rG � F_CONTIGUOUSr rd r� � C_CONTIGUOUSr. ) r/ r� rN r� r� r r� r^ r_ r� r- r� r )rJ r �fdr� r� r? r� r' r' rD �test_relaxed_strides� s z*TestNewBufferProtocol.test_relaxed_stridesN)r` ra rb r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r' r' r' rD r� � s F * r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestArrayAttributeDeletionc C sZ t �d�}dddddddg}t� �.}|�td � x|D ]}ttt||� q6W W d Q R X d S ) Nr& r? r� r- r) r� r� r� z Assigning the 'data' attribute)r/ r� r# r= r� r r� �delattr)rJ rI �attrr@ ru r' r' rD �,test_multiarray_writable_attributes_deletion� s zGTestArrayAttributeDeletion.test_multiarray_writable_attributes_deletionc C sF t �d�}dddddddd d d d d g }x|D ]}ttt||� q,W d S )Nr& r� rN r9 rA r� r� r r� r< �__array_struct__r� r� )r/ r� r r� r� )rJ rI r� ru r' r' rD �0test_multiarray_not_writable_attributes_deletion� s zKTestArrayAttributeDeletion.test_multiarray_not_writable_attributes_deletionc C s6 t �d�j}dddg}x|D ]}ttt||� q W d S )Nr& rY rX rO )r/ r� rN r r� r� )rJ rI r� ru r' r' rD �1test_multiarray_flags_writable_attribute_deletion� s zLTestArrayAttributeDeletion.test_multiarray_flags_writable_attribute_deletionc C sF t �d�j}dddddddd d d d g }x|D ]}ttt||� q,W d S ) Nr& � contiguousr� r� r� rW rU rV rT rR rS r� )r/ r� rN r r� r� )rJ rI r� ru r' r' rD �5test_multiarray_flags_not_writable_attribute_deletion� s zPTestArrayAttributeDeletion.test_multiarray_flags_not_writable_attribute_deletionN)r` ra rb r� r� r� r� r' r' r' rD r� � s r� c C s� G dd� dt �} | d�}tt�|�d� tt�|g�dg� tt�||g�ddg� tt�|�jt�d�� d|jd< tt�|�d� d |jd< tttj|� d|jd< tt�|�dgg� d|jd< tttj|� G d d � d t �}tt�|� �d � d S ) Nc @ s( e Zd Zdd� Zdd� Zedd� �ZdS )z!test_array_interface.<locals>.Fooc S s || _ ddi| _d S )N�typestrz=f8)r� �iface)rJ r� r' r' rD r) � s z*test_array_interface.<locals>.Foo.__init__c S s t | j�S )N)r� r� )rJ r' r' rD � __float__� s z+test_array_interface.<locals>.Foo.__float__c S s | j S )N)r� )rJ r' r' rD r< � s z5test_array_interface.<locals>.Foo.__array_interface__N)r` ra rb r) r� �propertyr< r' r' r' rD �Foo� s r� g �?z=f8r' r? )r, r, )r& c @ s e Zd Ze�d�ZejZdS )z'test_array_interface.<locals>.ArrayLiker, N)r` ra rb r/ r� r< r' r' r' rD � ArrayLike� s r� r, ) r0 r r/ r� r) r� r r� r1 )r� r� r� r' r' rD �test_array_interface� s r� c C s` t �ddgddgddgdd��} t jd| d �}t �|jd �}t �|jd �}t|j|j� d S ) Nr/ rN r8 r rd re )r� r� r� r9 rG )r) �descrr� )r/ r) r� r< r r9 )Zmy_dtyperI Zdescr_tZ typestr_tr' r' rD � test_array_interface_itemsize s r� c C sH t �d�j} y| d= | dd�= W n" tk r4 Y n t�Y nX d S )Nr r, r& )r/ r� r� r� rE )�itr' r' rD �test_flat_element_deletion s r� c C s( t jdddgd�} tt| d jd� d S )Nr& )r� r� )r� r� )r) r r� )r/ r2 r r1 � __delitem__)rI r' r' rD � test_scalar_element_deletion s r� c @ s e Zd Zdd� ZdS )�TestMemEventHookc C s$ t � t�d�}~t�� t� d S )Ni� )r r/ r2 �gc�collectr )rJ rI r' r' rD �test_mem_seteventhook s z&TestMemEventHook.test_mem_seteventhookN)r` ra rb r� r' r' r' rD r� s r� c @ s e Zd Zdd� ZdS )� TestMapIterc C s� t �d��d��t�}ddddgddddgf}dddd g}t|||� t|d d d d gddddgddddgg� t �d��t�}t �dddg�f}dddg}t|||� t|ddddddg� d S ) Nr� )r r[ r, r& r r r� r� re g g �?g @g 3@g Z@g @g @g @g @g "@g D@g &@r{ r[ gfffffY@g �I@g @g @)r/ rH r} rI r� r r r� )rJ rI r$ �valsr^ r' r' rD � test_mapiter' s zTestMapIter.test_mapiterN)r` ra rb r� r' r' r' rD r� & s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestAsCArrayc C s, t jdt jd�}t|d�}t|d |� d S )Nr� )r) r )r/ rH r� r r )rJ r� �from_cr' r' rD � test_1darray= s zTestAsCArray.test_1darrayc C s6 t jdt jd��dd�}t|dd�}t|d |� d S )Nr� )r) r rd r& r[ )r& r[ )r/ rH r� r} r r )rJ r� r� r' r' rD � test_2darrayB s zTestAsCArray.test_2darrayc C s: t jdt jd��ddd�}t|ddd�}t|d |� d S )Nr� )r) r& r r[ r, )r, r& r )r/ rH r� r} r r )rJ r� r� r' r' rD � test_3darrayG s zTestAsCArray.test_3darrayN)r` ra rb r� r� r� r' r' r' rD r� < s r� c @ s e Zd Zdd� Zdd� ZdS )�TestConversionc C s� x�t jd D ]�}tdt jd|d�kd|f � tdt jd|d�k d|f � xht jd D ]Z}tt jd|d�t jd|d�kd||f � tt jd|d�t jd|d�k d||f � qZW q W x�dD ]�}tdt jd|d�k d|f � tdt jd|d�k d|f � tdt jd|d�kd|f � x�d D ]�}tt jd|d�t jd|d�kd||f � tt jd|d�t jd|d�k d||f � tt jd|d�t jd|d�kd||f � �q(W q�W �xd t jd D ]�}tdt jd|d�kd|f � tdt jd|d�k d|f � tdt jd|d�kd|f � x�d t jd D ]�}tt jd|d�t jd|d�kd||f � tt jd|d�t jd|d�k d||f � tt jd|d�t jd|d�kd||f � �q8W �q�W d S ) Nr� r, r )r) ztype %s failedztype %s and %s failed�BHILQPr. �bhilqpZbhlqpr� )r/ r r r� )rJ r� r� r' r' rD �&test_array_scalar_relational_operationN s< z5TestConversion.test_array_scalar_relational_operationc C s� t tt�dg��d� t tt�dg��d� t tt�dgg��d� tttt�ddg�� G dd� dt�}tttt�|� �� tttt�|� g�� t�d g�}||d< yt}W n t k r� t }Y nX t|t|� d S ) NFTr� r, r& c @ s e Zd Zdd� ZeZdS )z:TestConversion.test_to_bool_scalar.<locals>.NotConvertiblec S s t �d S )N)r� )rJ r' r' rD �__bool__~ s zCTestConversion.test_to_bool_scalar.<locals>.NotConvertible.__bool__N)r` ra rb r� � __nonzero__r' r' r' rD �NotConvertible} s r� r ) r r� r/ r� r r1 r0 r� �RecursionError� NameErrorr� )rJ r� Zself_containing�Errorr' r' rD �test_to_bool_scalarw s z"TestConversion.test_to_bool_scalarN)r` ra rb r� r� r' r' r' rD r� M s )r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )� TestWherec C s t jt jt jt jt jt jt jt jg}�x�|D �]�}t j dt jd�}t t � ||d�|d��|d�� t t � | |d�|d��|d�� t t � d|d�|d��|d�� t t � d|d�|d��|d�� t � |�� |�}t �|�}|� |�}d|d< |d |d< t t � |||�|� t t � |||�|� t t � |||d �|� t t � ||d |�|� t t � |d d d� |d d d� |d d d� �|d d d� � t t � |dd d� |dd d� |dd d� �|dd d� � t t � |d d d � |d d d � |d d d � �|d d d � � t t � |dd d � |dd d � |dd d � �|dd d � � t t � |d d d � |d d d � |d d d � �|d d d � � t t � |d d d � |d d d � |d d d � �|d d d � � t t � |dd d � |dd d � |dd d � �|dd d � � q,W d S ) N�5 )r) r r, TFr� r& r r� r� )r/ r� rj rl rn r� r� r� r� r� r �where� ones_likerI r )rJ �dtsr r r� r� r� r' r' rD r# � s0 " <<<<<<zTestWhere.test_basicc C s0 t t�dd d �t�d �� tjg td��dd�}tjg tjd��dd�}t t�|d|�t�g ��dd�� t�ddddd d d d d dddddddddddg�}td�}tjdd|d ||d d ||d ||d ||d!|d"d#gtd�}tjddd$dd$d$ddd$d$dd$d$dd$d$dd$ddgtd�}|d d � }|t�|� |t�|�< t t�|||�|� |d d � }|t�| � |t�| �< t t�|||�|� t t�|||�|� tjd%d>j d�}td�}t t�d||�j tj � td'�}t t�d||�j tj � td(�}t t�d||�j tj � td)�}t t�d||�j tj� d S )*NT)r) r r gq= ףp��g{�G�zĿgH�z�G�gףp= �ӿg{�G�z��gffffff�g g�x�&1�?g/�$��?g ףp= ǿg�x�&1�?gh��|?5�?g/�$��?gףp= ��?g��|?5^�?gh��|?5�?g��ʡE�?g Zd;�O�?g= ףp=��g^�I +�?�NaNZ5zZ0lZWzZXq�csZQNZFdZkpZ36r9 r, g �?g @�Infinityz -Infinityg��P.5�_) r r/ r r� r� r} r r� r0 r� r r) )rJ r� r^ r� r� r� r� r' r' rD � test_exotic� s: " zTestWhere.test_exoticc C s� ddg}t �d�}t �d�}t �t �|�d d �t jf ||�}t|d |d � t|d |d � |j}|j}t �|||�}t|d d �df |d d �df � t|d d �df |d d �df � d S )NTF)r& rT r r, )r/ r2 r� r r� r� r r� )rJ r rI r^ r� r' r' rD � test_ndim� s ""zTestWhere.test_ndimc C s t �ddddddddddddg �}t �d�}t jdddddd dd d d ddg t jd �}t jdd dddd d d d d d dg t jd �}tt �|||�|� |�t j�}|�t j�}tt �|||�|� |�t j �}d||dk< tt �|||�|� |dk}d||dk< d||< tt �|||�|� d S )NFTr, g @g g @g @g �g �g $�g $@g �?)r) i r i�au) r/ r� rm r r r rI r� rn r� )rJ r rI r^ r� Ztmpmaskr' r' rD �test_dtype_mix� s$ zTestWhere.test_dtype_mixc C s� t �ddddddddddddg �}t jdddddddd d dddg t jd �}t jd d d �}t jdddddddd d dddg t jd �}tt �|||�|� |�d�}tt �|||�|� |�d�}tt �|||�|� |�d �}tt �|||�|� d S )NFTg @g �?g @g @g �g �g $�g $@)r) r, z>i4g z>f8z<i4)r/ r� r r� r r rI )rJ r r� rI r^ r' r' rD � test_foreign� s zTestWhere.test_foreignc C sH ddg}t �d�}t �d�}ttt j|||� ttt j|d ||� d S )NT)r[ rL )rL rL r )r/ r� r r1 r )rJ r rI r^ r' r' rD � test_error s zTestWhere.test_errorc C s| t �d�}t �d�}tt �d||�d� tt �d||�d� t �d�}t �d�}tt �d||�d� tt �d||�d� d S )NrM A� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxTFZabcdZxxxxxxxx)r/ r� r r )rJ rI r^ r' r' rD r� s zTestWhere.test_stringc C sB t �d�}t �t �|dk��}t|t �t jg g gt jd��� d S )N)r, r, g �X@)r) )r/ r2 rK r r � atleast_2dr� r )rJ r� Zibadr' r' rD �test_empty_result s zTestWhere.test_empty_resultc C sX ddddddg}t j�d� t jj|� }x*td�D ] }|�� }|�� }t||� q2W d S )NrG r& r r[ rL r{ )r/ r� r4 r� rp �nonzeror )rJ r? r� rv � benchmarkr r' r' rD � test_largedim' s zTestWhere.test_largedimN) r` ra rb r# r r r r r r� r r r' r' r' rD r� � s ) r� c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )� TestSizeOfc C s t �g �}tt�|�dk� d S )Nr )r/ r� r r� � getsizeof)rJ r� r' r' rD �test_empty_array7 s zTestSizeOf.test_empty_arrayc C s@ |d�j }x0dD ](}tj||d�}tt�|�|| k� qW d S )Nr )rG r� r i� )r) )r9 r/ rH r r� r )rJ r) Z elem_sizerw r� r' r' rD � check_array; s zTestSizeOf.check_arrayc C s | � tj� d S )N)r r/ rl )rJ r' r' rD �test_array_int32B s zTestSizeOf.test_array_int32c C s | � tj� d S )N)r r/ rn )rJ r' r' rD �test_array_int64E s zTestSizeOf.test_array_int64c C s | � tj� d S )N)r r/ r� )rJ r' r' rD �test_array_float32H s z TestSizeOf.test_array_float32c C s | � tj� d S )N)r r/ r )rJ r' r' rD �test_array_float64K s z TestSizeOf.test_array_float64c C s* t �d�}tt�|d �t�|�k � d S )Nr .)r/ r� r r� r )rJ r� r' r' rD � test_viewN s zTestSizeOf.test_viewc C s4 t �d�}tt�|�t�|�ddd��� �k � d S )Nr r, )r/ r� r r� r r} r� )rJ r� r' r' rD r� R s zTestSizeOf.test_reshapec C sP t �d�}t�|�}|�d� t|t�|�k� |�d� t|t�|�k � d S )Nr r� � )r/ r� r� r r� r )rJ r� �oldr' r' rD � test_resizeV s zTestSizeOf.test_resizec C s t �d�}tt|jd� d S )Nr rI )r/ r� r r� � __sizeof__)rJ r� r' r' rD r ^ s zTestSizeOf.test_errorN) r` ra rb r r r r r r r r� r r r' r' r' rD r 5 s r c @ s e Zd Zdd� Zdd� ZdS )� TestHashingc C s t �d�}ttt|� d S )Nr )r/ r� r r� rq )rJ r� r' r' rD �test_arrays_not_hashablee s z$TestHashing.test_arrays_not_hashablec C s t �g �}| �t|tj�� d S )N)r/ r� � assertFalser� � collections�Hashable)rJ r� r' r' rD �test_collections_hashablei s z%TestHashing.test_collections_hashableN)r` ra rb r r" r' r' r' rD r c s r c @ s� e Zd ZeZejejejejej ej ej ej ej ejejejej ejejejejejejgZejd dk r|ejs|e�ej � G dd� de j �Z G dd� de j �Z G dd� de!�Z"d d � Z#d d � Z$d d� Z%dd� Z&dS )�TestArrayPriorityr r c @ s e Zd ZdZdd� ZdS )zTestArrayPriority.Foog Y@c O s t j||��| �S )N)r/ r� ru )r� r� r� r' r' rD r� � s z TestArrayPriority.Foo.__new__N)r` ra rb r� r� r' r' r' rD r� } s r� c @ s e Zd ZdZdd� ZdS )zTestArrayPriority.Barg @Y@c O s t j||��| �S )N)r/ r� ru )r� r� r� r' r' rD r� � s z TestArrayPriority.Bar.__new__N)r` ra rb r� r� r' r' r' rD �Bar� s r$ c @ s� e Zd ZdZdd� Ze ZZe ZZe Z Z e Z Z e Z Ze ZZe ZZe ZZe ZZe ZZe ZZe ZZ e Z Z eZ eZ eZ!eZ"eZ#eZ$dS )zTestArrayPriority.Otherg @�@c C s | � � S )N)r� )rJ r� r' r' rD �_all� s z TestArrayPriority.Other._allN)%r` ra rb r� r% r �__radd__r �__rsub__r r �__pow__�__rpow__�__div__�__rdiv__�__mod__�__rmod__� __truediv__� __rtruediv__� __floordiv__� __rfloordiv__�__and__�__rand__�__xor__�__rxor__�__or__�__ror__� __lshift__� __rlshift__� __rshift__� __rrshift__r� r� r� r� r� r� r' r' r' rD �Other� s* r< c C sh t �ddg�}| �ddg�}xF| jD ]<}t|�}tt|||�| j�|� tt|||�| j�|� q$W d S )Nr, r& )r/ r� r$ � binary_opsr� r r� )rJ rI r^ r� r r' r' rD �test_ndarray_subclass� s z'TestArrayPriority.test_ndarray_subclassc C sb t �ddg�}| �� }xF| jD ]<}t|�}tt|||�| j�|� tt|||�| j�|� q W d S )Nr, r& )r/ r� r< r= r� r r� )rJ rI r^ r� r r' r' rD �test_ndarray_other� s z$TestArrayPriority.test_ndarray_otherc C sh | � ddg�}| �ddg�}xF| jD ]<}t|�}tt|||�| j�|� tt|||�| j�|� q$W d S )Nr, r& )r� r$ r= r� r r� )rJ rI r^ r� r r' r' rD �test_subclass_subclass� s z(TestArrayPriority.test_subclass_subclassc C sb | � ddg�}| �� }xF| jD ]<}t|�}tt|||�| j�|� tt|||�| j�|� q W d S )Nr, r& )r� r< r= r� r r� )rJ rI r^ r� r r' r' rD �test_subclass_other� s z%TestArrayPriority.test_subclass_otherN)'r` ra rb r7 r� rv r� rq r8 rs rr rt �and_�or_rz rw rx �gt�ge�lt�le�ne�eqr= r� r� � py3kwarningr� �divr/ r= r� r$ r0 r< r> r? r@ rA r' r' r' rD r# n s r# c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestBytestringArrayNonzeroc C s | � tjdgtjd�� d S )Nr� )r) )r r/ r� r� )rJ r' r' rD �"test_empty_bstring_array_is_falsey� s z=TestBytestringArrayNonzero.test_empty_bstring_array_is_falseyc C s( t jdgt jd�}d|d< | �|� d S )Nr� )r) z r )r/ r� r� r )rJ rI r' r' rD �'test_whitespace_bstring_array_is_falsey� s zBTestBytestringArrayNonzero.test_whitespace_bstring_array_is_falseyc C s( t jdgt jd�}d|d< | �|� d S )Nr� )r) z r )r/ r� r� r )rJ rI r' r' rD �%test_all_null_bstring_array_is_falsey� s z@TestBytestringArrayNonzero.test_all_null_bstring_array_is_falseyc C s( t jdgt jd�}d|d< | �|� d S )Nr� )r) z r )r/ r� r� r� )rJ rI r' r' rD �(test_null_inside_bstring_array_is_truthy� s zCTestBytestringArrayNonzero.test_null_inside_bstring_array_is_truthyN)r` ra rb rM rN rO rP r' r' r' rD rL � s rL c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestUnicodeArrayNonzeroc C s | � tjdgtjd�� d S )Nr� )r) )r r/ r� r )rJ r' r' rD �"test_empty_ustring_array_is_falsey� s z:TestUnicodeArrayNonzero.test_empty_ustring_array_is_falseyc C s( t jdgt jd�}d|d< | �|� d S )N�eggs)r) z r )r/ r� r r )rJ rI r' r' rD �'test_whitespace_ustring_array_is_falsey� s z?TestUnicodeArrayNonzero.test_whitespace_ustring_array_is_falseyc C s( t jdgt jd�}d|d< | �|� d S )NrS )r) z r )r/ r� r r )rJ rI r' r' rD �%test_all_null_ustring_array_is_falsey� s z=TestUnicodeArrayNonzero.test_all_null_ustring_array_is_falseyc C s( t jdgt jd�}d|d< | �|� d S )NrS )r) z r )r/ r� r r� )rJ rI r' r' rD �(test_null_inside_ustring_array_is_truthy� s z@TestUnicodeArrayNonzero.test_null_inside_ustring_array_is_truthyN)r` ra rb rR rT rU rV r' r' r' rD rQ � s rQ c @ s e Zd Zdd� Zdd� ZdS )� TestCTypesc C s@ t �dddgdddgg�}| �t|jj� tt|jj�d� d S )Nr, r& r r[ rL r{ )r& r )r/ r� r� r �_ctypesr r� r? )rJ �test_arrr' r' rD �test_ctypes_is_available� s z#TestCTypes.test_ctypes_is_availablec C sb ddl m} d |_zBt�dddgdddgg�}| �|jj|j� tt |jj �d � W d t|_X d S ) Nr )r� r, r& r r[ rL r{ )r& r ) � numpy.corer� r r/ r� �assertIsInstancerX �_missing_ctypesr r� r? )rJ r� rY r' r' rD � test_ctypes_is_not_available� s z'TestCTypes.test_ctypes_is_not_availableN)r` ra rb rZ r^ r' r' r' rD rW � s rW c C s t �d�} tt| jdd� d S )NrL � â)r* )r/ rH r r1 r( )rI r' r' rD �2test_orderconverter_with_nonASCII_unicode_ordering s r` c C s� G dd� dt �} G dd� d| �}G dd� d| �}tjddgd d �}xN||fD ]B}|� }t||kd � t||kd � t||kd � t||kd � qLW d S ) Nc @ s e Zd Zdd� Zdd� ZdS )z*test_equal_override.<locals>.MyAlwaysEqualc S s dS )NrI r' )rJ r� r' r' rD r� s z1test_equal_override.<locals>.MyAlwaysEqual.__eq__c S s dS )NrH r' )rJ r� r' r' rD r� s z1test_equal_override.<locals>.MyAlwaysEqual.__ne__N)r` ra rb r� r� r' r' r' rD � MyAlwaysEqual s ra c @ s e Zd ZdZdS )z-test_equal_override.<locals>.MyAlwaysEqualOldi' N)r` ra rb r� r' r' r' rD �MyAlwaysEqualOld s rb c @ s e Zd ZdZdS )z-test_equal_override.<locals>.MyAlwaysEqualNewN)r` ra rb r� r' r' r' rD �MyAlwaysEqualNew s rc )r r, )r& r zi4,i4)r) rI rH )r0 r/ r� r )ra rb rc r� Zmy_always_equal_clsZmy_always_equalr' r' rD �test_equal_override s rd r� )�� __future__r r r r rW r� r\ r� r7 r^ r r5 r r~ r� � contextlibr r� rt � __builtin__rt r �numpyr/ � numpy.compatr r � test_printr r� r r r r r r r Z numpy.testingr r r r r r r r r r r r r r r r! r" r# r% r$ r� r� rE rF rc rx r� r� r� r� r� r� r5 rg r� r0 rm ro r� r� r� r� r� r� r� r r r r$ r4 r6 rB rI rT r� r� r� r� r� r� r� r� r� r� r� r, r@ rC rK rL r] rb re rh ro rn r r� r� Znumpy.core._internalr� r� r� r� r� r� r� r� r� r� r� r� r� r r r# rL rQ rW r` rd r` r' r' r' rD �<module> s $P$ Nj R` Z r; l Jf B6 J,3/ >/O A P< D,]&g j )"$ A &. V