Edit File: test_recfunctions.cpython-37.pyc
B ��Fd%z � @ s6 d dl mZmZmZ d dlZd dlmZ d dlm Z d dl m Z d dl m Z mZmZmZ d dlmZmZmZmZmZmZmZmZmZ ejj j Z ejj j Z ejj j Z G dd� de �Z G d d � d e �Z!G d d � d e �Z"G d d� de �Z#G dd� de �Z$G dd� de �Z%G dd� de �Z&G dd� de �Z'e(dk�r2e� dS )� )�division�absolute_import�print_functionN)� MaskedRecords)� assert_equal)�TestCase�run_module_suite�assert_� assert_raises) � drop_fields� rename_fields�get_fieldstructure�recursive_fill_fields�find_duplicates� merge_arrays� append_fields� stack_arrays�join_byc @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )�TestRecFunctionsc C st t �ddg�}t �dddg�}t jddgdd tfgd �}t jd d gd tfddtfdtfgfgd �}||||f| _d S )N� � � � � )�Ag �?)�Bg @)r z|S3r )�dtype)r )r g @)� )� g @�a�b�ba�bb)�np�array�float�int�data)�self�x�y�z�w� r- ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-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� F)r z|S3r r r! r" r )r' � zip_descrr r# r r& r% )r( r, r) r* r+ �testr- r- r. �test_zip_descr s0 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|d k� d S )N)r )r g @)r )r g @r r r! r" )r ))r g @))r g @)r )r )r )g @)r )g @)r# r$ r& r% r r r )r( r r3 �controlr- r- r. �test_drop_fields> 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 g @g >@r r g @g N@r r r! r" )r r �BB)r r" )r# r$ r& r% r �viewr r )r( r r3 �newdtyper5 r- r- r. �test_rename_fields\ s " z#TestRecFunctions.test_rename_fieldsc C s\ t �ddtfg�}t|�}t|d� t �dtfddtfdtfgfg�}t|�}t|d� d S ) N)r z|S3r )r r r r r! r" )r )r )r! r" )r# r r% � get_namesr r& )r( �ndtyper3 r- r- r. �test_get_namesg s "z TestRecFunctions.test_get_namesc C s\ t �ddtfg�}t|�}t|d� t �dtfddtfdtfgfg�}t|�}t|d� d S ) N)r z|S3r )r r r r r! r" )r r r! r" )r# r r% �get_names_flatr r& )r( r< r3 r- r- r. �test_get_names_flatq 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||� d S ) N)r z|S3r )r r r �BA)r7 z|S1)r r r@ r7 r7 �BBA�BBB)r r r@ r7 rA rB )r# r r% r r r& )r( r< r3 r5 r- r- r. �test_get_fieldstructure{ s z(TestRecFunctions.test_get_fieldstructurec C s� t jddddddgddddddgdtfd d tfd gfgd �}t|d dd�}ddg}tt|d �|� t|d ||d � t|ddd�}dddddg}tt|d �|� t|d ||d � t|d dd�}ddddg}tt|d �|� t|d ||d � t|d dd�}ddddg}tt|d �|� t|d ||d � t|ddd�}dddddg}tt|d �|� t|d ||d � d S )N)r )g @r )r )g @r )r )g �?r )r )g @�C)r )r r )r )r r )r )r r r r r@ )r7 z|S1)�maskr FT)� ignoremask� return_indexr r ���)�keyrG r � r r r7 )�mar$ r& r% r r �sorted)r( r r3 r5 r- r- r. �test_find_duplicates� s2 z%TestRecFunctions.test_find_duplicatesc C s� dt fg}tjdddddddgdddddddgd��|�}t|ddd�}dddd g}tt|d �|� t|d ||d � t|d dd�}ddddd d g}tt|d �|� t|d ||d � d S ) Nr r r rJ r )rE T)rF rG r rH F� )r& rK r$ r8 r r rL )r( r< r r3 r5 r- r- r. � test_find_duplicates_ignoremask� s z0TestRecFunctions.test_find_duplicates_ignoremaskN) �__name__� __module__� __qualname__r/ r4 r6 r: r= r? rC rM rO r- r- r- r. r s ! !r c @ s e Zd Zdd� Zdd� ZdS )�TestRecursiveFillFieldsc C sf t jddgdtfdtfgd�}t jd|jd�}t||�}t jdddgdtfdtfgd�}t||� d S )N)r g $@)r g 4@r r )r )rJ )r g )r# r$ r& r% �zerosr r r )r( r r r3 r5 r- r- r. �test_simple_flexible� s z,TestRecursiveFillFields.test_simple_flexiblec C st t jddgddgdtfdtfgd�}t jd|jd �}t||�}t jddd gddd gdtfdtfgd�}t||� d S ) N)r g $@)r g 4@)r r )r r r r )rE r )rJ )r )r g )r r )rK r$ r& r% rT r r r )r( r r r3 r5 r- r- r. �test_masked_flexible� s z,TestRecursiveFillFields.test_masked_flexibleN)rP rQ rR rU rV r- r- r- r. rS � s rS c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )�TestMergeArraysc C st t �ddg�}t �dddg�}t jddgdd tfgd �}t jd d gd tfddtfdtfgfgd �}||||f| _d S )Nr r r r r )r g �?)r g @)r z|S3r )r )r )r g @)r )r g @r r r! r" )r# r$ r% r& r' )r( r) r* r+ r, r- r- r. r/ � s zTestMergeArrays.setUpc C s| | j \}}}}t|�}tjddgdtfgd�}t||� t|f�}t||� t|dd�}t||� t|dd�}t||� d S )N)r )r �f0)r F)r0 T)r' r r# r$ r& r )r( �_r) r+ r3 r5 r- r- r. � test_solo� s zTestMergeArrays.test_soloc C s^ | j d }t|dd�}t||� t|dd�}tjddgdtfdtfd tfgd �}t||� d S ) Nr F)r0 T)r r g @)r r g @r r! r" )r )r' r r r# r$ r& r% )r( r, r3 r5 r- r- r. �test_solo_w_flatten� s z#TestMergeArrays.test_solo_w_flattenc C s� | j \}}}}t||fdd�}tjdddgdtfdtfgd�}t||� t||fd d�}tjdddgd d d gdtfdtfgd �}t||� t|j|j� d S ) NF)�usemask)r r )r r )rH r rX �f1)r T)r r )r r )rE r )r' r r# r$ r& r rK rE )r( rY r) r* r3 r5 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 ) NT)r0 )r r g �?)r r g @rX )r z|S3r )r F)r )r g �?)r )r g @r] )r' r r# r$ r& r% r )r( rY r) r+ r3 r5 r- r- r. � test_flatten s z TestMergeArrays.test_flattenc C s� | j \}}}}t||fdd�}tjddgdtfdtfdtfdtfgd �}t||� t||fd d�}dtfd dtfd dtfdtfgfgfg}tjd dg|d �}t||� d S )NT)r0 )r r r g @)r r r g @rX r r! r" )r Fr] r )g �?)r )r g @)r )r )r g @)r' r r# r$ r& r% r )r( r, r) rY r3 r5 Z controldtyper- r- r. �test_flatten_wflexible s z&TestMergeArrays.test_flatten_wflexiblec C s� | j \}}}}tjdddgdddgd�}t||fdd�}tjdd d gd d d gdtfdtfgd�}t||� t||fddd�}t||� tt|t�� d S )Nr r rJ r )rE T)r\ )r r )r r )rH rJ )r r )r r )r r rX r] )rE r )r\ Z asrecarray) r' rK r$ r r& r r � isinstancer )r( rY r) �mxr3 r5 r- r- r. �test_wmasked_arrays/ s z#TestMergeArrays.test_wmasked_arraysc C sd t t�ddg��dtfg�t�dddg�f�}tjddd gd d d gdtfd tfgd �}t||� d S )Nr r r g $@g 4@g >@)r g $@)r g 4@)rH g >@)r r )r r r] )rE r )r r# r$ r8 r& rK r% r )r( r3 r5 r- r- r. �test_w_singlefield< s z"TestMergeArrays.test_w_singlefieldc C sT | j d }t|t�dddg��dtfg�f� tjdddgd d tfdtfgd � d S ) NrH r r r rD )r g �?r )r g @r )z-1rH r )r z|S3r )r )r' r r# r$ r8 r& r% )r( r+ r- r- r. �test_w_shorter_flexE s $ z#TestMergeArrays.test_w_shorter_flexc C sd | j \}}}}t|d |d |d fdd�}tjdgdtfdtfddd tfgfgd �}t||� d S ) Nr F)r\ )r r )r r rX r] �f2)r z|S3r )r )r' r r# r$ r& r% r )r( rY r) r* r+ r3 r5 r- r- r. �test_singlerecordT s z!TestMergeArrays.test_singlerecordN) rP rQ rR r/ rZ r[ r^ r_ r` rc rd re rg r- r- r- r. rW � s rW c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestAppendFieldsc C st t �ddg�}t �dddg�}t jddgdd tfgd �}t jd d gd tfddtfdtfgfgd �}||||f| _d S )Nr r r r r )r g �?)r g @)r z|S3r )r )r )r g @)r )r g @r r r! r" )r# r$ r% r& r' )r( r) r* r+ r, r- r- r. r/ a s zTestAppendFields.setUpc C sX | j \}}}}t|ddddgd�}tjdddgd d d gd tfdtfgd �}t||� d S ) Nr r r r )r' )r r )r r )rH r )r r )r r rX )rE r )r' r rK r$ r& r )r( rY r) r3 r5 r- r- r. �test_append_singlej s z#TestAppendFields.test_append_singlec C sf | j \}}}}t|ddddgddggd�}tjdd d gd d d gd tfdtfdtfgd�}t||� d S )N)r r r r r �d �� )r' )r r rj )r r rk )rH r rH )r r r )r r r rX r r )rE r )r' r rK r$ r& r )r( rY r) r3 r5 r- r- r. �test_append_doubles s z#TestAppendFields.test_append_doublec C sV | j d }t|ddddgd�}tjddd gd d d gd d tfdtfgd�}t||� d S )NrH rD r r r )r' )r g �?r )r g @r )rH g �r )r r r )r r r )r z|S3r )rE r )r' r rK r$ r% r& r )r( r+ r3 r5 r- r- r. �test_append_on_flex| s z$TestAppendFields.test_append_on_flexc C sf | j d }t|ddddgd�}tjddd gd d d gd tfd dtfdtfgfdtfgd�}t||� d S )Nr rD r r r )r' )r )r g @r )r )r g @r )rH )rH g �r )r )r r r )r )r r r r r r! r" )rE r )r' r rK r$ r& r% r )r( r, r3 r5 r- r- r. �test_append_on_nested� s z&TestAppendFields.test_append_on_nestedN)rP rQ rR r/ ri rl rm rn r- r- r- r. rh ^ s rh c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�TestStackArraysc C st t �ddg�}t �dddg�}t jddgdd tfgd �}t jd d gd tfddtfdtfgfgd �}||||f| _d S )Nr r r r r )r g �?)r g @)r z|S3r )r )r )r g @)r )r g @r r r! r" )r# r$ r% r& r' )r( r) r* r+ r, r- r- r. r/ � s zTestStackArrays.setUpc C sT | j \}}}}t|f�}t||� | �||k� t|�}t||� | �||k� d S )N)r' r r � assertTrue)r( rY r) r3 r- r- r. rZ � s zTestStackArrays.test_soloc C s� | j \}}}}t||fdd�}t�ddddg�}t||� t||fdd�}t�dddddg�}t||� t||fdd�}t�dddddg�}t||� d S )NF)r\ r r r r r )r' r r# r$ r )r( rY r) r* r3 r5 r- r- r. �test_unnamed_fields� s z#TestStackArrays.test_unnamed_fieldsc C s | j \}}}}t||f�}tjddddgddddgdtfdd tfgd �}t||� t|j|j� t||f�}tjd d d dgddddgdd tfdtfgd �}t||� t|j|j� t|||f�}tjd d d d d dgddddddgdd tfdtfgd �}t||� d S )N)r rH rH )r rH rH )rH r r )rH r r )r r r )r r r rX )r z|S3r )rE r )r r rH )r r rH )rH rH r )rH rH r )r r r )r r r rf )r' r rK r$ r& r% r rE )r( rY r) r+ r3 r5 r- r- r. � test_unnamed_and_named_fields� s4 z-TestStackArrays.test_unnamed_and_named_fieldsc C s� | j \}}}}tjdddgddtfdtfgd�}t||f�}tjdd dddgddtfdtfgd d d d d gd �}t||� t|j|j� t|||f�}ddtfdtfd tfg}tjdddddddg|dddddddgd �}t||� t|j|j� d S )N)r g $@g Y@)r g 4@g i@)�cg >@g �r@)r z|S3r rD )r )r r rH )r r rH )r r r )r r r )r rE �f3)r r rH rH )r r rH rH )r g $@g Y@rH )r g 4@g i@rH )rs g >@g �r@rH )rH rH rH r )rH rH rH r )r r r r )r r r r )r r r r ) r' r# r$ r% r rK r rE r& )r( rY r) r+ �zzr3 r5 r< r- r- r. �test_matching_named_fields� s. z*TestStackArrays.test_matching_named_fieldsc C s� | j \}}}}tjdddgddtfdtfgd�}dd d d d �}t||f|d �}tjdddddgddtfdtfgdddddgd�}t||� t|j |j � t|j|j� d S )N)r g $@g Y@)r g 4@g i@)rs g >@g �r@)r z|S3r rD )r z???g 8��g ���g �i��)r r rD �D)�defaults)r r g ���)r r g ���)r r r )r r r )r rE )r' r# r$ r% r rK r rE )r( rY r+ ru rx r3 r5 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� yt||fd d �}W n tk r� Y nX t �d S ) Nr r rD )r r rJ )r r r )rE r )r r rN )r )r r r T)Z autoconvertF) r& �boolr% rK r$ r r rE � TypeError�AssertionError)r( �adtyper �bdtyper r5 r3 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 )rs rD )r r rJ )r r r )rE r )r r rN )r )r r r )r& rz r% rK r$ r r rE )r( r} r r~ r r3 r5 r- r- r. �test_checktitles s z TestStackArrays.test_checktitlesN) rP rQ rR r/ rZ rq rr rv ry r r� r- r- r- r. ro � s ro c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )� 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 �< rj �n r r rs )r r � �A �K �d)r# r$ �list�zip�aranger& r r )r( r- r- r. r/ ' s zTestJoinBy.setUpc C sb | j | j }}td||dd�}tjdddddgdtfd tfd tfd tfd tfgd �}t||� d S )Nr �inner)�jointype)r �7 r� �i rj )rN �8 �B �j �e )� �9 �C �k �f )� �: �D �l �g )� �; �E �m �h �b1�b2rs r� )r )r r r r# r$ r& r )r( r r r3 r5 r- r- r. �test_inner_join/ s zTestJoinBy.test_inner_joinc C sN | j | j }}td||� tjdddddgdtfdtfd tfd tfgd � d S ) N)r r )r r� r� rj )rN r� r� r� )r� r� r� r� )r� r� r� r� )r� r� r� r� r r rs r� )r )r r r r# r$ r& )r( r r r- r- r. � test_join; s zTestJoinBy.test_joinc C s� | j | j }}td||d�}tjddddddd d d d d dddddddddgddddddddddddddddddddgdtfdtfdtfd tfgd �}t||� d S ) N)r r �outer)r r� rj rH )r �3 r� rH )r �4 r� rH )rJ �5 r� rH )r �6 r� rH )r r� r� rH )r r� rH rj )rN r� r� rH )rN r� rH r� )r� r� r� rH )r� r� rH r� )r� r� r� rH )r� r� rH r� )r� r� r� rH )r� r� rH r� )r �F rH r� )� �G rH r� )� �H rH r� )� �I rH r� )� �J rH r� )r r r r )r r r r r r rs r� )rE r )r r r rK r$ r& r )r( r r r3 r5 r- r- r. �test_outer_joinO s2 zTestJoinBy.test_outer_joinc C sz | j | j }}td||d�}tjddddddd d d d g d d d d d d d d d d g dtfdtfdtfdtfgd�}t||� d S )N)r r Z leftouter)r r� rj rH )r r� r� rH )r r� r� rH )rJ r� r� rH )r r� r� rH )r r� r� rH )rN r� r� rH )r� r� r� rH )r� r� r� rH )r� r� r� rH )r r r r r r rs r� )rE r )r r r rK r$ r& r )r( r r r3 r5 r- r- r. �test_leftouter_joink s z TestJoinBy.test_leftouter_joinc C sX t jddddgd�}t jddddgd�}tddg||dd d �}t|jjddd d g� d S ) NrJ )r �i4)r �f4)rs �u1)r rs r r� F)r� r\ �a1�a2)r# rT �onesr r r �names)r( r r �jr- r- r. �test_different_field_order| s z%TestJoinBy.test_different_field_orderc C sB t jddddgd�}t jddddgd�}tttdddg||� d S )NrJ )r r� )r r� )rs r� )r r r )r# rT r� r � ValueErrorr )r( r r r- r- r. �test_duplicate_keys� s z TestJoinBy.test_duplicate_keysN) rP rQ rR r/ r� r� r� 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� rj r� r r rs )r r� r� r� )r# r$ r� r� r� r& r r )�clsr- r- r. r/ � s zTestJoinBy2.setUpc C sp | j | j }}td||dddd�}tjdddd d d d d ddg dtfdtfdtfdtfdtfgd�}t||� d S )Nr r1 �2r� )� r1postfix� r2postfixr� )r r� r� rj rj )r r� r� r� r� )r r� r� r� r� )rJ r� r� r� r� )r r� r� r� r� )r r� r� r� r� )rN r� r� r� r� )r� r� r� r� r� )r� r� r� r� r� )r� r� r� r� r� r r� rs r� )r )r r r r# r$ r& r )r( r r r3 r5 r- r- r. �test_no_r1postfix� s z TestJoinBy2.test_no_r1postfixc C s | j ttd| j| jddd� d S )Nr r1 )r� r� )� assertRaisesr� r r r )r( r- r- r. �test_no_postfix� s zTestJoinBy2.test_no_postfixc C sp | j | j }}td||dddd�}tjdddd d d d d ddg dtfdtfdtfdtfdtfgd�}t||� d S )Nr �1r1 r� )r� r� r� )r r� r� rj rj )r r� r� r� r� )r r� r� r� r� )rJ r� r� r� r� )r r� r� r� r� )r r� r� r� r� )rN r� r� r� r� )r� r� r� r� r� )r� r� r� r� r� )r� r� r� r� r� r� r rs r� )r )r r r r# r$ r& r )r( r r r3 r5 r- r- r. �test_no_r2postfix� s z TestJoinBy2.test_no_r2postfixc C s6 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 jddddddddddg dtfd tfdtfdtfd tfd tfgd �}td dg||d d d d!�}t |j |j � t ||� d S )"Nr r� r r r� r� r �kr r rs )r r� r� r )r r r� r� r r )r� r r� r� r� r )r r r� r� r� r )r� r r� r� r� rJ )r r r� r� r� r )r� r r� r� r� r )r rJ r� r� � rN )r� rJ r� r� � r� )r r r� r� � r� )r� r r� r� � r� r� r� �c1�c2r� r� r� )r� r� r� ) r# r$ r� r� �tile�repeatr� r& r r r )r( r r r5 r3 r- r- r. �test_two_keys_two_vars� s" & & z"TestJoinBy2.test_two_keys_two_varsN) rP rQ rR � classmethodr/ 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� �dictr' )r( r� r- r- r. r/ � 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� g �?g @r r )r r r rD F)r' r\ N)r' r# r$ �objectr% r& r r )r( r� r) r* r3 r5 r- r- r. �test_append_to_objects� s z*TestAppendFieldsObj.test_append_to_objectsN)rP rQ rR �__doc__r/ r� r- r- r- r. r� � s r� �__main__))� __future__r r r �numpyr# �numpy.marK �numpy.ma.mrecordsr Znumpy.ma.testutilsr Z numpy.testingr r r r Znumpy.lib.recfunctionsr r r r r r r r r �libZ recfunctionsr; r> r2 r rS rW rh ro r� r� r� rP r- r- r- r. �<module> s, , 2 6 dA