Viewing: test_arrayprint.cpython-37.pyc
B ��Fd�' � @ s� d dl mZmZmZ d dlZd dlZd dlmZm Z m Z m Z G dd� de �Z G dd� de�ZG dd � d e�ZG d d � d �Zd d � Zedkr�e � dS )� )�division�absolute_import�print_functionN)�TestCase�run_module_suite�assert_� assert_equalc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestArrayReprc C s$ t �t jt jg�}tt|�d� d S )Nzarray([ nan, inf]))�np�array�nan�infr �repr)�self�x� r ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/core/tests/test_arrayprint.py� test_nan_inf s zTestArrayRepr.test_nan_infc C s� G dd� dt j�}t �ddg��|�}tt|�d� t �ddgddgg��|�}tt|�d� t jd d gd ��|�}tt|�d � d S ) Nc @ s e Zd ZdS )z(TestArrayRepr.test_subclass.<locals>.subN)�__name__� __module__� __qualname__r r r r �sub s r � � z sub([1, 2])� � zsub([[1, 2], [3, 4]]))r r )�a�i4)�dtypez@sub([[(1,), (1,)], [(1,), (1,)]], dtype=[('a', '<i4')]))r �ndarrayr �viewr r �ones)r r Zx1dZx2dZxstructr r r � test_subclass s zTestArrayRepr.test_subclassc C sz t �d �}||d< tt|�d� t �d d g�}||d< tt|�d� t �d �}t �d �}||d< ||d< tt|�d� d S )Nr z-array(array(..., dtype=object), dtype=object)r z5array([None, array(..., dtype=object)], dtype=object)zBarray(array(array(..., dtype=object), dtype=object), dtype=object))r r r r )r Zarr0d�arr1d�first�secondr r r �test_self_containing% s z"TestArrayRepr.test_self_containingc C s6 t �d d g�}ddg|d< dg|d< tt|�d� d S )Nr r r r z.array([list([1, 2]), list([3])], dtype=object))r r r r )r r# r r r �test_containing_list7 s z"TestArrayRepr.test_containing_listN)r r r r r" r& r' r r r r r s r c @ s e Zd Zdd� ZdS )�TestComplexArrayc l sR dddt jt j t jg��fdd��D �}t jt jt jg� � fdd�|D �}dddd d d d d d d d dddddddddddddddddddddd d d d d d d d!d!d"d#d#d$d%d%d&d'd'd(d)d)d*d+d+d,d-d-d.d/d/d0d1d1d1d2d2d2d3d3d3d4d4d5d6d6d7d8d8d9d:d:d:d;d;d;d<d<d<d=d=d>d?d?d@dAdAdBdCdCdCdDdDdDdEdEdEgl}x$t||�D ]\}}t||k� �q4W d S )FNr r ���c s g | ]}� D ]}t ||��q qS r )�complex)�.0Zrp�ip)�rvalsr r � <listcomp>C s z-TestComplexArray.test_str.<locals>.<listcomp>c s( g | ] }� D ]}t t�|g|���q qS r )�strr r )r+ �c�dt)�dtypesr r r. E s z [ 0.+0.j]z [ 0.0+0.0j]z [ 0.+1.j]z [ 0.0+1.0j]z [ 0.-1.j]z [ 0.0-1.0j]z [ 0.+infj]z [ 0.0+infj]z [ 0.-infj]z [ 0.0-infj]z [ 0.+nanj]z [ 0.0+nanj]z [ 1.+0.j]z [ 1.0+0.0j]z [ 1.+1.j]z [ 1.0+1.0j]z [ 1.-1.j]z [ 1.0-1.0j]z [ 1.+infj]z [ 1.0+infj]z [ 1.-infj]z [ 1.0-infj]z [ 1.+nanj]z [ 1.0+nanj]z [-1.+0.j]z [-1.0+0.0j]z [-1.+1.j]z [-1.0+1.0j]z [-1.-1.j]z [-1.0-1.0j]z [-1.+infj]z [-1.0+infj]z [-1.-infj]z [-1.0-infj]z [-1.+nanj]z [-1.0+nanj]z [ inf+0.j]z [ inf+0.0j]z [ inf+1.j]z [ inf+1.0j]z [ inf-1.j]z [ inf-1.0j]z [ inf+infj]z [ inf-infj]z [ inf+nanj]z [-inf+0.j]z [-inf+0.0j]z [-inf+1.j]z [-inf+1.0j]z [-inf-1.j]z [-inf-1.0j]z [-inf+infj]z [-inf-infj]z [-inf+nanj]z [ nan+0.j]z [ nan+0.0j]z [ nan+1.j]z [ nan+1.0j]z [ nan-1.j]z [ nan-1.0j]z [ nan+infj]z [ nan-infj]z [ nan+nanj])r r r � complex64�cdouble� clongdouble�zipr )r Zcvals�actual�wanted�res�valr )r2 r- r �test_strA sT zTestComplexArray.test_strN)r r r r; r r r r r( @ s r( c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestArray2Stringc C s6 t �d�}tt �|�dk� tt j|dd�dk� dS )zBasic test of array2string.r z[0 1 2]r )�max_line_widthz[0 1 2]N)r �aranger � array2string)r r r r r � test_basicp s zTestArray2String.test_basicc C s( t jt �d�dd� d�}t|dk� dS )z0This should only apply to 0-D arrays. See #1218.g �?c S s dt | � S )NzValue in 0-D array: )r/ )r r r r �<lambda>y � z5TestArray2String.test_style_keyword.<locals>.<lambda>)�stylezValue in 0-D array: 1.5N)r r? r r )r Zstylestrr r r �test_style_keywordv s z#TestArray2String.test_style_keywordc C s2 dd� }t �d�}tjd dkr*d}d}nd}d}tt j|d |id �d k� tt j|d |id �d k� tt j|d d d� id �dk� tt j|ddd� id �|� tt j|ddd� id �|� t �d�}tt j|ddd� id �dk� tt j|ddd� id �dk� t �ddg�}tt j|dd d� id �d k� d S ) z6Test custom format function for each element in array.c S s, t �| �dk rdS t �| �dk r$dS dS d S )Nr �.r �o�O)r �abs)r r r r �_format_function~ s z?TestArray2String.test_format_function.<locals>._format_functionr r z [0x0 0x1 0x2]z [0o0 0o1 0o2]z[0x0L 0x1L 0x2L]z [0L 01L 02L]�all)� formatterz[. o O]�int_kindc S s d| S )Nz%.4fr )r r r r rA � rB z7TestArray2String.test_format_function.<locals>.<lambda>z[0.0000 1.0000 2.0000]�intc S s t | �S )N)�hex)r r r r rA � rB c S s t | �S )N)�oct)r r r r rA � rB g @� float_kindc S s d| S )Nz%.2fr )r r r r rA � rB z[0.00 1.00 2.00]�floatc S s d| S )Nz%.2fr )r r r r rA � rB �abc�def�numpystrc S s | d S )Nr r )�sr r r rA � rB z[abcabc defdef]N)r r> �sys� version_infor r? r r )r rI r Zx_hexZx_octrU r r r �test_format_function| s2 z%TestArray2String.test_format_functionc C s t �dt jdfdt jdfg�}t jddg|d�}tt �|�d� t jd d gd �}|d d � �t � d �� tt �|�d� t jddgfddgfgdgd�}tt �|�d� t jddgddggfddgddggfgdgd�}tt �|�d� t jddd�}tt �|�d � d S ) N�name� Zgrades)r )ZSarah)g @g @)ZJohn)g @g @)r z,[('Sarah', [ 8., 7.]) ('John', [ 6., 7.])]� )�AzM8[s])�shaper � �NaTz�[('1970-01-01T00:00:00',) ('1970-01-01T00:00:00',) ('1970-01-01T00:00:00',) ('1970-01-01T00:00:00',) ('1970-01-01T00:00:00',) ('NaT',) ('NaT',) ('NaT',) ('NaT',) ('NaT',)]r r) �{ )�Br r z [([ 1, -1],) ([123, 1],)]r r r � )ra r )r r z1[([[ 0, 1], [ 2, 3]],) ([[12, 0], [ 0, 0]],)])g �?g�7��� @g @zf8,f8,f8z( 1., 2.12345679, 3.)) r r �str_�float64r r r? �zeros�fill� datetime64)r r1 r r\ Z struct_intZ struct_2dintZ array_scalarr r r �test_structure_format� s$ & z&TestArray2String.test_structure_formatN)r r r r@ rD rX rh r r r r r< o s &r< c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )�TestPrintOptionsz.Test getting and setting global print options.c C s t �� | _d S )N)r �get_printoptions�oldopts)r r r r �setUp� s zTestPrintOptions.setUpc C s t jf | j� d S )N)r �set_printoptionsrk )r r r r �tearDown� s zTestPrintOptions.tearDownc C s< t �dddg�}tt|�d� t jdd� tt|�d� d S )Ng �?r gރB��?z.array([ 1.5 , 0. , 1.23456789])r )� precisionz"array([ 1.5 , 0. , 1.2346]))r r r r rm )r r r r r r@ � s zTestPrintOptions.test_basicc C s� t jdd� x�dgdfdgdfdgdfdgd fd gd fd gd fdgdfdgdfdddgdfdgdfdgdftdd�gdff D ]$\}}t �|�}tt|�d| � q|W d S ) Nr )ro g z 0.g333333�?g333333ӿz-0.gffffff�?z 1.g �?z 2.g ��z-2.g�G�z�.�z-15.g Y@z 100.g�������?r) gq= ף�^@z 0., -1., 123.�0i�z-12gffffff�z 0.-1.jz array([%s]))r rm r* r r r )r �values�stringr r r r �test_precision_zero� s * z$TestPrintOptions.test_precision_zeroc C s0 t �d�}t jddd� id� tt|�d� d S )Nr rJ c S s t | d �S )Nr )r/ )r r r r rA � rB z1TestPrintOptions.test_formatter.<locals>.<lambda>)rK zarray([-1, 0, 1]))r r> rm r r )r r r r r �test_formatter� s z TestPrintOptions.test_formatterc C s t �d�}t jddd� id� tt|�d� t jdd id� tt|�d� t jdd d� id� tt|�d� t jdd id� tt|�d� t jdd d� id� tt|�d� t jd d id� tt|�d� t �d �}t jd dd� id� tt|�d� t jdd id� tt|�d� d S )Nr rJ c S s t | d �S )Nr )r/ )r r r r rA � rB z7TestPrintOptions.test_formatter_reset.<locals>.<lambda>)rK zarray([-1, 0, 1])rM zarray([0, 1, 2])c S s t | d �S )Nr )r/ )r r r r rA � rB c S s t | d �S )Nr )r/ )r r r r rA � rB rL g @rQ c S s t | d �S )Nr )r/ )r r r r rA � rB zarray([-1.0, 0.0, 1.0])rP zarray([ 0., 1., 2.]))r r> rm r r )r r r r r �test_formatter_reset� s$ z%TestPrintOptions.test_formatter_resetN) r r r �__doc__rl rn r@ rs rt ru r r r r ri � s ri c C sB dd l } | jd dkr d}nd}tjdgtd�}tt|�|� d S )Nr r u array(['é'], dtype=object)z array([u'\xe9'], dtype=object)� é)r )rV rW r r �objectr r )rV �expectedr r r r �test_unicode_object_array� s rz �__main__)� __future__r r r rV �numpyr Z numpy.testingr r r r rx r r( r< ri rz r r r r r �<module> s 5/Q5
Return