|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/numpy/lib/tests/__pycache__/ |
Upload File : |
o
6��aà � @ 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 �r 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? �C/usr/lib/python3/dist-packages/numpy/lib/tests/test_recfunctions.py�setup s
�
�zTestRecFunctions.setupc 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��flatten� Fr$ r" r/ r1 r2 r0 )r8 � zip_descrr r4 r& r7 r6 )r: 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/ r0 r1 r2 r% )r( )r, �r �r+ )r )r) )r+ )r. r? )r4 r5 r7 r6 r r �r: r/ rF �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/ r0 r1 r2 r% r �BB)r/ r2 )r4 r5 r7 r6 r �viewr r& )r: r/ rF �newdtyperK r? r? r@ �test_rename_fieldsa s "��
z#TestRecFunctions.test_rename_fieldsc C � 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/ r0 r1 r2 )r/ )r0 )r1 r2 )r/ )r0 r? r? )r4 r& r6 � get_namesr r7 �r: �ndtyperF r? r? r@ �test_get_namesl �
"
z TestRecFunctions.test_get_namesc C rR ) Nr$ r" rS r/ r0 r1 r2 )r/ r0 r1 r2 �r/ r0 r? )r4 r& r6 �get_names_flatr r7 rU r? r? r@ �test_get_names_flat~ rX 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" rS r �BA�rN z|S1)r r" r\ rN rN �BBA�BBB)r r" r\ rN r^ r_ )r4 r& r6 r r r7 )r: rV rF rK 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 rb ra )r )r r" ra )r )r# �C)�r �r r rd rd rd )r re )r �r r r r" r\ r] ��maskr& FT��
ignoremask� return_indexr r ���)�keyrk )r r r � r- )r r r r+ rN )r r r rn r+ )�mar5 r7 r6 r
r �sortedrJ 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 rn rn )r r r r r r r �rh Tri )r r rn r+ rl r F)r r r rn r+ � )r7 ro r5 rO r
r rp )r: rV r/ rF rK 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��alignr r% �
) r4 r& �zerosr r �itemsize�recordr �type)r: �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)r0 zf4,u2)�c�f4r r% )r+ r- �f8))r r r- )r+ r- � )r� � � )r r� � )�r; r )r< r� )r= r� r; r= rl )�axis)r) g @� "@� &@)r � @� @� $@r r )r r r) r� r- )r. r� � @r� r )r� r� g *@g ,@� )g 0@� g 2@g 3@r~ r0 )�f0r� )�f1�u2)r� r� �r )gUUUUUU@gUUUUUU@gUUUUUU!@r� )r� )r< r )r= r r r rn rs r� r� � r� r� rw � ))�x0r )�x1)r r )�x2)r �r r r< r/ r� )r rs c S s t �| |f�S �N)r4 r& )r| �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�ir| r? r? r@ �
<listcomp>$ s zXTestRecFunctions.test_structured_to_unstructured.<locals>.structured.<locals>.<listcomp>)r4 r& � enumerate)�dtsr? r? r@ �
structured# s zDTestRecFunctions.test_structured_to_unstructured.<locals>.structuredc S s2 t �d| �}t||d�}t|| �}|j|j|jfS )Nr? r% )r4 rx r r r� r& )r| r& �arr�ret�backarrr? r? r@ �inspect& s
zATestRecFunctions.test_structured_to_unstructured.<locals>.inspect)rs rI rH )r- )rn r r� )r4 rx r r r5 �mean�arange�reshaper r& r r �baser7 �int32r �
ValueError�NotImplementedError)r: r/ �outr0 r� �want�d�dd�ddd�point�triangler� �resr� r� r� r| 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 )r~ �r0 r� �r� �u1r% �r0 r� r� �r r )rn r+ F)�zero_unassigned)r r r )r rn r+ )r r r )r rn r+ r/ r� )r� )�rn )r� )�r rn ))r r ))r rn rn r r? )r4 �onesr r r5 r r& )r: r/ �newdtr0 r? r? r@ � test_field_assignment_by_nameA s&
z.TestRecFunctions.test_field_assignment_by_nameN)�__name__�
__module__� __qualname__rA rG rL rQ rW r[ r` rq rt r} 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 � )r4 r5 r7 r6 rx r& r r �r: r/ r0 rF rK 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 rf r r" rg r� r% r� )r� rf re )ro r5 r7 r6 rx 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/ r0 r1 r2 �bcr3 r9 r? r? r@ rA w s � �zTestMergeArrays.setupc C s| | j \}}}}t|�}tjddgdtfgd�}t||� t|f�}t||� t|dd�}t||� t|dd�}t||� d S )NrH r� r� r% FrB T)r8 r r4 r5 r7 r �r: �_r; r= rF rK r? r? r@ � test_solo� s
zTestMergeArrays.test_soloc C s^ | j d }t|dd�}t||� t|dd�}tjddgdtfdtfd tfgd
�}t||� d S ) Nr FrB T)r r r) )r+ r- r. r/ r1 r2 r% )r8 r r r4 r5 r7 r6 �r: r> rF rK 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 )rl r r� r� r% T�re re rf rg )r8 r r4 r5 r7 r ro rh �r: r� r; r< rF rK 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 )
NTrB )r r r )r r" r# r� r$ r" r% F)r r )r r! r� �r8 r r4 r5 r7 r6 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 )NTrB )r r r r) )r r+ r- r. r� r/ r1 r2 r% Fr� r0 r� )r r� )r r� r� )r: r> r; r� rF rK � 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 rn �r r r rr Tr� )�r r r� )rl rn )r� re rf r� r� rg )r� �
asrecarray) r8 ro r5 r r7 r r �
isinstancer )r: r� r; �mxrF rK 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� rM )r� r� )rl rM r� r� rg )r r4 r5 rO r7 ro r6 r )r: rF rK 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 )Nrl r rc )�r r r �r" r# r )z-1rl r r$ r" r% )r8 r r4 r5 rO r7 r6 )r: 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� )r: r� r; r< r= rF rK r? r? r@ �test_singlerecord� s ��z!TestMergeArrays.test_singlerecordN)
r� r� r� rA 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 r r r3 r9 r? r? r@ rA � � �
�zTestAppendFields.setupc C sR | j \}}}}t|dg d�d�}tjg d�g d�dtfdtfgd�}t||� d S )Nr r �r8 r� r� r� rg �r8 r ro r5 r7 r �r: r� r; rF rK 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 ) NrS r �d �� r� ))r r r� )r r r� )rl r rl )�r r r r� )r r r r� r r" rg 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 )
Nrl rc r r� )r� r� )rl � �r )r� r� �r r r r$ r" rg )r8 r ro r5 r6 r7 r )r: r= rF rK 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 rc r r� ))r r( r )r+ r, r )rl )rl r� r )�r re r r� )r r� r r/ r0 r1 r2 rg )r8 r ro r5 r7 r6 r r� r? r? r@ �test_append_on_nested s
��z&TestAppendFields.test_append_on_nestedN)r� r� r� rA 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 r r r3 r9 r? r? r@ rA 1 r� zTestStackArrays.setupc C sP | j \}}}}t|f�}t||� t||u � t|�}t||� t||u � d S r� )r8 r
r r )r: r� r; rF 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 )r8 r
r4 r5 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 rl rl )r rl rl )rl r r )rl r" r )�r r r r r� r� r� r$ r" rg )�r r rl �r" r rl �rl rl r �rl rl r )�r r r r r� r� r� )r r r r r r )r r r r r� r� )r8 r
ro r5 r7 r6 r rh 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@�r0 r� � i@�r� rM � �r@r$ r" rc r% )r r r r
r �r r r� r� r� �r& rh �f3))r r rl rl )r" r rl rl )r/ r� r rl )r0 r� r rl )r� rM r rl )rl rl rl r )rl rl rl r )�r r r r r �r r r r r r �r r r r r ) r8 r4 r5 r6 r
ro r rh r7 )r: r� r; r= �zzrF rK rV 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" rc r% z???g 8��� ���g �i��)r r" rc �D)�defaults))r r r )r" r r r r
r r r )r8 r4 r5 r6 r
ro r rh )r: r� r= r r rF rK 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 � d S 1 s`w Y d S )
Nr r" rc r� �r r r rg �r+ r- rs r% r� T)� autoconvertF)
r7 �boolr6 ro r5 r
r rh r � TypeError)r: �adtyper/ �bdtyper0 rK rF 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 )r0 r" )r� rc r� r rg r r% r� )r7 r r6 ro r5 r
r rh )r: r" r/ r# r0 rF rK 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" rH r% r/ r� r r0 r� r r� rM r rc � Ar r � Br# � a� b� cFT)r8 rh r& )r4 r5 r6 r
ro r& r rh )r: r= r r� �expectedr? r? r@ �
test_subdtype� s: � � ��
�
��
z TestStackArrays.test_subdtypeN) r� r� r� rA r� r r r r r$ r% r, r? r? r? r@ r / s r c @ sz e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
e j j
dd�dd� �Zdd� Zdd� Zdd� 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/ r0 r� r% r- r� �A �K r� �r4 r5 �list�zipr� r7 r/ r0 �r: r? r? r@ rA � s
��
� �zTestJoinBy.setupc C s\ | j | j}}td||dd�}tjg d�dtfdtfdtfdtfdtfgd �}t||� d S )
Nr/ �inner)�jointype))r- �7 r1 �i r� )rs �8 �B �j �e )r� �9 �C �k �f )r� �: �D �l �g )r� �; �E �m �h �b1�b2r� r� r% �r/ r0 r r4 r5 r7 r r� r? r? r@ �test_inner_join� s
��zTestJoinBy.test_inner_joinc C sH | j | j}}td||� tjg d�dtfdtfdtfdtfgd� d S )NrY ))r- r9 r: r� )rs r; r= r> )r� r? rA rB )r� rC rE rF )r� rG rI rJ r/ r0 r� r� r% )r/ r0 r r4 r5 r7 �r: r/ r0 r? r? r@ � test_join� s
�
�zTestJoinBy.test_joinc C s^ t jdgdtfgd�}t jdt �g d��fgdtfdgd�}td||�}t||�tj�� d S )NrH rm r% r r� )�value�uint16rn )r4 r5 r7 r r rO ro � MaskedArray)r: �foo�barr� r? r? r@ �test_join_subdtype s �
� z TestJoinBy.test_join_subdtypec C �Z | j | j}}td||d�}tjg d�g d�dtfdtfdtfdtfgd �}t||� d S )
NrY �outer)�r r. r� rl �r �3 r> rl �r �4 rB rl �rn �5 rF rl �r+ �6 rJ rl �r- r9 r: rl )r- r1 rl r� �rs r; r= rl )rs r< rl r> �r� r? rA rl )r� r@ rl rB �r� rC rE rl )r� rD rl rF �r� rG rI rl )r� rH rl rJ )r �F rl r: )r� �G rl r= )r� �H rl rA )rw �I rl rE )r� �J rl rI )r r r r r r �r r r r r rl r rl r rl r rl rl rl rl rl rl r/ r0 r� r� rg �r/ r0 r ro r5 r7 r r� r? r? r@ �test_outer_join s
��zTestJoinBy.test_outer_joinc C rW )
NrY � leftouter)
rY rZ r\ r^ r` rb rc rd re rf )
r r r r r r r r r r r/ r0 r� r� rg rm 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 ) Nrn �r~ r� r� r% �r� r� r~ r� r0 r7 F)r8 r� )r0 r� �a1�a2)r4 rx r� r r r& �names)r: r/ r0 �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 )Nrn rq r% rr )r/ r0 r0 )r4 rx r� r r� r rO r? r? r@ �test_duplicate_keysA s z TestJoinBy.test_duplicate_keyszSee comment at gh-9343)�reasonc C sj t �ddg�}t �ddg�}t �g d��}t jddg|d�}t jdd g|d�}td
||�}t|j|� d S ) N)rm �S5�rQ �<f4�rm �S10)r} ��value1r| )�value2r| ��Sarahr� ��Johnr. r% �r� r� �r� r� rm �r4 r& r5 r r �r: �a_dtype�b_dtype�expected_dtyper/ r0 r� r? r? r@ �#test_same_name_different_dtypes_keyF � 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 ) Nr} r{ )rQ �<f8))rm z|S10r )r� r� r� r� r% r� r� rm r� r� r? r? r@ � test_same_name_different_dtypesS r� 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�posrn )�fr| )r r r r� r� r% )�gr| )rn r r )r4 r& r7 r5 �pir r )r: r� r/ r� r0 r� 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,f4Tru )�k�vrn r� )rn r r� r� r� ))r� �i1)�v1r� )�v2r� ) r4 r& ru r �len�descrr5 r r )r: r| r/ r0 r� r� r? r? r@ �test_padded_dtypen s z TestJoinBy.test_padded_dtypeN)r� r� r� rA rN rP rV rn rp rw rx �pytest�mark�xfailr� r� r� r� 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� r0 r/ r0 r� r% r1 r2 r� r3 )�clsr? r? r@ rA � s
��
� �zTestJoinBy2.setupc C �` | j | j}}td||dddd�}tjg d�dtfdtfdtfd tfd
tfgd �}t||� d S ) Nr/ rD �2r7 �� r1postfix� r2postfixr8 �
)r r. r1 r� r� )r r[ r< r> r> )r r] r@ rB rB )rn r_ rD rF rF )r+ ra rH rJ rJ )r- r9 rg r: r: )rs r; rh r= r= )r� r? ri rA rA )r� rC rj rE rE )r� rG rk rI rI r0 rL r� r� r% rM r� r? r? r@ �test_no_r1postfix� � �
��z TestJoinBy2.test_no_r1postfixc C s t ttd| j| jddd� d S )Nr/ rD )r� r� )r r� r r/ r0 r6 r? r? r@ �test_no_postfix� s
�zTestJoinBy2.test_no_postfixc C r� ) Nr/ �1rD r7 r� r� rK r0 r� r� r% rM r� r? r? r@ �test_no_r2postfix� r� 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 r� r/ r0 r� r% r1 r2 r )
)r r r. r1 r r )r� r r[ r< r� r )r r r] r@ r� r )r� r r_ rD rw rn )r r ra rH r� r+ )r� r r9 rg r� r- )r rn r; rh � rs )r� rn r? ri r� r� )r r+ rC rj � r� )r� r+ rG rk � r� rK rL �c1�c2r� r� r7 r� ) r4 r5 r4 r5 �tile�repeatr� r7 r r r& )r: r/ r0 rK rF r? r? r@ �test_two_keys_two_vars� s( &��&��
���z"TestJoinBy2.test_two_keys_two_varsN) r� r� r� � classmethodrA r� r� r� r� r? r? r? r@ r� s
r� 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)�datetimer� �dictr8 )r: r� r? r? r@ rA � s zTestAppendFieldsObj.setupc 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 rc F)r8 r� N)r8 r4 r5 �objectr6 r7 r r )r: r� r; r< rF rK r? r? r@ �test_append_to_objects� s
��z*TestAppendFieldsObj.test_append_to_objectsN)r� r� r� �__doc__rA r� r? r? r? r@ r� � s r� ).r� �numpyr4 �numpy.maro �numpy.ma.mrecordsr �numpy.ma.testutilsr �
numpy.testingr r �numpy.lib.recfunctionsr r r r r
r r r
r r r r r r r �lib� recfunctions�_get_fieldspec�
get_fieldspecrT rZ �
_zip_descrrE �
_zip_dtype� zip_dtyper r� r� r� r r- r� r� r? r? r? r@ �<module> s2 D
K 6 , &A