Edit File: test_mrecords.cpython-37.pyc
B ��Fd5P � @ s� d Z ddlmZmZmZ ddlZddlZddlZddl m Z ddlm Z ddl m Z m Z ddlmZmZmZ ddlmZmZ ddlmZmZmZmZmZmZ dd l m Z m Z m Z G d d � d e�Z G d d � d e�Z!G dd� de�Z"dd� Z#e$dkr�e� dS )z^Tests suite for mrecords. :author: Pierre Gerard-Marchant :contact: pierregm_at_uga_dot_edu � )�division�absolute_import�print_functionN)�recarray)�masked�nomask)�TestCase�run_module_suite�temppath)� fromrecords� fromarrays)� MaskedRecords� mrecarrayr � fromtextfiler �addfield)�assert_� assert_equal�assert_equal_recordsc @ 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#S )$� TestMRecordsc O s t j| f|�|� | �� d S )N)r �__init__�setup)�self�args�kwds� r ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/ma/tests/test_mrecords.pyr ! s zTestMRecords.__init__c C sl dddddg}dddd d g}d d d ddg}dt fdtfdg}dddddg}tjtt|||��||d�| _d S )N� � � � � g�������?g������@gffffff @g������@g @s ones twos threes fours five�a�b)�cz|S8r )�mask�dtype)�int�float�ma�array�list�zip�base)r �ilist�flist�slist�ddtyper$ r r r r % s zTestMRecords.setupc C s� | j }|�t�}t|j|j� t|j|j� tt|j t �� t|j |j �t �� x dD ]}t|| || � qVW t|�t�|� d S )N)r! r" r# ) r, �viewr r � recordmaskr �_maskr � isinstance�_datar )r r, �mbase�fieldr r r � test_byview/ s zTestMRecords.test_byviewc C s� | j �� }|�t�}x2dD ]*}tt||�|| � t|| || � qW |d }tt|t�� t|j|j� t|� � d� t|j t � t|j � � d� t|d |d d � |d }tt|t�� t|j|j� t|� � d� t|j d� t|j � � d � t|d |d d � t|d tk� |d d � }tt|t�� t|j|j� t|j dd g� t|jtjdd g|j jd �� t||d d � �t�� x*dD ]"}tt||�|d d � | � �q�W d S ) N)r! r" r# r )r g�������?s one)FFFr! ���)NNNT)TTTr r )r% )r, �copyr1 r r �getattrr r4 r% �tolistr2 r r3 �itemr r r$ �npr) )r r, r6 r7 Z mbase_firstZ mbase_lastZmbase_slr r r �test_get; s> zTestMRecords.test_getc C sH | j �� }|�t�}|�� }d|_d|jjd d �< t|d jdddddg� t|d jdddddg� d|_t|d jdgd � tt � |d �dgd � t|j dgd � t|j� � t jdddddgtd �� t|_t|jjdgd � t|jj dgd � tt � |d �dgd � tt �|d �d gd � t|j� � t jd dd d dgtd �� |�t��� }d|jd d �< t|jddd ddg� t|jjdddddg� t|jd d �< t|j|d � t|jjdddddg� ddtfg}t jdddg|d �}|�t�}t jdddgdddgd�}||d< t|j|� t|jjdddg� d S )N)i?B g@��x �DzN/Ar r! r r F)r r r )r r r )r% r# s N/A)r r r r r r" )�alphaz|S1�num)r! r )r" r )r# r � � � )r$ )r, r: r1 r � fill_valuer! r5 r r3 r( � getmaskarrayr2 r< r> r) �boolr r# r$ �getdatar" r& r rA )r r, r6 �ndtype�dataZrdata�valr r r �test_set_fieldsc sX z TestMRecords.test_set_fieldsc C s� | j �� }|�t�}t|d d< t|jdddddg� t|jjdddddg� tt � d�t j � d�gdt fd tfgd �}t|d d< t|jdddddg� t|jjdddddg� d S ) Nr! ���r r r r r r r" )r% )r, r: r1 r r r r! r3 r r> �arange�random�randr& r' )r r, r6 r r r �test_set_fields_mask� s z!TestMRecords.test_set_fields_maskc C s� | j �� }|�t�}t|_tt�|d �dgd � t|d j |d j � t|d j |d j � t|j � � t j dgd t d�� t|_tt�|d �dgd � t|j � � t j d gd t d�� d S ) Nr" r r r! r# )r r r )r% r )r r r )r, r: r1 r r r$ r r( rF r3 r< r> r) rG r )r r, r6 r r r � test_set_mask� s zTestMRecords.test_set_maskc C s� | j �� }|�t�}dddddg|_t|jjdddddg� t|jjdddddg� t|jjdddddg� dddddg|_t|jjdddddg� t|jjdddddg� t|jjdddddg� d S )Nr r ) r, r: r1 r r$ r r! r" r# )r r, r6 r r r �test_set_mask_fromarray� s z$TestMRecords.test_set_mask_fromarrayc C s� | j �� �t�}tjdddddgdtfdtfdtfgd�}||_t|j jddd d dg� t|j jd d dddg� t|j jddd d dg� d |_||_ t|j jddd d dg� t|j jd d dddg� t|j jddd d dg� d S ) N)r r r )r r r )r r r r! r" r# )r% r r F) r, r: r1 r r> r) rG r$ r r! r" r# � fieldmask)r r6 �nmaskr r r �test_set_mask_fromfields� s z%TestMRecords.test_set_mask_fromfieldsc C s� | j �� }|�t��� }t|d< t|j�� tj dddddgt d�� t|j dddddg� |�t��� }d|d d�< t|j j d d d d d g� t|j jdddddg� t|jj d d d ddg� t|jjdddddg� t|jj dddddg� t|jjdddddg� |�t��� }t|d d�< t|j j ddd d d g� t|j jdddddg� t|jj ddd ddg� t|jjdddddg� t|jj dddddg� t|jjdddddg� d S )NrM )r r r )r r r )r% r r )r r r r r r r g @gffffff @g������@g @� 5s threes fours fiveg�������?g������@s ones two)r, r: r1 r r r r3 r< r> r) rG r2 r! r5 r" r# )r r, r6 r r r �test_set_elements� s4 z TestMRecords.test_set_elementsc C s | j �� }|�t�}|�� y�d|dd �< t|jjdddddg� t|jjddd dd g� t|j jd d d ddg� t|jj dddddg� t|jj |jj � t|jj |j j � W n( t k r� Y n t k r� � Y n X t d��yd|dd �< W n t tfk �r Y n X td��d S )N)r r r rM r r r r g�������?g������@gffffff @g @s ones twos threerW s fiver z)Flexible hard masks should be supported !z.Should have expected a readable buffer object!)r, r: r1 r � harden_maskr r! r5 r" r# r3 �NotImplementedError�AssertionError� Exception� TypeError)r r, r6 r r r �test_setslices_hardmask� s, z$TestMRecords.test_setslices_hardmaskc C s� | j �� }|�t�}|�� | �|j� t|_t |j |j � |� � | �|j � t|_t |j t � |j|j�� | �t �|d j �tk� t|d j |d j � d S )Nr" r! )r, r: r1 r rY � assertTrue� _hardmaskr r$ r r3 � soften_maskr( �make_mask_none�shaper% � make_maskr )r r, r6 r r r � test_hardmask s zTestMRecords.test_hardmaskc C sd | j �� }|�t�}t�|�}t�|�}t|j|j� t |j |j � t|j |j � t |j |j � d S )N) r, r: r1 r �pickle�dumps�loadsr r% r r5 r3 )r r, �mrec�_Zmrec_r r r � test_pickling& s zTestMRecords.test_picklingc C s� t jdddgdddgtd�}t jdddgdddgtd�}t jd d d gdddgd d�}d tfdtfdg}t|||g|dd�}|�� }t|d tjdtd�� t|d tjdtd�� t|d tjdd d�� d S )Nr r r r )r$ r% g�������?g������@gffffff @�one�two�threez|S8r! r" )r# z|S8)i�� g �i�@zN/A)r% rE )r r i�� )r% )g�������?g������@g �i�@r# )rl rm zN/A)r( r) r& r' r �filledr r> )r �_a�_b�_cr0 ri Z mrecfilledr r r � test_filled1 s zTestMRecords.test_filledc C s� t jdddgdddgtd�}t jdddgdddgtd�}t jd d d gdddgd d�}d tfdtfdg}t|||g|dd�}t|�� dddg� d S )Nr r r r )r$ r% g�������?g������@gffffff @rl rm rn z|S8r! r" )r# z|S8)i�� g �i�@zN/A)r% rE )r g�������?N)r g������@s two)NNs three)r( r) r& r' r r r< )r rp rq rr r0 ri r r r � test_tolist@ s zTestMRecords.test_tolistc C s0 t dtdd�}d|d d< t|d d d� d S )Nr r, )�formats�namesrB r )r r' r )r �xr r r �test_withnamesM s zTestMRecords.test_withnamesc C s� t ddtfddtfgd�}t|d< t|�d��� d� t ddgd�}t|d< t|�d��� tjd |j d��� � t d d d�}t|d< d |d< |�d� t |�d�tjd d g|j d�� d S )Nr �i)�sz|S8�f)r% r )r � 1g �?)�f0z<f8)r r )r r zi4, (2,3)float, float)r r r )r r r ) r r& r' r r ro r= r> r) r% r )r ZeasyZsolo�multr r r �test_exotic_formatsS s z TestMRecords.test_exotic_formatsN)�__name__� __module__� __qualname__r r r8 r? rL rQ rR rS rV rX r^ re rk rs rt rx r r r r r r s" (7 r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestViewc C sr t �d�t j�d� }}dt jfdt jfg}t jtt||��|d�}t||g|dd�}d|j d< ||||f| _ d S ) NrB r! r" )r% )g "�g �X�)r% rE )FTr ) r> rN rO rP r' r) r* r+ r r$ rJ )r r! r" rI �arrri r r r �setUpi s zTestView.setUpc C sB | j \}}}}|�� }| �t|t�� t||� t|j|j� d S )N)rJ r1 r_ r4 r r r3 )r ri r! r" r� �testr r r �test_view_by_itselfr s z TestView.test_view_by_itselfc C sl | j \}}}}tjdf}|�|�}| �t|tj�� t|tj t t ||��tjd�� | �|d tj k� d S )Nr )r% )r r ) rJ r> r' r1 r_ r4 r( � MaskedArrayr r) r* r+ r )r ri r! r" r� �ntyper� r r r �test_view_simple_dtypey s z TestView.test_view_simple_dtypec C s� | j \}}}}dtjfdtjfg}|�|�}| �t|t�� t||�|�� | �|d d tk� t |j t� |�� | �|j d k� d S )N�A�Br ) rJ r> r' r1 r_ r4 r r r r r% � _fill_value)r ri r! r" r� Zalttyper� r r r �test_view_flexible_type� s z TestView.test_view_flexible_typeN)r� r� r� r� r� r� r� r r r r r� g s r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�TestMRecordsImportc O s t j| f|�|� | �� d S )N)r r r )r r r r r r r � s zTestMRecordsImport.__init__c C s� t jdddgdddgtd�}t jdddgdddgtd�}t jd d d gdddgd d�}d tfdtfdg}t|||g|dd�}t|j|j|jf|d�}|||f| _d S )Nr r r r )r$ r% g�������?g������@gffffff @s ones twos threez|S8r! r" )r# z|S8)s 99999s 99999.s N/A)r% rE )r% )r( r) r& r' r � recfromarraysr5 rJ )r rp rq rr r0 ri �nrecr r r r � s zTestMRecordsImport.setupc C s� t jdddgdddgtd�}t jdddgdddgtd�}t jd d d gdddgd d�}| j\}}}x0td |||f�D ] \}}tt||�j|j� qrW t jddd gdddgd�} t t | |j d�|d � d S )Nr r r r )r$ r% g�������?g������@gffffff @rl rm rn z|S8)r! r" r# )r$ )r% ) r( r) r& r' rJ r+ r r; r3 r r r% ) r rp rq rr ri r� rj r{ �l�_xr r r �test_fromarrays� s z"TestMRecordsImport.test_fromarraysc C s | j \}}}dddg}t|dd�}t|dd�}t||� t|�}t|j|j� x(|jjD ] }tt||�t|j|�� qXW t|� � dd�}t|jdt fdt fd g� x.t d d �D ] \} } tt|| �t|j| �� q�W t|�}t|j|j� t|j|� � � t|j|j� d S ) N)r �abcg ��� @r )r �xyg ���@r )r � g ����?r zc1, c2, c3, c4)rv zc1,c2,c3�c1�c2)�c3z|S5)r� r� r� )r! r" r# )rJ �recfromrecordsr r r r% rv r; r5 r< r& r' r+ ro r3 ) r ri r� r0 Zpalist�paZmpa�_mrecr7 r{ �nr r r �test_fromrecords� s& z#TestMRecordsImport.test_fromrecordsc C s� | j \}}}t|�� |dddgd�}t|j|j� t|j�� dddg� t|�� |dd�}t|j|j� t|j�� dddg� t|�� ||jd�}t|j|j� t|j�� |j�� � t|�� ||j�� d�}t|j|j� t|j�� |j�� � d S )Nr r )r% r$ )r r r )r r r T)rJ r r< r r5 r r3 )r ri r� r0 r� r r r �test_fromrecords_wmask� s z)TestMRecordsImport.test_fromrecords_wmaskc C s� d}t � �4}t|d��}|�|� W d Q R X t|ddd�}W d Q R X | �t|t�� t|jddddg� t|j j ddddg� t|j dddd g� d S ) Nz�# 'One (S)','Two (I)','Three (F)','Four (M)','Five (-)','Six (C)' 'strings',1,1.0,'mixed column',,1 'with embedded "double quotes"',2,2.0,1.0,,1 'strings',3,3.0E5,3,,1 'strings',4,-1e-10,,,1 �w�,ZABCDEFG)� delimitor�varnamesr r g �OAg�����|۽) r �open�writer r_ r4 r r �F�Er3 �C)r Zfcontent�pathr{ Zmrectxtr r r �test_fromtextfile� s z$TestMRecordsImport.test_fromtextfilec C sT | j \}}}dddgdddg }}t|tj||d��}t|j|� t|jj|� d S )N�d �� i, r r )r$ )rJ r r( r) r �f3r3 )r ri r� r0 �d�mr r r � test_addfield� s z TestMRecordsImport.test_addfieldN) r� r� r� r r r� r� r� r� r� r r r r r� � s r� c C s2 t jddgddgdtfdtjfgd�} | d d S ) N)r �2)r �4)r r )r r r! r" )r$ r% r )r( � masked_arrayr& r> �object)�yr r r �#test_record_array_with_object_field� s r� �__main__)%�__doc__� __future__r r r �warningsrf �numpyr> �numpy.mar( r r r Z numpy.testingr r r Znumpy.core.recordsr r� r r� �numpy.ma.mrecordsr r r r Znumpy.ma.testutilsr r r r r� r� r� r� r r r r �<module> s& J&i