Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/numpy/lib/tests/__pycache__/test_shape_base.cpython-37.pyc
Назад
B �5�g�^ � @ sz d dl Zd dlZd dlZd dlZd dlmZmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZ d dlmZmZmZmZmZ ejdkZdd� Z G dd� d�Z G d d � 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$G dd� d�Z%G dd� d�Z&G dd � d �Z'G d d � d �Z(G d d � d �Z)G d!d"� d"�Z*G d#d$� d$�Z+G d%d&� d&�Z,d'd(� Z-dS ))� N)�apply_along_axis�apply_over_axes� array_split�split�hsplit�dsplit�vsplit�dstack� column_stack�kron�tile� expand_dims�take_along_axis�put_along_axis)�assert_� assert_equal�assert_array_equal� assert_raises� assert_warnsl c s t �� �� fdd��}|S )z: hack in keepdims behavior into a function taking an axis c s. � | fd|i|��}|d kr d}t j||d�S )N�axisr )r )�npr )�ar �kwargs�res)�func� �U/home/digitalm-up/venv/lib/python3.7/site-packages/numpy/lib/tests/test_shape_base.py�wrapped s z _add_keepdims.<locals>.wrapped)� functools�wraps)r r r )r r � _add_keepdims s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestTakeAlongAxisc C s� ddl m} |ddd�}tjtjt� fttj�ttj�t� fttj �ttj �t� ftj tj tdd�fg}xl|D ]d\}}}xXt t|j��dg D ]@}||fd |i|��}||fd |i|��} t|t|| |d �� q�W qnW dS ) z- Test it translates from arg<func> to <func> r )�rand� � � � )ZkthNr )r )� numpy.randomr" r �sortZargsort�dictr �min�argmin�max�argmax� partitionZ argpartition�list�range�ndimr r ) �selfr" r �funcsr Zargfuncr r Za_funcZai_funcr r r �test_argequivalent s z$TestTakeAlongAxis.test_argequivalentc C s� t �d�}t jdt jd�}t||dd� ttt|t �d�dd� ttt||�t �dd� ttt||�t �dd� tt j t||dd� dS )z4 Test it errors when indices has too few dimensions )� r5 )r5 r& )�dtype� )r r5 N) r �ones�intpr r � ValueError�array� IndexErrorZastype�bool�float� AxisError)r2 r �air r r � test_invalid1 s z TestTakeAlongAxis.test_invalidc C s: t �d�}t jdt jd�}t||dd�}t|j|j� dS )zC Test everything is ok with empty results, even with inserted dims )r# r$ r% )r# r r% )r6 r7 )r N)r r8 r9 r r �shape)r2 r r@ �actualr r r � test_emptyB s z TestTakeAlongAxis.test_emptyc C s8 t �d�}t jdt jd�}t||dd�}t|jd� dS )zD Test that non-indexing dimensions are broadcast in both directions )r# r$ r7 )r7 r& r% )r6 r7 )r )r# r& r% N)r r8 r9 r r rB )r2 r r@ rC r r r �test_broadcastJ s z TestTakeAlongAxis.test_broadcastN)�__name__� __module__� __qualname__r4 rA rD rE r r r r r! s r! c @ s e Zd Zdd� Zdd� ZdS )�TestPutAlongAxisc C s� t �dddgdddgg�}xbtt|j��d g D ]J}|�� }tt j�||d�}t||d|d� tt j �||d�}t ||� q0W d S ) Nr5 � � �< �( �2 )r i����) r r; r/ r0 r1 �copyr r- r r+ r )r2 Za_baser r Zi_maxZi_minr r r �test_replace_maxS s z!TestPutAlongAxis.test_replace_maxc C sL t �d�}t jdt jd��d�d }t||ddd� tt||dd�d� d S ) zD Test that non-indexing dimensions are broadcast in both directions )r# r$ r7 r5 )r6 )r7 r& r% r$ rK r7 )r N)r r8 �aranger9 �reshaper r r )r2 r r@ r r r rE c s z TestPutAlongAxis.test_broadcastN)rF rG rH rP rE r r r r rI R s rI c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej fd d �Z ej fd d�Z ej fdd�Z dd� Z dd� Zdd� Zdd� Zdd� ZdS ) �TestApplyAlongAxisc C s6 t �dd�}tttd|�t|�t �|jd � � d S )N)rK r5 �dr r7 )r r8 r r �lenrB )r2 r r r r � test_simplel s z TestApplyAlongAxis.test_simplec C s6 t �dd�}tttd|�t|�t �|jd � � d S )N)r5 �e rT r r7 )r r8 r r rU rB )r2 r r r r �test_simple101q s z!TestApplyAlongAxis.test_simple101c C s@ t �d��d�}ttt jd|�dddgdddgd d d gg� d S ) N� )r# r# r# r rJ �! �$ �' �* �- �0 �3 )r rQ rR r r �sum)r2 r r r r �test_3dv s zTestApplyAlongAxis.test_3dc C s� dd� }G dd� dt j�}t �ddgddgg��|�}t �ddgd d gg��|�}t|d|�}tt||�� t||� t|d|�}tt||�� t||� d S ) Nc S s | d S )Nr& r )�rowr r r �double| s z9TestApplyAlongAxis.test_preserve_subclass.<locals>.doublec @ s e Zd ZdS )z<TestApplyAlongAxis.test_preserve_subclass.<locals>.MyNDArrayN)rF rG rH r r r r � MyNDArray s re r r7 r& r# r$ � )r �ndarrayr; �viewr r � isinstancer )r2 rd re �m�expected�resultr r r �test_preserve_subclass{ s z)TestApplyAlongAxis.test_preserve_subclassc C sL G dd� dt j�}dd� }t �d��|�}tt|d|�t �dddg�� d S )Nc @ s e Zd ZdZdS )z9TestApplyAlongAxis.test_subclass.<locals>.MinimalSubclassr7 N)rF rG rH �datar r r r �MinimalSubclass� s ro c S s | j S )N)rn )r; r r r �minimal_function� s z:TestApplyAlongAxis.test_subclass.<locals>.minimal_function)rf r# r r7 )r rg �zerosrh r r r; )r2 ro rp r r r r � test_subclass� s z TestApplyAlongAxis.test_subclassc C sL t �d��|�}tt jd|�}tt||�� t|t �dddg��|�� d S )N)rf r# r rf ) r r8 rh r ra r ri r r; )r2 �clsr r r r r �test_scalar_array� s z$TestApplyAlongAxis.test_scalar_arrayc C s� dd� }t �d��|�}t|d|�}tt||�� t|t �dddg��|�� t|d|�}tt||�� t|t �ddddddg��|�� d S )Nc S s t | jd� t�tj| dd��S )z/ Sum x, returning a 0d array of the same class r7 T)Zkeepdims)r r1 r �squeezera )�xr r r � sum_to_0d� s z3TestApplyAlongAxis.test_0d_array.<locals>.sum_to_0d)rf r# r rf r7 r# )r r8 rh r r ri r r; )r2 rs rw r r r r r � test_0d_array� s z TestApplyAlongAxis.test_0d_arrayc s2 �fdd��t �d��d�� t�d� �}t j� �fdd�t� jd �D �d d ����}tt |�t |�� t||� t�d� �}t j� �fd d�t� jd �D �dd ����}tt |�t |�� t||� t �d ��d ��t�d��}t j��fdd�t�jd �D �d d ����}tt |�t |�� t||� d S )Nc s0 t | jd� | ddd� | dd�df �� �S )z/produces an asymmetric non-square matrix from xr7 N���)r r1 rh )rv )rs r r �f1to2� s z5TestApplyAlongAxis.test_axis_insertion.<locals>.f1to2� )rf r# r c s g | ]}�� d d �|f ��qS )Nr )�.0�i)�a2drz r r � <listcomp>� s z:TestApplyAlongAxis.test_axis_insertion.<locals>.<listcomp>r7 ry )r c s g | ]}�� |d d �f ��qS )Nr )r| r} )r~ rz r r r � s �Z )rf r% r# c s6 g | ].� t j��� fd d�t�jd �D �dd��qS )c s" g | ]}�� |d d ��f ��qS )Nr )r| r} )�a3drz �jr r r � s zETestApplyAlongAxis.test_axis_insertion.<locals>.<listcomp>.<listcomp>r )r )r �stackr0 rB )r| )r� rz )r� r r � s r& ) r rQ rR r r� r0 rB rh r �type)r2 rs rC rk r )r~ r� rs rz r �test_axis_insertion� s, z&TestApplyAlongAxis.test_axis_insertionc C s4 G dd� dt j�}| �|� | �|� | �|� d S )Nc @ s e Zd ZdS )zFTestApplyAlongAxis.test_subclass_preservation.<locals>.MinimalSubclassN)rF rG rH r r r r ro � s ro )r rg rt rx r� )r2 ro r r r �test_subclass_preservation� s z-TestApplyAlongAxis.test_subclass_preservationc C s� dd� }t �d��d�}t|d|�}tt|t jj�� t|j d� t |d d �d d �df j ||d d �df �j � t |d d �d d �df j ||d d �df �j � t |d d �d d �df j ||d d �df �j � d S ) Nc S s@ t | jd� | ddd� | dd�df }tj�|d dk|�S )z/produces an asymmetric non-square matrix from xr7 Nry r% r )r r1 r �maZ masked_where)rv r r r r rz � s z8TestApplyAlongAxis.test_axis_insertion_ma.<locals>.f1to2r{ )rf r# r r# r7 r& ) r rQ rR r r ri r� Z masked_arrayr r1 r �mask)r2 rz r r r r r �test_axis_insertion_ma� s 00z)TestApplyAlongAxis.test_axis_insertion_mac C sH dd� }t �|dt �ddgddgg��}t|t �ddgddgg�� d S )Nc S s | d | d fS )Nr7 r r )rv r r r � sample_1d� s z7TestApplyAlongAxis.test_tuple_func1d.<locals>.sample_1dr7 r& r# r$ )r r r; r )r2 r� r r r r �test_tuple_func1d� s z$TestApplyAlongAxis.test_tuple_func1dc C s| dd� }t �d�}ttt j|d|� ttt j|d|� dd� }t �d�}t �|d|�}t|t �d �� ttt j|d|� d S ) Nc S s t d� d S )NF)r )rv r r r � never_call� s z1TestApplyAlongAxis.test_empty.<locals>.never_call)r r r r7 c S s t t| �dk� dS )Nr r7 )r rU )rv r r r � empty_to_1� s z1TestApplyAlongAxis.test_empty.<locals>.empty_to_1)r5 r r5 )r �emptyr r: r r r8 )r2 r� r r� rC r r r rD � s z TestApplyAlongAxis.test_emptyc C s� t �ddhddhddhgddhd d hd d hgg�}t �d d� d|�}t �ddddhddd d hddd d hg�}t||� x.t �|j�D ] }tt|| �t|| �� q�W d S )Nr7 � r& � r# rZ r$ �, r% �7 rf �B c S s t j| � S )N)�set�union)r r r r �<lambda> � z>TestApplyAlongAxis.test_with_iterable_object.<locals>.<lambda>r )r r; r r ZndindexrB r� )r2 rT rC rk r} r r r �test_with_iterable_object s ( z,TestApplyAlongAxis.test_with_iterable_objectN)rF rG rH rV rX rb rm rr r rg rt rx r� r� r� r� rD r� r r r r rS k s %rS c @ s e Zd Zdd� ZdS )�TestApplyOverAxesc C sH t �d��ddd�}tt j|ddg�}t|t �dgdgdggg�� d S ) N� r& r# r$ r rL �\ �| )r rQ rR r ra r r; )r2 r Zaoa_ar r r rV s z TestApplyOverAxes.test_simpleN)rF rG rH rV r r r r r� s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestExpandDimsc C sV d}t �|�}xBtdd�D ]4}t||�}t|j| dk� tt �|�j|k� qW d S )N)r& r# r$ r% ���r$ r7 )r r� r0 r r rB ru )r2 �sr r �br r r �test_functionality s z!TestExpandDims.test_functionalityc C sn t �d�}t j|dd�jdks"t�t j|dd�jdks:t�t j|dd�jdksRt�t j|d d�jd ksjt�d S ) N)r# r# r# )r r7 r& )r )r7 r7 r7 r# r# r# )r ry ���)r7 r# r# r# r7 r7 )r r# r% )r7 r# r# r7 r# r7 )r ���r� )r7 r7 r# r7 r# r# )r r� r rB �AssertionError)r2 r r r r �test_axis_tuple! s z TestExpandDims.test_axis_tuplec C s\ d}t �|�}tt jt|d� tt jt|d� t �d�}tt jt|d� tt jt|d� d S )N)r& r# r$ r% i����r% )r# r# r# )r i����)r r% )r r� r r? r )r2 r� r r r r �test_axis_out_of_range( s z%TestExpandDims.test_axis_out_of_rangec C s t �d�}ttt|dd� d S )N)r# r# r# )r7 r7 )r )r r� r r: r )r2 r r r r �test_repeated_axis2 s z!TestExpandDims.test_repeated_axisc C sf t �d��d�}t jj||d dkd�}t j|dd�}tt|t jj�� t |j d� t |j j d� d S ) Nr5 )r& r% r# r )r� r7 )r )r& r7 r% ) r rQ rR r� r; r r ri Z MaskedArrayr rB r� )r2 r �expandedr r r �test_subclasses6 s z TestExpandDims.test_subclassesN)rF rG rH r� r� r� r� r� r r r r r� s r� c @ sf e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej j e d d �d d� �Z dd� Z dd� Zdd� ZdS )�TestArraySplitc C s t �d�}ttt|d� d S )Nr5 r )r rQ r r: r )r2 r r r r �test_integer_0_splitA s z#TestArraySplit.test_integer_0_splitc C s� t �d�}t|d�}t �d�g}t||� t|d�}t �d�t �dd�g}t||� t|d�}t �d�t �dd�t �dd�g}t||� t|d�}t �d�t �dd�t �dd �t �d d�g}t||� t|d�}t �d�t �dd�t �dd�t �dd �t �d d�g}t||� t|d�}t �d�t �dd�t �dd�t �dd �t �d d �t �d d�g}t||� t|d�}t �d�t �dd�t �dd�t �dd�t �dd �t �d d �t �d d�g}t||� t|d �}t �d�t �dd�t �dd�t �dd�t �dd�t �dd �t �d d �t �d d�g}t||� t|d �}t �d�t �dd�t �dd�t �dd�t �dd�t �dd�t �dd �t �d d �t �d d�g }t||� t|d�}t �d�t �dd�t �dd�t �dd�t �dd�t �dd�t �dd�t �dd �t �d d �t �d d�g }t||� t|d �}t �d�t �dd�t �dd�t �dd�t �dd�t �dd�t �dd�t �dd �t �d d �t �d d�t �g �g }t||� d S ) Nr5 r7 r& r% r# r$ � rf � � r� )r rQ r �compare_resultsr; )r2 r r �desiredr r r �test_integer_splitE sb " " z!TestArraySplit.test_integer_splitc C s� t �t �d�t �d�g�}t|ddd�}t �t �d�g�t �t �d�g�t �d�g}t||� t|jj|d jjk� t|dddgdd�}t �d�t �t �d�g�t �t �d�g�g}t||� t|jj|d jjk� d S ) Nr5 r# r )r )r r5 ry r7 r& ) r r; rQ r rq r� r r6 r� )r2 r r �tgtr r r �test_integer_split_2D_rows� s z)TestArraySplit.test_integer_split_2D_rowsc C s� t �t �d�t �d�g�}t|ddd�}t �t �d�t �d�g�t �t �dd�t �dd�g�t �t �dd�t �dd�g�g}t||� d S )Nr5 r# ry )r r$ r� )r r; rQ r r� )r2 r r r� r r r �test_integer_split_2D_cols� s z)TestArraySplit.test_integer_split_2D_colsc C sv t �t �d�t �d�g�}t|d�}t �t �d�g�t �t �d�g�t �d�g}t||� t|jj|d jjk� dS )z2 This will fail if we change default axis r5 r# )r r5 ry N) r r; rQ r rq r� r r6 r� )r2 r r r� r r r � test_integer_split_2D_default� s z,TestArraySplit.test_integer_split_2D_defaultzNeeds 64bit platform)�reasonc C s` t �dgd�}t|d�}t �dgd�}|gd }x*tt|��D ]}t|| j|| j� q>W d S )Nr )l r& r$ )i @r& )r Z broadcast_tor r0 rU r rB )r2 r r �chunkr� r} r r r �,test_integer_split_2D_rows_greater_max_int32� s z;TestArraySplit.test_integer_split_2D_rows_greater_max_int32c C s\ t �d�}dddg}t||dd�}t �dd�t �dd�t �dd�t �dd�g}t||� d S )Nr5 r7 r% r� ry )r r )r rQ r r� )r2 r �indicesr r� r r r �test_index_split_simple� s z&TestArraySplit.test_index_split_simplec C sZ t �d�}dddg}t||dd�}t �g �t �dd�t �dd�t �dd�g}t||� d S )Nr5 r r% r� ry )r )r rQ r r; r� )r2 r r� r r� r r r �test_index_split_low_bound� s z)TestArraySplit.test_index_split_low_boundc C sn t �d�}dddddg}t||dd�}t �g �t �dd�t �dd�t �dd�t �g �t �g �g}t||� d S )Nr5 r r% r� � ry )r )r rQ r r; r� )r2 r r� r r� r r r �test_index_split_high_bound� s z*TestArraySplit.test_index_split_high_boundN)rF rG rH r� r� r� r� r� �pytest�markZskipif�IS_64BITr� r� r� r� r r r r r� @ s = r� c @ s e Zd Zdd� Zdd� ZdS )� TestSplitc C s8 t �d�}t|d�}t �d�t �dd�g}t||� d S )Nr5 r& r% )r rQ r r� )r2 r r r� r r r �test_equal_split� s zTestSplit.test_equal_splitc C s t �d�}ttt|d� d S )Nr5 r# )r rQ r r: r )r2 r r r r �test_unequal_split� s z TestSplit.test_unequal_splitN)rF rG rH r� r� r r r r r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestColumnStackc C s t ttd� d S )Nr7 )r � TypeErrorr )r2 r r r �test_non_iterable� s z!TestColumnStack.test_non_iterablec C sL t �d�}t �d�}t �ddgddgddgg�}t �||f�}t||� d S )N)r7 r& r# )r& r# r$ r7 r& r# r$ )r r; r r )r2 r r� rk rC r r r �test_1D_arrays� s z TestColumnStack.test_1D_arraysc C sd t �dgdgdgg�}t �dgdgdgg�}t �ddgddgddgg�}t �||f�}t||� d S )Nr7 r& r# r$ )r r; r r )r2 r r� rk rC r r r �test_2D_arrays� s z TestColumnStack.test_2D_arraysc C s. t t�� tdd� td�D �� W d Q R X d S )Nc s s | ]}t �d �V qdS )r# N)r rQ )r| �_r r r � <genexpr>� s z1TestColumnStack.test_generator.<locals>.<genexpr>r& )r � FutureWarningr r0 )r2 r r r �test_generator� s z TestColumnStack.test_generatorN)rF rG rH r� r� r� r� r r r r r� � s r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )� TestDstackc C s t ttd� d S )Nr7 )r r� r )r2 r r r r� � s z TestDstack.test_non_iterablec C s@ t �d�}t �d�}t||g�}t �ddggg�}t||� d S )Nr7 r& )r r; r r )r2 r r� r r� r r r � test_0D_array� s zTestDstack.test_0D_arrayc C sD t �dg�}t �dg�}t||g�}t �ddggg�}t||� d S )Nr7 r& )r r; r r )r2 r r� r r� r r r � test_1D_array� s zTestDstack.test_1D_arrayc C sX t �dgdgg�}t �dgdgg�}t||g�}t �ddggddggg�}t||� d S )Nr7 r& )r r; r r )r2 r r� r r� r r r � test_2D_array s zTestDstack.test_2D_arrayc C sN t �ddg�}t �ddg�}t||g�}t �ddgddggg�}t||� d S )Nr7 r& )r r; r r )r2 r r� r r� r r r �test_2D_array2 s zTestDstack.test_2D_array2c C s. t t�� tdd� td�D �� W d Q R X d S )Nc s s | ]}t �d �V qdS )r# N)r rQ )r| r� r r r r� s z,TestDstack.test_generator.<locals>.<genexpr>r& )r r� r r0 )r2 r r r r� s zTestDstack.test_generatorN) rF rG rH r� r� r� r� r� r� r r r r r� � s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � TestHsplitz&Only testing for integer splits. c C s t ttdd� d S )Nr7 )r r: r )r2 r r r r� s z TestHsplit.test_non_iterablec C s: t �d�}yt|d� td� W n tk r4 Y nX d S )Nr7 r& r )r r; r r r: )r2 r r r r r� ! s zTestHsplit.test_0D_arrayc C sF t �ddddg�}t|d�}t �ddg�t �ddg�g}t||� d S )Nr7 r& r# r$ )r r; r r� )r2 r r r� r r r r� ) s zTestHsplit.test_1D_arrayc C sb t �ddddgddddgg�}t|d�}t �ddgddgg�t �ddgddgg�g}t||� d S )Nr7 r& r# r$ )r r; r r� )r2 r r r� r r r r� / s ,zTestHsplit.test_2D_arrayN)rF rG rH �__doc__r� r� r� r� r r r r r� s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � TestVsplitz&Only testing for integer splits. c C s t ttdd� d S )Nr7 )r r: r )r2 r r r r� ; s z TestVsplit.test_non_iterablec C s t �d�}ttt|d� d S )Nr7 r& )r r; r r: r )r2 r r r r r� > s zTestVsplit.test_0D_arrayc C sB t �ddddg�}yt|d� td� W n tk r< Y nX d S )Nr7 r& r# r$ r )r r; r r r: )r2 r r r r r� B s zTestVsplit.test_1D_arrayc C s^ t �ddddgddddgg�}t|d�}t �ddddgg�t �ddddgg�g}t||� d S )Nr7 r& r# r$ )r r; r r� )r2 r r r� r r r r� J s (zTestVsplit.test_2D_arrayN)rF rG rH r� r� r� r� r� r r r r r� 7 s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TestDsplitc C s t ttdd� d S )Nr7 )r r: r )r2 r r r r� T s z TestDsplit.test_non_iterablec C s t �d�}ttt|d� d S )Nr7 r& )r r; r r: r )r2 r r r r r� W s zTestDsplit.test_0D_arrayc C s$ t �ddddg�}ttt|d� d S )Nr7 r& r# r$ )r r; r r: r )r2 r r r r r� [ s zTestDsplit.test_1D_arrayc C sN t �ddddgddddgg�}yt|d� td� W n tk rH Y nX d S )Nr7 r& r# r$ r )r r; r r r: )r2 r r r r r� _ s zTestDsplit.test_2D_arrayc C s� t �ddddgddddggddddgddddggg�}t|d�}t �ddgddggddgddggg�t �ddgddggddgddggg�g}t||� d S )Nr7 r& r# r$ )r r; r r� )r2 r r r� r r r � test_3D_arrayh s $(zTestDsplit.test_3D_arrayN)rF rG rH r� r� r� r� r� r r r r r� R s r� c @ s e Zd Zdd� ZdS )� TestSqueezec C s� ddl m} |ddddd�}|ddddd�}|dddd�}tt�|�t�|d�� tt�|�t�|d�� tt�|�t�|d�� d ggg}t�|�}t|d � t|jd� tt|�tj � d S ) Nr )r" rK r5 r7 )rK r5 r5 )rK r5 rK )rK r5 g �?) r' r" r r ru rR r r1 r� rg )r2 r" r r� �cr r r r � test_basict s zTestSqueeze.test_basicN)rF rG rH r� r r r r r� s s r� c @ s e Zd Zdd� ZdS )�TestKronc C s� G dd� dt j�}t �ddg�}||j|j|j�}ttt||��t j� ttt||��|� ttt||��t j� ttt||��|� d S )Nc @ s e Zd ZdZdS )z*TestKron.test_return_type.<locals>.myarrayg N)rF rG rH Z__array_priority__r r r r �myarray� s r� r& ) r rg r8 rB r6 rn r r� r )r2 r� r r� r r r �test_return_type� s zTestKron.test_return_typeN)rF rG rH r� r r r r r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestTilec C s t �dddg�}ddgddgg}tt|d�ddddddg� tt|d�ddddddgddddddgg� tt|d�ddddddgg� tt|d�ddddgddddgg� tt|d�ddgddgddgddgg� tt|d�ddddgddddgddddgddddgg� d S ) Nr r7 r& r# r$ )r& r& )r7 r& )r& r7 )r r; r r )r2 r r� r r r r� � s , $( zTestTile.test_basicc C s0 t �d�}t|d�}|d7 }t|t �d�� d S )Nr% r7 r& )r rQ r r )r2 r r� r r r �(test_tile_one_repetition_on_array_gh4679� s z1TestTile.test_tile_one_repetition_on_array_gh4679c C sL t �g gg�}t �g g g�}t|d�j}t|d�j}t|d� t|d� d S )Nr& )r# r& r% )r& r )r# r& r )r r; r rB r )r2 r r� r� rT r r r rD � s zTestTile.test_emptyc C s� ddl m} ddddddg}d d d d d dg}xT|D ]L}|dd|d�}x8|D ]0}t�||j�}t||�}t||�} t|| � qJW q2W d S )Nr )�randint)r& )r7 r& )r& r7 )r& r& )r& r# r& )r# r& )r# )r& r# )r# r$ r# )r# r& r# )r$ r# r& r$ r5 )�size)r' r� r r8 r6 r r r ) r2 r� ZrepsrB r� r� �rr ZlargeZklarger r r �test_kroncompare� s zTestTile.test_kroncompareN)rF rG rH r� r� rD r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�TestMayShareMemoryc C s� t �d�}t �d�}tt �||�� tt �||d d d� �� tt �||d d d� �� tt �||dd �d d d�f �� tt �|d d d� |� � tt �|d d d� |� � tt �|dd �d d d�f |� � tt �|dd �d d d�f |�� d S )N)rN rL )rJ rL rf ry r& r7 )r r8 r Zmay_share_memory)r2 rT Zd2r r r r� � s " $z TestMayShareMemory.test_basicN)rF rG rH r� r r r r r� � s r� c C s, x&t t|��D ]}t| | || � qW d S )N)r0 rU r )r r� r} r r r r� � s r� ).�numpyr r �sysr� Znumpy.lib.shape_baser r r r r r r r r r r r r r Z numpy.testingr r r r r �maxsizer� r r! rI rS r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r �<module> s4 @ 4 '( ' ! (
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка