?????????? ????????? - ??????????????? - /usr/lib64/python3.9/site-packages/numpy/ma/tests/__pycache__/test_mrecords.cpython-39.pyc
???????
a z[yc�M � @ s� d Z ddlZddlmZ ddlmZ ddlmZmZ ddlm Z ddl m Z m Z ddlmZmZm Z mZm Z mZ ddlmZmZmZ dd lmZ G d d � d �ZG d d � d �ZG dd� d�Z dd� Z dS )z^Tests suite for mrecords. :author: Pierre Gerard-Marchant :contact: pierregm_at_uga_dot_edu � N)�recarray)�masked�nomask)�temppath)� fromrecords� fromarrays)� MaskedRecords� mrecarrayr � fromtextfiler �addfield)�assert_� assert_equal�assert_equal_records)�picklec @ s� e Zd Zg d�Zg d�Zg d�ZdefdefdgZg d�Z e j e e eee��e ed�Zd d � Zd d � Zd d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Z d%d&� Z d'S )(� TestMRecords�� � � � � �皙�����?皙����@�ffffff @皙����@� @�� one� two� three� four� five�a�b��c�|S8�r r r r r ��mask�dtypec C s� | j }|�t�}t|j|j� t|j|j� tt|j t �� t|j |j �t �� dD ]}t|| || � qTt|�t�|� d S )N�r# r$ r& ) �base�viewr r � recordmaskr �_maskr � isinstance�_datar )�selfr- �mbase�field� r6 �B/usr/lib64/python3.9/site-packages/numpy/ma/tests/test_mrecords.py� test_byview$ s zTestMRecords.test_byviewc C s� | j �� }|�t�}dD ]*}tt||�|| � t|| || � q|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 tu � |d d � }tt|t�� t|j|j� t|j dd g� t|jtjdd g|j jd �� t||d d � �t�� dD ]"}tt||�|d d � | � �q�d S ) Nr, r )r r r )FFFr# ���)NNNT)TTTr r �r+ )r- �copyr. r r �getattrr r1 r+ �tolistr/ r r0 �itemr r r* �np�array)r3 r- r4 r5 Z mbase_firstZ mbase_lastZmbase_slr6 r6 r7 �test_get0 sD ���zTestMRecords.test_getc C s | j �� }|�t�}|�� }d|_d|jjd d �< t|d jg d�� t|d jg d�� d|_t|d jdgd � tt � |d �dgd � t|j dgd � t|j� � t jg d �td �� t|_t|jjdgd � t|jj dgd � tt � |d �dgd � tt �|d �d gd � t|j� � t jg d �td �� |�t��� }d|jdd �< t|jg d�� t|jjg d�� t|jdd �< t|j|d � t|jjg d�� ddtfg}t jg d�|d �}|�t�}t jg d�g d�d�}||d< t|j|� t|jjg d�� d S )N)i?B g@��x �D�N/Ar r# )r r r r r r( r r F)�r r r �r r r rC rC rD r: r& � N/A)�r r r rD rF rF rD r �r r r r r )r r r r r r$ �r r r r r )Zalphaz|S1�num))r# r )r$ r )r& r )� � � �r r r �r* )r- r; r. r � fill_valuer# r2 r r0 �ma� getmaskarrayr/ r= r? r@ �boolr r& r* Zgetdatar$ �intr rI )r3 r- r4 �ndtype�dataZrdata�valr6 r6 r7 �test_set_fieldsX sP �� �� z TestMRecords.test_set_fieldsc C s� | j �� }|�t�}t|d d< t|jg d�� t|jjg d�� tt � d�t j � d�gdt fdtfgd�}t|d d< t|jg d�� t|jjg d �� d S ) Nr# ���r rH r r$ r: )r r r r r )r r r r r )r- r; r. r r r r# r0 r r? �arange�random�randrS �float�r3 r- r4 r6 r6 r7 �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 rC )r- r; r. r r r* r rP rQ r0 r= r? r@ rR r r] r6 r6 r7 � test_set_mask� s � �zTestMRecords.test_set_maskc C s� | j �� }|�t�}g d�|_t|jjg d�� t|jjg d�� t|jjg d�� g d�|_t|jjg d�� t|jjg d�� t|jjg d�� d S )N)r r r r r �r r r r r ) r- r; r. r r* r r# r$ r& r] r6 r6 r7 �test_set_mask_fromarray� s z$TestMRecords.test_set_mask_fromarrayc C s� | j �� �t�}tjg d�dtfdtfdtfgd�}||_t|j jg d�� t|j jg d�� t|j jg d�� d|_||_ t|j jg d�� t|j jg d�� t|j jg d�� d S ) N)�r r r rc �r r r rd rC r# r$ r& r: )r r r r r )r r r r r F) r- r; r. r r? r@ rR r* r r# r$ r& Z fieldmask)r3 r4 Znmaskr6 r6 r7 �test_set_mask_fromfields� s �z%TestMRecords.test_set_mask_fromfieldsc C s^ | j �� }|�t��� }t|d< t|j�� tj g d�t d�� t|j g d�� |�t��� }d|d d�< t|j j g d�� t|j jg d�� t|jj g d �� t|jjg d�� t|jj g d �� t|jjg d�� |�t��� }t|d d�< t|j j g d �� t|j jg d �� t|jj g d �� t|jjg d �� t|jj g d�� t|jjg d �� d S )NrX )rC r_ rC r_ r_ r: rH �r r r r )r r r r r ra )� @rg r r r )� 5rh r r! r" r )r r r r r r r )r- r; r. r r r r0 r= r? r@ rR r/ r# r2 r$ r& r] r6 r6 r7 �test_set_elements� s< �� � �z TestMRecords.test_set_elementsc C s� | j �� }|�t�}|�� z|d|dd �< t|jjg d�� t|jjg d�� t|j jg d�� t|jj g d�� t|jj |jj � t|jj |j j � W n$ t y� Y n t y� � Y n 0 t d��zd|dd �< W n t tfy� Y n 0 td ��d S ) Nrf rX rG )r r r r r )r r r rh r" r( z)Flexible hard masks should be supported !r z.Should have expected a readable buffer object!)r- r; r. r � harden_maskr r# r2 r$ r& r0 �NotImplementedError�AssertionError� Exception� TypeErrorr] r6 r6 r7 �test_setslices_hardmask� s. � z$TestMRecords.test_setslices_hardmaskc C s� | j �� }|�t�}|�� t|j� t|_t |j |j � |� � t|j � t|_t |j t � |j|j�� tt �|d j �tu � t|d j |d j � d S )Nr$ r# )r- r; r. r rj r Z _hardmaskr r* r r0 Z soften_maskrP Zmake_mask_none�shaper+ Z make_maskr r] r6 r6 r7 � test_hardmask s �zTestMRecords.test_hardmaskc C s~ | j �� }|�t�}tdtjd �D ]T}tj||d�}t�|�}t |j |j � t |j |j � t |j |j � t |j |j � q$d S )Nr r )Zprotocol)r- r; r. r �ranger ZHIGHEST_PROTOCOL�dumps�loadsr r+ r r2 r0 )r3 r- �mrec�proto�_Zmrec_r6 r6 r7 � test_pickling s zTestMRecords.test_picklingc C s� t jg d�g d�td�}t jg d�g d�td�}t jg d�g d�dd�}dtfdtfd g}t|||g|d d �}|�� }t|d tjd td �� t|d tjdtd �� t|d tjddd �� d S )N�r r r rF r) �r r r ��one�twoZthreer' r# r$ r% �韆 � �i�@rB �r+ rO )r r r r: )r r r� r& )r| r} rB )rP r@ rS r\ r �filledr r? )r3 �_a�_b�_c�ddtyperu Z mrecfilledr6 r6 r7 � test_filled' s ���zTestMRecords.test_filledc C s� t jg d�g d�td�}t jg d�g d�td�}t jg d�g d�dd�}dtfd tfd g}t|||g|d d �}t|�� g d �� d S )Nry rF r) rz r{ rM r' r# r$ r% r~ r� ))r r N)r r r )NNr )rP r@ rS r\ r r r= )r3 r� r� r� r� ru r6 r6 r7 � test_tolist6 s ��zTestMRecords.test_tolistc C s0 t dtdd�}d|d d< t|d d d� d S )Nr r- )�formats�namesrJ r )r r\ r )r3 �xr6 r6 r7 �test_withnamesC 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)�sr' �fr: r )r � 1g �?)Zf0z<f8)r r )r r zi4, (2,3)float, floatr_ rC ) r rS r\ r r r� r> r? r@ r+ r )r3 ZeasyZsoloZmultr6 r6 r7 �test_exotic_formatsI s" � ��z TestMRecords.test_exotic_formatsN) �__name__� __module__� __qualname__Zilist�flistZslistrS r\ r� r* rP r@ �list�zipr- r8 rA rW r^ r` rb re ri ro rq rx r� r� r� r� r6 r6 r6 r7 r s* (7 r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestViewc C sn t �d�t j�d� }}dtfdtfg}t jtt||��|d�}t||g|dd�}d|j d< ||||f| _ d S ) NrJ r# r$ r: )g "�g �X�r� )FTr ) r? rY rZ r[ r\ r@ r� r� r r* rU )r3 r# r$ rT �arrru r6 r6 r7 � setup_method_ s zTestView.setup_methodc C s@ | j \}}}}|�� }tt|t�� t||� t|j|j� d S )N)rU r. r r1 r r r0 )r3 ru r# r$ r� �testr6 r6 r7 �test_view_by_itselfh s z TestView.test_view_by_itselfc C sd | j \}}}}tdf}|�|�}tt|tj�� t|tj t t ||��td�� t|d tj u � d S )Nr r: )r r ) rU r\ r. r r1 rP Z MaskedArrayr r? r@ r� r� r )r3 ru r# r$ r� Zntyper� r6 r6 r7 �test_view_simple_dtypeo s z TestView.test_view_simple_dtypec C s~ | j \}}}}dtfdtfg}|�|�}tt|t�� t||�|�� t|d d tu � t|j t � |�� t|j d u � d S )N�A�Br ) rU r\ r. r r1 r r r r r+ r? Z _fill_value)r3 ru r# r$ r� Zalttyper� r6 r6 r7 �test_view_flexible_typew s z TestView.test_view_flexible_typeN)r� r� r� r� r� r� r� r6 r6 r6 r7 r� ] s r� c @ s� e Zd Zejg d�g d�ed�Zejg d�g d�ed�Zejg d�g d�dd�Z defdefd gZ e eee ge d d �Z e ejeje jfe d �Ze ee fZd d� Zdd� Zdd� Zdd� Zdd� ZdS )�TestMRecordsImportry rF r) rz )r r r r' r# r$ r% )s 99999s 99999.rE r� r: c C s� t jg d�g d�td�}t jg d�g d�td�}t jg d�g d�dd�}| j\}}}td|||f�D ] \}}tt||�j|j� qdt jg d�g d �t d�} t t | |j d �|d � d S ) Nry rF r) rz r{ r' r, )r r r| rM r: r ) rP r@ rS r\ rU r� r r< r0 �objectr r r+ ) r3 r� r� r� ru �nrecrw r� �lZ_xr6 r6 r7 �test_fromarrays� s z"TestMRecordsImport.test_fromarraysc C s | j \}}}g d�}t|dd�}t|dd�}t||� t|�}t|j|j� |jjD ] }tt||�t|j|�� qTt|� � dd�}t|jdt fdt fdg� t dd �D ] \} } tt|| �t|j| �� q�t|�}t|j|j� t|j|� � � t|j|j� d S ) N))r �abcg ��� @r )r Zxyg ���@r )r � g ����?r zc1, c2, c3, c4)r� zc1,c2,c3�c1�c2)�c3z|S5)r� r� r� r, )rU �recfromrecordsr r r r+ r� r< r2 r= rS r\ r� r� r0 ) r3 ru r� r� Zpalist�paZmpa�_mrecr5 r� �nr6 r6 r7 �test_fromrecords� s" z#TestMRecordsImport.test_fromrecordsc C s� | j \}}}t|�� |g d�d�}t|j|j� t|j�� g d�� t|�� |dd�}t|j|j� t|j�� g d�� t|�� ||jd�}t|j|j� t|j�� |j�� � t|�� ||j�� d�}t|j|j� t|j�� |j�� � d S )Nrc )r+ r* )rC r_ rC T)r_ r_ r_ )rU r r= r r2 r r0 )r3 ru r� r� r� r6 r6 r7 �test_fromrecords_wmask� s �z)TestMRecordsImport.test_fromrecords_wmaskc C s� d}t � �R}t|d��}|�|� W d � n1 s60 Y t|ddd�}W d � n1 sb0 Y tt|t�� t|jg d�� t|j j g d�� t|j g d�� 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)Z delimiterZvarnames)r r r r )r r g �OAg�����|۽) r �open�writer r r1 r r �F�Er0 �C)r3 Zfcontent�pathr� Zmrectxtr6 r6 r7 �test_fromtextfile� s � (,z$TestMRecordsImport.test_fromtextfilec C sP | j \}}}g d�g d� }}t|tj||d��}t|j|� t|jj|� d S )N)�d �� i, rM rN )rU r rP r@ r Zf3r0 )r3 ru r� r� �d�mr6 r6 r7 � test_addfield� s z TestMRecordsImport.test_addfieldN)r� r� r� rP r@ rS r� r\ r� r� r� r ru � recfromarraysr2 r� rU r� r� r� r� r� r6 r6 r6 r7 r� � s � � r� c C s0 t jddgddgdtfdtfgd�} | d d S ) N)r �2)r �4)r r )r r r# r$ r) r )rP Z masked_arrayrS r� )�yr6 r6 r7 �#test_record_array_with_object_field� s �r� ) �__doc__Znumpyr? Znumpy.marP r r r Z numpy.testingr Znumpy.core.recordsr r� r r� Znumpy.ma.mrecordsr r r r Znumpy.ma.testutilsr r r Z numpy.compatr r r� r� r� r6 r6 r6 r7 �<module> s D&c
| ver. 1.6 |
Github
|
.
| PHP 8.2.30 | ??????????? ?????????: 0 |
proxy
|
phpinfo
|
???????????