OwlCyberSecurity - MANAGER
Edit File: test_records.cpython-38.pyc
U �p�]�L � @ s� d dl mZmZmZ d dlZzd dlmZ W n ek rH d dl ZY nX d dl Z d dl m Z d dl Z d dlZd dlmZ d dlmZmZmZmZmZmZ d dlmZ G dd� de�Ze j j edkd d �G d d � d e��Z G d d� de�Z dd� Z dS )� )�division�absolute_import�print_functionN)�path)�Path)�assert_� assert_equal�assert_array_equal�assert_array_almost_equal� assert_raises�temppath)�picklec @ 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'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 )N�� �dbe�333333�?� Zde��������?zcol1,col2,col3��namesr )r r r �col1�i� �col2�U� �S�col3�f) �np�rec� fromrecordsr �item�dtypeZkind�sys� version_info�itemsize��self�r� r* �?/usr/lib/python3/dist-packages/numpy/core/tests/test_records.py�test_fromrecords s �z TestFromrecords.test_fromrecordsc C s0 dt fdt fg}tjjg |d�}t|jd� dS )z0 Verify fromrecords works with a 0-length input �a�b�r# �r N)�floatr r 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 � )� � )r7 r6 )r5 r )r r4 r4 r r6 r7 r5 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��formatsr2 � byteorderr4 )idc s efgidcba�r r �arrayr r" r' r* r* r+ �test_method_array? s z!TestFromrecords.test_method_arrayc C s4 t jjdddddddgdd �}t|d �� d � d S ) N�r4 � r- �r � r. �r �! �c�r5 �, �d�r6 �7 �ex�r7 �B r �� �M �g�u1,f4,a1�r? r4 )r g 6@� brA r' r* r* r+ �test_method_array2C s ��z"TestFromrecords.test_method_array2c C s> t jjdddddddgdd �}t|d d d � d �� d � d S ) NrD rF rH rK rN rQ rS rW rX r4 r )r5 g F@� drA r' r* r* r+ �test_recarray_slicesH 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 )Nr4 r r r5 r- �dd�xyzZ12皙�����?za,b,cr )r r] � @�" )r rB r � fromarraysr r" r- )r( Zx1Zx2Zx3r) r* r* r+ �test_recarray_fromarraysM 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 ) Nr9 zrecarray_from_file.fits�rbi� �f8,i4,a5r r= r> ) r �join�dirname�__file__�open�seekr r �fromfilerB �closer )r( Zdata_dir�filename�fdr: r; r* r* r+ �test_recarray_fromfileV s z&TestFromrecords.test_recarray_fromfilec C s� d}t j|dd�}t j|dd�}t j|dd�}tt|��D ]}ttdd��||< q:t jj|||gdd�}tt|��D ]D}t|j| ttdd��k� t|j | dk� t|j | dk� qtd S ) N� �Or/ �f8r4 zdate,data1,data2r g ) r �zeros�range�len�listr rb r �dateZdata1Zdata2)r( �countr- r. rJ r Zminer* r* r+ �test_recarray_from_obja s z&TestFromrecords.test_recarray_from_objc C s� t jddgddgd�}t j�|�}tt|�t�d�� t �t jddd��}ttt j�|��� d �� t j�t jd d d��}tt|�� d �d � t j�t jd dd��}tt|�� d�d k� d S )N)r4 g�������?)r g�������?)�foo�<i4)�barz<f8r/ zi rec.array([(1, 0.1), (2, 0.2)], dtype=[('foo', '<i4'), ('bar', '<f8')])r5 rr z rec.arrayr �i4,i4z numpy.record����i4z dtype=int32) r rB r r �repr�textwrap�dedent�onesr � startswith�find�r( r- r* r* r+ �test_recarray_repro s � �z"TestFromrecords.test_recarray_reprc C sl t jjddd�}tt|�t�d�� |d }tt|�d� z t jd d� tt|�d � W 5 t jdd� X d S ) N)r4 r` Z2003z<i4,<f8,<M8[Y]r/ zt rec.array((1, 2., '2003'), dtype=[('f0', '<i4'), ('f1', '<f8'), ('f2', '<M8[Y]')])r* z(1, 2., '2003')F)Zlegacyz1.13z#(1, 2.0, datetime.date(2003, 1, 1)))r r rB r r� r� r� Zset_printoptions)r( Zarr_0d�recordr* r* r+ �test_0d_recarray_repr� s z%TestFromrecords.test_0d_recarray_reprc 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�r4 �ABC�r ZDEFrz �r| �S4r/ znumpy.�numpy) r rB r8 r �view�recarrayr# r� �evalr� r �type�ndarray)r( r- Z recordarrZrecarrZ recordviewZ recordarr_rZrecarr_rZ recordview_rr* r* r+ �test_recarray_from_repr� s$ � z'TestFromrecords.test_recarray_from_reprc C s 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 � G dd� dt j �}|� |�}tt|d �|� ddddg}t jjddddgdfddddgdfg|d�}t|j jjt j� t|j jjt 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}|D ]P} t j�| �} | � | jj�p�| jt j�} t| jj| jj� tt| �t| �� �q�d S ) Nr� r� rz r� r/ r4 r r r5 r6 zf4,i4rr )r- r )r. r )rJ r} rJ c @ s e Zd ZdS )z.TestFromrecords.test_recarray_views.<locals>.CN)�__name__� __module__� __qualname__r* r* r* r+ �C� s r� )r- �f4,f4)r. �V8)rJ )Zf4r )rM ��i8r} �r4 r4 s 11111111r} r� )r r r� r� )r rB r8 Zint64r r r# r� r� r� r� r� r- r. ZvoidrJ Zfloat32rM �int32�fieldsr� ) r( r- r. r) �rvr� rJ Z test_dtypeZarrs�arrr Zarr2r* r* r+ �test_recarray_views� s^ �� ��� z#TestFromrecords.test_recarray_viewsc C s~ t jjdddgdd�}t jjdddgdd�}t|j|jk� t|j|jk� tt|��D ] }t|| � � || � � k� qXd S )N)r4 �abcg ��� @r )r Zxyg ���@r4 )r � g ����?r zc1, c2, c3, c4r ) r r rB r! r r# r2 rt ru 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)r4 r� gffffff@)r r^ g������@)r Zwrsr zfield, shape, meanr r_ g������@gffffff @�mean�r4 r �A�Br� r2 r6 �field)r r rB r� r r r� �varr2 r r� � isinstance�collections_abc�Callable)r( r� r* r* r+ � test_recarray_conflict_fields� s �� z-TestFromrecords.test_recarray_conflict_fieldsc C s� t jjddgdtfdtfgd�}t|jddg� t|d jd� t|jdd g� t|d jd � t �dtfdtfg�}t jjddg|d�}t|jddg� t|d jd� t|jdd g� t|d jd � d S ) N)r4 r- )r �bbbr- r. r/ r4 r r r� r~ ) r r r! r8 �objectr r- r. r# )r( r- Zndtyper* 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)�abc r4 )r� r �rz r� r| r/ r r4 F)r rB r8 r� r� r rz 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 )NZS5r r7 )r� �D)r� r� r4 )�abcdeZfgehi)r� )r r r4 )r� Zjklmnr� r| r� r� �baz�quxr/ rz r4 s fgehir� )r r# r rB r8 r r� rz r� r| r� r� r� �dictr� r� r� r� )r( Z qux_fieldsr- r* r* r+ �test_recarray_returntypes s< � ��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 ) NZtestr � �f0�f1)r� z|S4)r� z|Sr/ r5 s test� )r r rb r r# r& )r( Zcolsr �dtr* r* r+ �test_zero_width_strings9 s z'TestFromrecords.test_zero_width_stringsN)r� r� r� r, r3 r< rC rZ r\ rc ro ry r� r� r� r� r� r� r� r� r� r� r* r* r* r+ r s& > r zNo pathlib.Path)�reasonc @ s e Zd Zdd� ZdS )� TestPathUsagec C s� t dd��r}t|�}tj�d� tj�d��d�}d|d< |�d��}|�|� W 5 Q R X tj j j |ddd �}t ||� W 5 Q R X d S ) Nz.bin)�suffix�{ rp re )g �?rp r� r6 �wb)r? r2 ) r r r ZrandomZseedZrandZastyperi ZtofileZcoreZrecordsrk r )r( r r- rn �xr* r* r+ �test_tofile_fromfileK s �z"TestPathUsage.test_tofile_fromfileN)r� r� r� r� r* r* r* r+ r� H 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� Zej�dd d d g�d d � �Zej�dd d d g�d!d"� �Zd#d$� Zd%S )&� TestRecordc C s t jjddgdddgd�| _d S )N)r4 r r )r5 r6 r7 �r r{ �r r{ )r r{ r/ )r r r! r9 �r( r* r* r+ �setupZ s ��zTestRecord.setupc C s4 | j }t|jd d� d|d _t|jd d� d S �Nr r4 �r9 r r r� r* r* r+ �test_assignment1` s zTestRecord.test_assignment1c C s4 | j }t|jd d� d|jd< t|jd d� d S r� r� r� r* r* r+ �test_assignment2f s zTestRecord.test_assignment2c C s | j }dd� }tt||� d S )Nc S s d| d _ d S )Nr4 r )Zcol5)r� r* r* r+ �assign_invalid_columno s zATestRecord.test_invalid_assignment.<locals>.assign_invalid_column)r9 r �AttributeError)r( r- r� r* r* r+ �test_invalid_assignmentl s z"TestRecord.test_invalid_assignmentc C sp t jjddgdgd�}d|j_tt�� ddg|_W 5 Q R X tt��" |jddgf|j j d �� W 5 Q R X d S ) Nr0 )r4 )r r r/ Fr r r ) r r rB �flags� writeabler � ValueErrorr Zsetfieldr# r� r' r* r* r+ �test_nonwriteable_setfieldt s z%TestRecord.test_nonwriteable_setfieldc sb | j ddg }t|jjd� t|jjdddg� | j ddg � t� jjd� tt� fdd �� d S ) Nr r )r r r� r� )r� z|V4)r r c s � j jS )N)r# �descrr* ��yr* r+ �<lambda>� r� z5TestRecord.test_out_of_order_fields.<locals>.<lambda>)r9 r r# r r� r r� )r( r� r* r� r+ �test_out_of_order_fields} s �z#TestRecord.test_out_of_order_fieldsc C s| t jdg fgdt jfdt jdfgd�}tdtjd �D ]@}t|t�tj||d��� t|d t�tj|d |d��� q6d S )Nr4 r- r. r r/ r �Zprotocol) r rB r� rt r �HIGHEST_PROTOCOLr �loads�dumps�r( r- �protor* r* r+ � test_pickle_1� s &�zTestRecord.test_pickle_1c C s\ | j }tdtjd �D ]@}t|t�tj||d��� t|d t�tj|d |d��� qd S )Nr r4 r� r )r9 rt r r� r r� r� r� r* r* r+ � test_pickle_2� s �zTestRecord.test_pickle_2c C sh | j }tdtjd �D ]L}t�tj|d |d��}t|jj� t|jj � t|jj � t|jj � qd S )Nr r4 r r� ) r9 rt r r� r� r� r r� � c_contiguous� f_contiguousr� Zaligned)r( r- r� r� r* r* r+ � test_pickle_3� s zTestRecord.test_pickle_3c C sf t �ddg�}t jd|d��t j�}t �dddg�|_t jdd tfd tfd t fgd�}|d d g d S ) N)rz r� )r| rq r� r/ r4 r r )r r� r� �z) r r# rs r� r� rB rz r� r1 r8 )r( r� r) r� r* r* r+ �test_objview_record� s z TestRecord.test_objview_recordc C s: t jddtdfgd�}d|d _t|d jt �d�� d S )Nr4 r� r6 r/ r )r r� r1 r� r r� )r( r r* r* r+ �test_record_scalar_setitem� 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+ r� � r� z/TestRecord.test_missing_field.<locals>.<lambda>)r rs r8 r �KeyErrorr� r* r� r+ �test_missing_field� s z TestRecord.test_missing_fieldc C s, t �d�t jdddgd�g}t j�|�}d S )Nrp )r- z<u2)r. z<f4r/ )r Zaranger� r rb )r( Zarraysr� r* r* r+ �(test_fromarrays_nested_structured_arrays� s �z3TestRecord.test_fromarrays_nested_structured_arrays�nfieldsr r4 r c C s| t �dt jfdt jfdt jfgd |� �}t �d|��t j�}|jjt jksPt�|jt jks`t�||_|jjt jksxt�d S )Nr- r. rJ r ) r r# �uint8rs r� r� r� r� �AssertionError)r( r� r� r9 r* r* r+ �test_assign_dtype_attribute� s *z&TestRecord.test_assign_dtype_attributec C s� t �dt jfdt jfdt jfgd|� �}t �d|fg�}t �d|��t j�}t|t j�s^t�t|d t j�srt�|d }t|t j�s�t�t|d t j�s�t�dS )z> Test that nested structured types are treated as records too r- r. rJ N�innerr r ) r r# r� rs r� r� r� r� r� )r( r� r� �dt_outerr9 �data0r* r* r+ � test_nested_fields_are_records� s *z)TestRecord.test_nested_fields_are_recordsc C s� t �dt jfdt jfdt jfg�}|ddg }|j|jks>t�t �d|fg�}t �d|��t j�}t|d j|� |d }t|d j|� dS )z) test that trailing padding is preserved r- r. rJ r� r r N) r r# r� r& r� rs r� r� r )r( r� Z dt_padded_endr� r9 r� r* r* r+ �test_nested_dtype_padding� s " z$TestRecord.test_nested_dtype_paddingN)r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �pytest�markZ parametrizer� r� r� r* r* r* r+ r� Y s$ r� c C s� ddddddg} t tj�| �g k� ddddddg}t tj�|�dgk� ddddddddg}t tj�|�ddgk� ddddddddg}t tj�|�ddgk� d S )Nr4 r r r5 r6 r7 )r r r Zfind_duplicate)�l1�l2�l3r* r* r+ �test_find_duplicate� s r )!Z __future__r r r r$ Zcollections.abcr� r� � ImportError� collectionsr� �osr r� r� r Z numpy.compatr Z numpy.testingr r r r r r r r� r r� Zskipifr� r� r r* r* r* r+ �<module> s( 3