Edit File: test_records.cpython-37.pyc
B ��Fd= � @ s� d dl mZmZmZ d dlZd dlZd dlZd dlZd dlm Z d dl Z d dl m Z mZmZmZmZmZmZmZ G dd� de �ZG dd� de �Zd d � Zed kr�e� dS ) � )�division�absolute_import�print_functionN)�path)�TestCase�run_module_suite�assert_� assert_equal�assert_array_equal�assert_array_almost_equal� assert_raises� assert_warnsc @ 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%S )&�TestFromrecordsc C s� t jjdddgdddggdd�}t|d �� d � t|d jjd � tjd d krxt|d jjd� t|d jj d� n$t|d jjd� t|d jj d � t|d jjd� d S )Ni� �dbeg333333�?� �deg�������?zcol1,col2,col3)�namesr )i� r g333333�?�col1�i� �col2�U� �S�col3�f) �np�rec� fromrecordsr �item�dtype�kind�sys� version_info�itemsize)�self�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_records.py�test_fromrecords s z TestFromrecords.test_fromrecordsc C s4 dt jfdt jfg}t jjg |d�}t|jd� dS )z0 Verify fromrecords works with a 0-length input �a�b)r )r N)r �floatr r r �shape)r% r r& r' r' r( �test_fromrecords_0len s z%TestFromrecords.test_fromrecords_0lenc C s� dddgdddgg}ddd gd d d gg}d d d gd ddgg}t jj|d tfdtfgd�}t|d |� t|d |� t jj|d dgd�}t|d |� t|d |� t||� d S )N)� r )r � )� � )r2 r1 )r0 r )r r/ r/ r r1 r2 r0 r r* r+ )r )r )r r r �intr )r% �dataZ expected_aZ expected_b�r1�r2r' r' r( �test_fromrecords_2d$ s z#TestFromrecords.test_fromrecords_2dc C s* t jjddddd�}t|d �� d� d S )Ns� abcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgzi2,a3,i4r �big)�formatsr- � byteorderr/ )idc s efgidcba)r r �arrayr r )r% r& r' r' r( �test_method_array8 s z!TestFromrecords.test_method_arrayc C s4 t jjdddddddgdd �}t|d �� d � d S ) N)r/ � r* )r � r+ )r �! �c)r0 �, �d)r1 �7 �ex)r2 �B r )� �M �gzu1,f4,a1)r9 r/ )r g 6@� b)r r r; r r )r% r& r' r' r( �test_method_array2< s z"TestFromrecords.test_method_array2c C s> t jjdddddddgdd �}t|d d d � d �� d � d S ) N)r/ r= r* )r r> r+ )r r? r@ )r0 rA rB )r1 rC rD )r2 rE r )rF rG rH zu1,f4,a1)r9 r/ r )r0 g F@� d)r r r; r r )r% r& r' r' r( �test_recarray_slicesA s z$TestFromrecords.test_recarray_slicesc C s� t �ddddg�}t �ddddg�}t �d dddg�}t jj|||gd d �}t|d �� d � d |d< t|jt �ddddg�� d S )Nr/ r r r0 r* �dd�xyz�12g�������?za,b,c)r )r rM g @�" )r r; r � fromarraysr r r* )r% �x1�x2�x3r& r' r' r( �test_recarray_fromarraysF s z(TestFromrecords.test_recarray_fromarraysc C sz t �t �t�d�}t �|d�}t|d�}|�d� tjj|dddd�}|�d� tjj |dddd�}|� � t ||� d S ) Nr4 zrecarray_from_file.fits�rbi� zf8,i4,a5r r8 )r9 r- r: ) r �join�dirname�__file__�open�seekr r �fromfiler; �closer )r% Zdata_dir�filename�fdr5 r6 r' r' r( �test_recarray_fromfileO s z&TestFromrecords.test_recarray_fromfilec C s� d}t j|dd�}t j|dd�}t j|dd�}x&tt|��D ]}ttdd��||< q<W t jj|||gdd�}xTtt|��D ]D}t|j| ttdd��k� t|j | dk� t|j | dk� qzW d S ) N� �O)r �f8r/ zdate,data1,data2)r g ) r �zeros�range�len�listr rQ r �dateZdata1Zdata2)r% �countr* r+ r@ r Zminer' r' r( �test_recarray_from_objZ s z&TestFromrecords.test_recarray_from_objc C s t jddgdtfdgd�}t j�|�}|�t j�}|�t �t j|jf��}tdt |� dt i�}tdt |� dt i�}tdt |� dt i�}t t |�t j� t |jj t j� t ||� t t |�t j� t |jj t j� t ||� t t |�t j � t |jj t j� t ||� d S )N)r/ �ABC)r �DEF�foo)�bar�S4)r znumpy.�numpy) r r; r3 r �view�recarrayr �record�eval�reprr �type�ndarray)r% r* Z recordarrZrecarrZ recordviewZ recordarr_rZrecarr_rZ recordview_rr' r' r( �test_recarray_from_reprh s" z'TestFromrecords.test_recarray_from_reprc C sX t jddgdtfdgd�}t jdddd d gt jd�}tt j�|�jjt j� ttt j�|��t j � tt j�|�jjt j� ttt j�|��t j � t|� t j �jjt j� tt|� t j ��t j � t|� t j �jjt j� tt|� t j ��t j � t j�t j d d d��}|� d �� d �}tt|�t j � t|jjt j� t j�t j d d ddgd��}t|d jjt j� tt|d �t j � t t ��6 t|ddg jjt j� tt|ddg �t j � W d Q R X G dd� dt j �}|� |�}tt|d �|� ddddg}t jjddddgdfddddgdfg|d�}t|jjjt j� t|jjjt j� t|jjjt j� t|jjjt j� t j�t j d dd��}t|� d �jjt j� t|� d �jjt j� t|� d �jjt j� t|� d �jjt j� t j d d d�t j d d d�g}xX|D ]P} t j�| �} | � | jj�p"| jt j�} t| jj| jj� tt| �t| �� �q W d S ) N)r/ rk )r rl rm )rn ro )r r/ r r r0 r1 zf4,i4rc )r* �i4)r+ ry )r@ zi4,i4r@ r* r+ c @ s e Zd ZdS )z.TestFromrecords.test_recarray_views.<locals>.CN)�__name__� __module__� __qualname__r' r' r' r( �C� s r} )r* zf4,f4)r+ �V8)r@ )�f4r )rB )�i8zi4,i4)r/ r/ s 11111111zi4,i4zf4,f4)ry r r~ )r� zi4,i4)r r; r3 �int64r r r rv rs rr rq �onesr � FutureWarningr* r+ �voidr@ �float32rB �int32�fieldsrw ) r% r* r+ r& �rvr} r@ � test_dtype�arrs�arrr �arr2r' r' r( �test_recarray_views sX " z#TestFromrecords.test_recarray_viewsc C s� t �t jddd��}ttt j�|���d�� t j�t jddd��}tt|��d�d� t j�t jdd d��}tt|��d �dk� d S ) Nr0 rc )r z rec.arrayr zi4,i4z numpy.record���ry z dtype=int32) r r; r� r ru r � startswithr �find)r% r* r' r' r( �test_recarray_repr� s z"TestFromrecords.test_recarray_reprc C s� t jjdddgdd�}t jjdddgdd�}t|j|jk� t|j|jk� x0tt|��D ] }t|| � � || � � k� qZW d S )N)r/ �abcg ��� @r )r �xyg ���@r/ )r � g ����?r zc1, c2, c3, c4)r ) r r r; r r r r- re rf r )r% �ra�pa�kr' r' r( �test_recarray_from_names� s z(TestFromrecords.test_recarray_from_namesc C s� t jjdddgdd�}dddg|_t|d dddg� tt|j�t|j�k� d |_t|jd k� d d d g|_t |d d d d gg� d|_ t |d dddgg� tt |j t j �� d S )N)r/ r� gffffff@)r rN g������@)r Zwrsg�������?zfield, shape, mean)r g�������?g������@gffffff @�mean)r/ r �A�Br} r- r1 �field)r r r; r� r r rv �varr- r r� � isinstance� collections�Callable)r% r� r' r' r( � test_recarray_conflict_fields� s z-TestFromrecords.test_recarray_conflict_fieldsc C s� t jjddgdtfdt jfgd�}t|jddg� t|d jd� t|jdd g� t|d jd � t �dtfdt jfg�}t jjddg|d�}t|jddg� t|d jd� t|jdd g� t|d jd � d S ) N)r/ r* )r �bbbr* r+ )r r/ r r r� r� ) r r r r3 �objectr r* r+ r )r% r* �ndtyper' r' r( �$test_fromrecords_with_explicit_dtype� s z4TestFromrecords.test_fromrecords_with_explicit_dtypec C sD t jddgddtfgd�}|�t j�}t|jd |jd kd� d S ) N)zabc r/ )r� r )rm ro rn )r r r/ F)r r; r3 rq rr r rm )r% r* r' r' r( �test_recarray_stringtypes� s z)TestFromrecords.test_recarray_stringtypesc C s� t �d�dft �d�dfd�}t jjddgddd tfd tfgfd tfd |fgd �}tt|j�t j� tt|d �t j� tt|j �t j � tt|d �t j � t|j jjt j � tt|d �t j � t|j jjt j � tt |j jj�|� tt|j�t j� tt|d �t j� tt|d j �t j � tt|d d �t j � t|d j jd� t|d j d d� t|d d jd� t|d d d d� t|d j jd� t|d j d d� t|d d jd� t|d d d d� d S )N�S5r r2 )r} �D)zabc )r/ r/ r/ )�abcdeZfgehi)r� )r r r/ )r� Zjklmn)rm ro rn r� r� �baz�qux)r rm r/ s fgehir� )r r r r; r3 r rv rm rw rn rr rs r� �dictr� r� r� r� )r% Z qux_fieldsr* r' r' r( �test_recarray_returntypes s4 z)TestFromrecords.test_recarray_returntypesc C s� dgd dgd g}t j�|�}t|d dddg� t|d dddg� t �ddg�}t jj||d�}t|jd � t|d d d d g� t|d d d d g� d S ) N�testr � �f0�f1)r� z|S4)r� z|S)r r0 s test� )r r rQ r r r$ )r% �colsr �dtr' r' r( �test_zero_width_strings s z'TestFromrecords.test_zero_width_stringsN)rz r{ r| r) r. r7 r< rJ rL rU r` rj rx r� r� r� r� r� r� r� r� r' r' r' r( r s$ C r c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� ZdS )� TestRecordc C s t jjddgdddgd�| _d S )N)r/ r r )r0 r1 r2 )r z<i4)r z<i4)r z<i4)r )r r r r4 )r% r' r' r( �setUp. s zTestRecord.setUpc C s4 | j }t|jd d� d|d _t|jd d� d S )Nr r/ )r4 r r )r% r* r' r' r( �test_assignment14 s zTestRecord.test_assignment1c C s4 | j }t|jd d� d|jd< t|jd d� d S )Nr r/ )r4 r r )r% r* r' r' r( �test_assignment2: s zTestRecord.test_assignment2c C s | j }dd� }| �t||� d S )Nc S s d| d _ d S )Nr/ r )Zcol5)�xr' r' r( �assign_invalid_columnC s zATestRecord.test_invalid_assignment.<locals>.assign_invalid_column)r4 � assertRaises�AttributeError)r% r* r� r' r' r( �test_invalid_assignment@ s z"TestRecord.test_invalid_assignmentc C sp t jjddgdgd�}d|j_tt�� ddg|_W d Q R X tt��" |jddgf|j j d �� W d Q R X d S ) N)r )r/ )r ry )r Fr r r ) r r r; �flags� writeabler � ValueErrorr �setfieldr r� )r% r& r' r' r( �test_nonwriteable_setfieldH s z%TestRecord.test_nonwriteable_setfieldc C sN t t��" | jddg }| jddg }W dQ R X t|d d |d d � dS )z Ticket #1431.r r Nr r/ )r r� r4 r )r% r� �yr' r' r( �test_out_of_order_fieldsQ s z#TestRecord.test_out_of_order_fieldsc C s^ t jdg fgdt jfdt jdfgd�}t|t�t�|��� t|d t�t�|d ��� d S )Nr/ r* r+ r )r )r r; r� r �pickle�loads�dumps)r% r* r' r' r( � test_pickle_1Z s &zTestRecord.test_pickle_1c C s> | j }t|t�t�|��� t|d t�t�|d ��� d S )Nr )r4 r r� r� r� )r% r* r' r' r( � test_pickle_2` s zTestRecord.test_pickle_2c C sN | j }t�t�|d ��}t|jj� t|jj� t|jj� t|jj � d S )Nr ) r4 r� r� r� r r� � c_contiguous� f_contiguousr� �aligned)r% r* r� r' r' r( � test_pickle_3e s zTestRecord.test_pickle_3c C sz t �ddg�}t jd|d��t j�}t �dddg�|_t jdd tfd tfd t fgd�}t t �� |d d g W d Q R X d S ) N)rm r� )rn rb )r/ r )r r/ r r )r r� r� �z) r r rd rq rr r; rm r� r, r3 r r� )r% r� r& r� r' r' r( �test_objview_recordn s z TestRecord.test_objview_recordc C s: t jddtdfgd�}d|d _t|d jt �d�� d S )Nr/ r� r1 )r r )r rr r, r� r r� )r% r r' r' r( �test_record_scalar_setitemy s z%TestRecord.test_record_scalar_setitemc s0 t jddtfdtfgd�� tt� fdd�� d S )N)r r� r� )r c s � dg S )NZnofieldr' r' )r� r' r( �<lambda>� r� z/TestRecord.test_missing_field.<locals>.<lambda>)r rd r3 r r� )r% r' )r� r( �test_missing_field s z TestRecord.test_missing_fieldN)rz r{ r| r� r� r� r� r� r� r� r� r� r� r� r� r' r' r' r( r� - s r� c C s� ddddddg} t tj�| �g k� ddddddg}t tj�|�dgk� ddddddddg}t tj�|�ddgk� ddddddddg}t tj�|�ddgk� d S )Nr/ r r r0 r1 r2 )r r r �find_duplicate)�l1�l2�l3r' r' r( �test_find_duplicate� s r� �__main__)� __future__r r r r" r� r� �warnings�osr rp r Z numpy.testingr r r r r r r r r r� r� rz r' r' r' r( �<module> s ( W