Current File : //lib64/python3.9/site-packages/numpy/lib/tests/__pycache__/test_recfunctions.cpython-39.opt-1.pyc
a
z[yc�� � @ s. d dl Z d dlZd dlmZ d dlmZ d dlmZ d dl m
Z
m Z d dl m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZ ej j j Z ej j j Z ej j j!Z!ej j j"Z#ej j j$Z%G dd� d�Z&G dd � d �Z'G d
d � d �Z(G d d
� d
�Z)G dd� d�Z*G dd� d�Z+G dd� d�Z,G dd� d�Z-dS )� N)�
MaskedRecords)� assert_equal)�assert_�
assert_raises)� drop_fields�
rename_fields�get_fieldstructure�recursive_fill_fields�find_duplicates� merge_arrays�
append_fields� stack_arrays�join_by�
repack_fields�unstructured_to_structured�structured_to_unstructured�apply_along_fields�require_fields�assign_fields_by_namec @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�TestRecFunctionsc C sr t �ddg�}t �g d��}t jddgddtfgd�}t jd d
gd tfd d
tfdtfgfgd�}||||f| _d S �N� � ��
� � ��A� �?��B� @�r z|S3r! ��dtype�r �r � @�� �� � @�a�b�ba�bb��np�array�float�int�data��self�x�y�z�w� r> �G/usr/lib64/python3.9/site-packages/numpy/lib/tests/test_recfunctions.py� setup_method s
�
�z TestRecFunctions.setup_methodc C sH | j \}}}}t||fdd�}t|t�dtfdtfg�� t||fdd�}t|t�dtfdtfg�� t||fdd�}t|t�dtfddtfg�� t||fdd�}t|t�dtfdddtfgfg�� t||fdd�}t|t�dtfdtfdtfd tfg�� t||fdd�}t|t�dtfddtfd
dtfd tfgfgfg�� d S ) NT�Zflatten� Fr# r! r. r0 r1 r/ )r7 � zip_descrr r3 r% r6 r5 )r9 r= r: r; r< �testr> r> r? �test_zip_descr sD ���
��
��
���z TestRecFunctions.test_zip_descrc C s& t jddgdtfddtfdtfgfgd�}t|d�}t jdd gddtfdtfgfgd�}t||� t|d�}t jd
d gdtfgd�}t||� t|dg�}t jd d
gdtfddtfgfgd�}t||� t|ddg�}t jd
d gdtfgd�}t||� t|ddg�}t jddgg d�}t||� d S )Nr& r) r. r/ r0 r1 r$ )r' )r+ �r �r* )r )r( )r* )r- r> )r3 r4 r6 r5 r r �r9 r. rD �controlr> r> r? �test_drop_fieldsA s,
�
�
�
z!TestRecFunctions.test_drop_fieldsc C s� t jddddgffddddgffgd tfd
d tfd tdffgfgd
�}t|ddd��}dtfd
d tfdtdffgfg}|�|�}t|j|� t||� d S )Nr r r( � >@r* r, r- g N@r. r/ r0 r1 r$ r �BB)r. r1 )r3 r4 r6 r5 r �viewr r% )r9 r. rD ZnewdtyperI r> r> r? �test_rename_fieldsa s "��
z#TestRecFunctions.test_rename_fieldsc C s� t �ddtfg�}t|�}t|d� t �dtfddtfdtfgfg�}t|�}t|d� t �dtfdg fg�}t|�}t|d � t �g �}t|�}t|d
� d S ) Nr# r! �r r! r. r/ r0 r1 )r. )r/ )r0 r1 )r. )r/ r> r> )r3 r% r5 � get_namesr r6 �r9 �ndtyperD r> r> r? �test_get_namesl s
"
z TestRecFunctions.test_get_namesc C s� t �ddtfg�}t|�}t|d� t �dtfddtfdtfgfg�}t|�}t|d� t �dtfdg fg�}t|�}t|d � t �g �}t|�}t|d
� d S ) Nr# r! rO r. r/ r0 r1 )r. r/ r0 r1 �r. r/ r> )r3 r% r5 �get_names_flatr r6 rQ r> r> r? �test_get_names_flat~ s
"
z$TestRecFunctions.test_get_names_flatc C s� t �ddtfg�}t|�}t|g g d�� t �dtfddtfdgfg�}t|�}t|g g dgdgd�� t �dtfddtfdd tfd
tfgfgfg�}t|�}g g dgdgddgddgd �}t||� t �g �}t|�}t|i � d S ) Nr# r! rO r �BA�rL z|S1)r r! rW rL rL �BBA�BBB)r r! rW rL rY rZ )r3 r% r5 r r r6 )r9 rR rD rI r> r> r? �test_get_fieldstructure� s&
�� �
z(TestRecFunctions.test_get_fieldstructurec C sf t jg d�g d�dtfddtfdgfgd�}t|dd d
�}d d g}tt|d
�|� t|d ||d
� t|dd d�}g d�}tt|d
�|� t|d ||d
� t|dd d�}g d�}tt|d
�|� t|d ||d
� t|dd d�}g d�}tt|d
�|� t|d ||d
� t|dd d�}g d�}tt|d
�|� t|d ||d
� d S )N)�r �r" r! )r r] r\ )r )r r! r\ )r )r" �C)�r �r r r_ r_ r_ )r r` )r �r r r r! rW rX ��maskr% FT�Z
ignoremask� return_indexr r ���)�keyre )r r r � r, )r r r r* rL )r r r rh r* )�mar4 r6 r5 r
r �sortedrH r> r> r? �test_find_duplicates� s0
�z%TestRecFunctions.test_find_duplicatesc C s� dt fg}tjg d�g d�d��|�}t|ddd�}g d�}tt|d �|� t|d ||d � t|d
dd�}g d �}tt|d �|� t|d ||d � d S ) Nr. )r r r r r rh rh )r r r r r r r �rc Trd )r r rh r* rf r F)r r r rh r* � )r6 ri r4 rM r
r rj )r9 rR r. rD rI r> r> r? � test_find_duplicates_ignoremask� s
��z0TestRecFunctions.test_find_duplicates_ignoremaskc C s~ t jddd�}t jd|d�}tt|�t �d�� tt|�jd� ttt|�dd�|� t �t j|f�}tt|�jt ju � d S )Nzu1,f4,i8T�Zalignr r$ �
) r3 r% �zerosr r �itemsize�recordr �type)r9 �dtr. r> r> r? �test_repack_fields� s z#TestRecFunctions.test_repack_fieldsc C s� t jdg d�d�}t|�}t|t jddd�� t jg d�g d�d�}t jt|dd g �d
d �}t|t �g d ��� t jt|dg �d
d �}t|t �g d
��� t �d��d�}t||j �}t jddddgfddddgfddddgfdd d d gfgd d d!d"gfd#gd�}t||� t jg d�g d�d�}tt
t j|�t �g d$��� tt
t j|dd g �t �g d ��� t jg d�g d%�d�}t|�}t||j �}t |j |u � t |j |u � t jd&d'd(gddgd)d*ggfd+d,dgd-d.gd/d0ggfgg d1�d�}t|�}t||j �}t |j |u � t |j |u � t � dt
fd2t
fg�} t � d3| fd | fd4| fg�}
t �d|
�} t| t
d�} t| t jd5t
d�� d6d7� }
d8d9� }dAd:d;�}||
|t jt j�d(��}t||�d<t j|f� ||
|
t jd'�d'��}t||�d=t j|f� |t j�}t||�d>t j|f� |t j|
|
t jd'�d'��}t||�d?t j|f� |� }tttt �d(|�� tttt �d(|�t jd� tttt jd@t jd�� d S )BNr* )�r. �i4)r/ zf4,u2)�c�f4r r$ )r* r, �f8))r r r, )r* r, � )r| � � )r r~ � )�r: rx )r; rz )r< r{ r: r< rf )Zaxis)r( g @� "@� &@)r � @� @� $@r r )r r r( r� r, )r- r| � @r� r )r� r g *@g ,@� )g 0@� g 2@g 3@rw r/ )�f0rz )�f1Zu2)ry rz �r )gUUUUUU@gUUUUUU@gUUUUUU!@r� )r� )r; rx )r< rx r r rh rm r| r} � r~ r rp � ))Zx0rx )Zx1)rx r )Zx2)rx �r r r; r. ry )r rm c S s t �| |f�S �N)r3 r% )ru �shaper> r> r? �subarray s zBTestRecFunctions.test_structured_to_unstructured.<locals>.subarrayc W s t �dd� t| �D ��S )Nc S s g | ]\}}d � |�|f�qS )zx{})�format)�.0�iru r> r> r? �
<listcomp>$ � zXTestRecFunctions.test_structured_to_unstructured.<locals>.structured.<locals>.<listcomp>)r3 r% � enumerate)Zdtsr> r> r? �
structured# s zDTestRecFunctions.test_structured_to_unstructured.<locals>.structuredc S s2 t �d| �}t||d�}t|| �}|j|j|jfS )Nr> r$ )r3 rq r r r� r% )ru r% �arr�retZbackarrr> r> r? �inspect& s
zATestRecFunctions.test_structured_to_unstructured.<locals>.inspect)rm rG rF )r, )rh r )N)r3 rq r r r4 Zmean�arangeZreshaper r% r r �baser6 Zint32r �
ValueError�NotImplementedError)r9 r. �outr/ ry Zwant�d�ddZdddZpointZtriangler� �resr� r� r� ru r> r> r? � test_structured_to_unstructured� s�
� �
��
� � �
� ��
��z0TestRecFunctions.test_structured_to_unstructuredc C s\ t jdg d�d�}ddg}tt||�t �d|�� t jddg|d�}t||dd � t|t jd
d g|jd�� t||� t|t jd d
g|jd�� t jddddgfgd�}ddgfg}tt||�t �d|�� t jddg|d�}t||dd � t|t jddg|jd�� t||� t|t jddg|jd�� t �d�t �d� }}t||� t|d d� d S )Nr )rw �r/ r{ �ry Zu1r$ �r/ rz r� �r r )rh r* F)Zzero_unassigned)r r r )r rh r* )r r r )r rh r* r. r� )r� )�rh )r� )�r rh ))r r ))r rh rh r r> )r3 �onesr r r4 r r% )r9 r. Znewdtr/ r> r> r? � test_field_assignment_by_nameA s&
z.TestRecFunctions.test_field_assignment_by_nameN)�__name__�
__module__� __qualname__r@ rE rJ rN rS rV r[ rk rn rv r� r� r> r> r> r? r s ! ! Zr c @ s e Zd Zdd� Zdd� ZdS )�TestRecursiveFillFieldsc C sd t jddgdtfdtfgd�}t jd|jd�}t||�}t jg d�dtfdtfgd�}t||� d S )N�r r� �r � 4@r r! r$ r� �r� r� )r � )r3 r4 r6 r5 rq r% r r �r9 r. r/ rD rI r> r> r? �test_simple_flexible_ s
�z,TestRecursiveFillFields.test_simple_flexiblec C sp t jddgddgdtfdtfgd�}t jd|jd �}t||�}t jg d
�g d �dtfdtfgd�}t||� d S ) Nr� r� �r r ra r r! rb r� r$ r� )r� ra r` )ri r4 r6 r5 rq r% r r r� r> r> r? �test_masked_flexibleh s �
�z,TestRecursiveFillFields.test_masked_flexibleN)r� r� r� r� r� r> r> r> r? 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 )�TestMergeArraysc C sx t �ddg�}t �g d��}t jddgddtfgd�}t jd d
gd tfd d
tfdtfdg fgfgd�}||||f| _d S )Nr r r r r r# r! r$ �r )r r( r> �r* )r, r- r> r. r/ r0 r1 �bcr2 r8 r> r> r? r@ w s � �z TestMergeArrays.setup_methodc C s| | j \}}}}t|�}tjddgdtfgd�}t||� t|f�}t||� t|dd�}t||� t|dd�}t||� d S )NrF r� r� r$ FrA T)r7 r r3 r4 r6 r �r9 �_r: r< rD rI r> r> r? � test_solo� s
zTestMergeArrays.test_soloc C s^ | j d }t|dd�}t||� t|dd�}tjddgdtfdtfd tfgd
�}t||� d S ) Nr FrA T)r r r( )r* r, r- r. r0 r1 r$ )r7 r r r3 r4 r6 r5 �r9 r= rD rI r> r> r? �test_solo_w_flatten� s
�z#TestMergeArrays.test_solo_w_flattenc C s� | j \}}}}t||fdd�}tjg d�dtfdtfgd�}t||� t||fdd�}tjg d�g d�dtfdtfgd �}t||� t|j|j� d S )
NF��usemask�)r r )r r )rf r r� r� r$ T�r` r` ra rb )r7 r r3 r4 r6 r ri rc �r9 r� r: r; rD rI r> r> r? �
test_standard� s
�
�
z TestMergeArrays.test_standardc C s� | j \}}}}t||fdd�}tjddgdtfddtfgd�}t||� t||fd d�}tjd
d gdtfd ddtfgfgd�}t||� d S )
NTrA )r r r )r r! r" r� r# r! r$ F)r r )r r r� �r7 r r3 r4 r6 r5 r r� r> r> r? � test_flatten� s
�
��z TestMergeArrays.test_flattenc C s� | j \}}}}t||fdd�}tjddgdtfdtfdtfdtfgd �}t||� t||fd
d�}dtfd dtfd dtfdtfd
g fgfgfg}tjddg|d �}t||� d S )NTrA )r r r r( )r r* r, r- r� r. r0 r1 r$ Fr� r/ r� )r r� )r r� r� )r9 r= r: r� rD rI Z controldtyper> r> r? �test_flatten_wflexible� s$
��
��
�z&TestMergeArrays.test_flatten_wflexiblec C s� | j \}}}}tjg d�g d�d�}t||fdd�}tjg d�g d�dtfd tfgd
�}t||� t||fddd �}t||� tt|t�� d S ) N�r r rh �r r r rl Tr� )�r r r� )rf rh )r� r` ra r� r� rb )r� Z
asrecarray) r7 ri r4 r r6 r r �
isinstancer )r9 r� r: ZmxrD rI r> r> r? �test_wmasked_arrays� s
�
z#TestMergeArrays.test_wmasked_arraysc C s^ t t�ddg��dtfg�t�g d��f�}tjg d�g d�dtfdtfgd�}t||� d S ) Nr r r. )r� r� rK )r� r� )rf rK r� r� rb )r r3 r4 rM r6 ri r5 r )r9 rD rI r> r> r? �test_w_singlefield� s �
�z"TestMergeArrays.test_w_singlefieldc C sP | j d }t|t�g d���dtfg�f� tjg d�ddtfdtfgd� d S )Nrf r r^ )�r r r �r! r" r )z-1rf r r# r! r$ )r7 r r3 r4 rM r6 r5 )r9 r< r> r> r? �test_w_shorter_flex� s
"
�z#TestMergeArrays.test_w_shorter_flexc C sd | j \}}}}t|d |d |d fdd�}tjdgdtfdtfddd tfgfgd
�}t||� d S ) Nr Fr� )r r �r r r� r� �f2r# r! r$ r� )r9 r� r: r; r< rD rI r> r> r? �test_singlerecord� s ��z!TestMergeArrays.test_singlerecordN)
r� r� r� r@ r� r� r� r� r� r� r� r� r� r> r> r> r? r� t s
r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �TestAppendFieldsc C sr t �ddg�}t �g d��}t jddgddtfgd�}t jd d
gd tfd d
tfdtfgfgd�}||||f| _d S r r2 r8 r> r> r? r@ � s �
�z TestAppendFields.setup_methodc C sR | j \}}}}t|dg d�d�}tjg d�g d�dtfdtfgd�}t||� d S )Nr r �r7 r� r� r� rb �r7 r ri r4 r6 r �r9 r� r: rD rI r> r> r? �test_append_single s
�z#TestAppendFields.test_append_singlec C s` | j \}}}}t|dg d�ddggd�}tjg d�g d�dtfd tfd
tfgd �}t||� d S ) NrO r �d �� r� ))r r r� )r r r� )rf r rf )�r r r r� )r r r r� r r! rb r� r� r> r> r? �test_append_double s
�z#TestAppendFields.test_append_doublec C sP | j d }t|dg d�d�}tjg d�g d�ddtfdtfgd �}t||� d S )
Nrf r^ r r� )r� r� )rf � �r )r� r� �r r r r# r! rb )r7 r ri r4 r5 r6 r )r9 r< rD rI r> r> r? �test_append_on_flex s
�z$TestAppendFields.test_append_on_flexc C s` | j d }t|dg d�d�}tjg d�g d�dtfdd tfd
tfgfdtfgd �}t||� d S ) Nr r^ r r� ))r r' r )r* r+ r )rf )rf r� r )�r r` r r� )r r� r r. r/ r0 r1 rb )r7 r ri r4 r6 r5 r r� r> r> r? �test_append_on_nested s
��z&TestAppendFields.test_append_on_nestedN)r� r� r� r@ r� r� r� r� r> r> r> r? r� � s
r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�TestStackArraysc C sr t �ddg�}t �g d��}t jddgddtfgd�}t jd d
gd tfd d
tfdtfgfgd�}||||f| _d S r r2 r8 r> r> r? r@ 1 s �
�z TestStackArrays.setup_methodc C sP | j \}}}}t|f�}t||� t||u � t|�}t||� t||u � d S r� )r7 r
r r )r9 r� r: rD r> r> r? r� : s
zTestStackArrays.test_soloc C s� | j \}}}}t||fdd�}t�g d��}t||� t||fdd�}t�g d��}t||� t||fdd�}t�g d��}t||� d S )NFr� )r r r r )r r r r r )r r r r r )r7 r
r3 r4 r r� r> r> r? �test_unnamed_fieldsE s
z#TestStackArrays.test_unnamed_fieldsc C s� | j \}}}}t||f�}tjg d�g d�dtfddtfgd�}t||� t|j|j� t||f�}tjg d�g d�ddtfd tfgd�}t||� t|j|j� t|||f�}tjg d
�g d �ddtfd tfgd�}t||� d S ) N))r rf rf )r rf rf )rf r r )rf r! r )�r r r r� r� r� r� r# r! rb )�r r rf �r! r rf �rf rf r �rf rf r )�r r r r� r� r� r� )r� r� r� r� r� r� )r� r� r� r� r� r� )r7 r
ri r4 r6 r5 r rc r� r> r> r? � test_unnamed_and_named_fieldsU s*
�
�
�z-TestStackArrays.test_unnamed_and_named_fieldsc C s� | j \}}}}tjg d�ddtfdtfgd�}t||f�}tjg d�ddtfdtfgg d�d�}t||� t|j|j� t|||f�}ddtfdtfd tfg}tjg d
�|g d �d�}t||� t|j|j� d S ) N��r. r� � Y@�r/ r� � i@�ry rK � �r@r# r! r^ r$ )r� r� r� r� r� �r� r� r� r� r� �r% rc Zf3))r r rf rf )r! r rf rf )r. r� r� rf )r/ r� r� rf )ry rK r� rf )rf rf rf r )rf rf rf r )�r r r r r� �r r r r r� r� �r r r r r� ) r7 r3 r4 r5 r
ri r rc r6 )r9 r� r: r< �zzrD rI rR r> r> r? �test_matching_named_fieldsu s&
�
�
�
z*TestStackArrays.test_matching_named_fieldsc C s� | j \}}}}tjg d�ddtfdtfgd�}dddd d
�}t||f|d �}tjg d �ddtfdtfgg d
�d�}t||� t|j |j � t|j|j� d S )Nr� r# r! r^ r$ z???g 8��� ���g �i��)r r! r^ �D)�defaults))r r r� )r! r r� r� r� r� r� r� )r7 r3 r4 r5 r
ri r rc )r9 r� r< r� r� rD rI r> r> r? �
test_defaults� s
�
�
z TestStackArrays.test_defaultsc C s� dt fdtfdtfg}tjdgdg|d�}dt fdtfdtfg}tjdg|d�}tjddgdd g|d�}t||fd
d �}t||� t|j|j� tt �� t||fd d � W d � n1 s�0 Y d S )
Nr r! r^ r� �r r r rb �r* r, rm r$ r� T)Z autoconvertF)
r6 �boolr5 ri r4 r
r rc r � TypeError)r9 �adtyper. �bdtyper/ rI rD r> r> r? �test_autoconversion� s �
z#TestStackArrays.test_autoconversionc C s� dt fdtfdtfg}tjdgdg|d�}dt fdtfdtfg}tjdg|d�}t||f�}tjddgdd g|d�}t||� t|j|j� d S )
N)r. r )r/ r! )ry r^ r� r rb r r$ r� )r6 r r5 ri r4 r
r rc )r9 r r. r r/ rD rI r> r> r? �test_checktitles� s �
z TestStackArrays.test_checktitlesc C s t jddgddtdfgd�}t jddgd fd
d gd fd
dgdfgddtdfdtfgd�}t||f�}tjddgdfddgdfddgd fdd gd fddgdfgddgdfddgdfddgdfddgdfddgdfg|jd�}t|j|j� t||� t|j|j� d S ) Nr� )r! r r# r! rF r$ r. r� r� r/ r� r� ry rK r� r^ � Ar r � Br" � a� b� cFT)r7 rc r% )r3 r4 r5 r
ri r% r rc )r9 r< r� r� �expectedr> r> r? �
test_subdtype� s: � � ��
�
��
z TestStackArrays.test_subdtypeN) r� r� r� r@ r� r� r� r� r r r r r> r> r> r? r� / s 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 )�
TestJoinByc C s� t jttt �d�t �dd�t �dd���dtfdtfdtfgd �| _t jttt �d
d �t �d d
�t �dd���dtfdtfdtfgd �| _d S )Nr �2 �< r� �n r. r/ ry r$ r, r� �A �K r� �r3 r4 �list�zipr� r6 r. r/ �r9 r> r> r? r@ � s
��
��zTestJoinBy.setup_methodc C s\ | j | j }}td||dd�}tjg d�dtfdtfdtfdtfdtfgd �}t||� d S )
Nr. �inner)�jointype))r, �7 r �i r� )rm �8 �B �j �e )r| �9 �C �k �f )r} �: �D �l �g )r� �; �E �m �h �b1�b2ry r� r$ �r. r/ r r3 r4 r6 r r� r> r> r? �test_inner_join� s
��zTestJoinBy.test_inner_joinc C sH | j | j }}td||� tjg d�dtfdtfdtfdtfgd� d S )NrT ))r, r r r� )rm r r r! )r| r" r$ r% )r} r& r( r) )r� r* r, r- r. r/ ry r� r$ )r. r/ r r3 r4 r6 �r9 r. r/ r> r> r? � test_join� s
��zTestJoinBy.test_joinc C s^ t jdgdtfgd�}t jdt �g d��fgdtfdgd�}td||�}t||�tj�� d S )NrF rg r$ r r� )�valueZuint16rh )r3 r4 r6 r r rM ri Z MaskedArray)r9 ZfooZbarr� r> r> r? �test_join_subdtype s �
� z TestJoinBy.test_join_subdtypec C sZ | j | j }}td||d�}tjg d�g d�dtfdtfdtfdtfgd �}t||� d S )
NrT Zouter)�r r r� rf �r �3 r! rf �r �4 r% rf �rh �5 r) rf �r* �6 r- rf �r, r r rf )r, r rf r� �rm r r rf )rm r rf r! �r| r" r$ rf )r| r# rf r% �r} r& r( rf )r} r' rf r) �r� r* r, rf )r� r+ rf r- )r �F rf r )r~ �G rf r )r �H rf r$ )rp �I rf r( )r� �J rf r, )r� r� r� r� r� r� �r r r r r� rI r� rI r� rI r� rI rI rI rI rI rI r. r/ ry r� rb �r. r/ r ri r4 r6 r r� r> r> r? �test_outer_join s
��zTestJoinBy.test_outer_joinc C sZ | j | j }}td||d�}tjg d�g d�dtfdtfdtfdtfgd �}t||� d S )
NrT Z leftouter)
r6 r7 r9 r; r= r? r@ rA rB rC )
r� r� r� r� r� r� r� r� r� r� r. r/ ry r� rb rJ r� r> r> r? �test_leftouter_join( s
� z TestJoinBy.test_leftouter_joinc C sP t jdg d�d�}t jdg d�d�}tddg||ddd �}t|jjg d
�� d S ) Nrh �rw r� r� r$ �r� r� rw ry r/ r F)r r� )r/ ry Za1Za2)r3 rq r� r r r% �names)r9 r. r/ �jr> r> r? �test_different_field_order9 s z%TestJoinBy.test_different_field_orderc C s< t jdg d�d�}t jdg d�d�}tttg d�||� d S )Nrh rM r$ rN )r. r/ r/ )r3 rq r� r r� r r2 r> r> r? �test_duplicate_keysA s z TestJoinBy.test_duplicate_keysc C sj t �ddg�}t �ddg�}t �g d��}t jddg|d�}t jdd g|d�}td
||�}t|j|� d S ) N)rg ZS5�r4 �<f4�rg ZS10)rU �Zvalue1rT )�value2rT ��Sarahr� ��Johnr- r$ �rY r� �r[ r� rg �r3 r% r4 r r �r9 �a_dtype�b_dtype�expected_dtyper. r/ r� r> r> r? �#test_same_name_different_dtypes_keyF s z.TestJoinBy.test_same_name_different_dtypes_keyc C sj t �ddg�}t �ddg�}t �g d��}t jddg|d�}t jdd g|d�}td
||�}t|j|� d S ) NrU rS )r4 �<f8))rg z|S10rV )rW rd rX rZ r$ r\ r] rg r^ r_ r> r> r? � test_same_name_different_dtypesR s z*TestJoinBy.test_same_name_different_dtypesc C s� t �dtdfdg�}t jg d�t jfg d�dfg|d�}t �dtdfdg�}t jg d�dfg d �dfg|d�}t �dtdfddg�}t jg d�t jdfg|d�}td||�}t|j|� t||� d S )
N�posrh )�frT )r r r r� r� r$ )�grT )rh r r )r3 r% r6 r4 Zpir r )r9 r` r. ra r/ rb r r� r> r> r? �test_subarray_key_ s $" z TestJoinBy.test_subarray_keyc C sn t jddd�}d|_tt|j�d� t �ddg|�}t �dd g|�}td
||�}t �g d ��}t|j|� d S ) Nzi1,f4Tro )�k�vrh r� )rh r r� r� rj ))rj Zi1)Zv1rz )Zv2rz ) r3 r% rO r �lenZdescrr4 r r )r9 ru r. r/ r� rb r> r> r? �test_padded_dtypem s z TestJoinBy.test_padded_dtypeN)r� r� r� r@ r1 r3 r5 rK rL rQ rR rc re ri rm r> r> r> r? r � s
r c @ s8 e Zd Zedd� �Zdd� Zdd� Zdd� Zd d
� Zd S ) � TestJoinBy2c C s� t jttt �d�t �dd�t �dd���dtfdtfdtfgd �| _t jttt �d�t �d
d �t �dd���dtfdtfd tfgd �| _d S )
Nr r r r� r r. r/ ry r$ r r r� r )�clsr> r> r? r@ s
��
��zTestJoinBy2.setup_methodc C s` | j | j }}td||dddd�}tjg d�dtfdtfdtfd tfd
tfgd �}t||� d S ) Nr. rB �2r �� r1postfix� r2postfixr �
)r r r r� r� )r r8 r r! r! )r r: r# r% r% )rh r< r' r) r) )r* r> r+ r- r- )r, r rD r r )rm r rE r r )r| r" rF r$ r$ )r} r& rG r( r( )r� r* rH r, r, r/ r/ ry r� r$ r0 r� r> r> r? �test_no_r1postfix� s �
��z TestJoinBy2.test_no_r1postfixc C s t ttd| j| jddd� d S )Nr. rB )rr rs )r r� r r. r/ r r> r> r? �test_no_postfix� s �zTestJoinBy2.test_no_postfixc C s` | j | j }}td||dddd�}tjg d�dtfdtfdtfd tfd
tfgd �}t||� d S ) Nr. �1rB r rq rt r. r/ ry r� r$ r0 r� r> r> r? �test_no_r2postfix� s �
��z TestJoinBy2.test_no_r2postfixc
C s& t jttt �ddgd�t �t �d�d�t �dd�t �dd���dtfd tfd
tfd tfgd �}t jttt �ddgd�t �t �d�d�t �d
d�t �dd���dtfd tfd
tfd tfgd �}t jg d�dtfd tfdtfdtfdtfdtfgd �}td dg||dddd�}t |j
|j
� t ||� d S )Nr r~ r, r r r r rj r. r/ ry r$ r r r )
)r r r r r r )r~ r r8 r r~ r )r r r: r# r r )r~ r r< r' rp rh )r r r> r+ r� r* )r~ r r rD r� r, )r rh r rE � rm )r~ rh r" rF r� r| )r r* r& rG � r} )r~ r* r* rH � r� r. r/ Zc1Zc2rw rp r rq ) r3 r4 r r Ztile�repeatr� r6 r r r% )r9 r. r/ rI rD r> r> r? �test_two_keys_two_vars� s( &��&��
���z"TestJoinBy2.test_two_keys_two_varsN) r� r� r� � classmethodr@ ru rv rx r} r> r> r> r? rn ~ s
rn c @ s e Zd ZdZdd� Zdd� ZdS )�TestAppendFieldsObjz;
Test append_fields with arrays containing objects
c C s$ ddl m} t|ddd�d�| _d S )Nr )�datei� r )�obj)Zdatetimer� �dictr7 )r9 r� r> r> r? r@ � s z TestAppendFieldsObj.setup_methodc C s� | j d }tj|df|dfgdtfdtfgd�}tjddgtd�}t|d |d
d �}tj|ddf|ddfgdtfdtfd tfgd�}t||� d S )
z7Test append_fields when the base array contains objectsr� r r" r r! r$ r r r^ F)r7 r� N)r7 r3 r4 �objectr5 r6 r r )r9 r� r: r; rD rI r> r> r? �test_append_to_objects� s
��z*TestAppendFieldsObj.test_append_to_objectsN)r� r� r� �__doc__r@ r� r> r> r> r? r � s r ).ZpytestZnumpyr3 Znumpy.mari Znumpy.ma.mrecordsr Znumpy.ma.testutilsr Z
numpy.testingr r Znumpy.lib.recfunctionsr r r r r
r r r
r r r r r r r �libZ recfunctionsZ_get_fieldspecZ
get_fieldspecrP rU Z
_zip_descrrC Z
_zip_dtypeZ zip_dtyper r� r� r� r� r rn r r> r> r> r? �<module> s0 D
K 6 , %A
Mini Shell