%PDF- %PDF-
| Direktori : /lib64/python3.9/site-packages/numpy/lib/tests/__pycache__/ |
| Current File : //lib64/python3.9/site-packages/numpy/lib/tests/__pycache__/test_shape_base.cpython-39.pyc |
a
z[yc�d � @ sz d dl Zd dlZd dlZd dlZd dlmZmZmZm Z m
Z
mZmZm
Z
mZmZmZmZmZmZ d dlmZmZmZmZmZ ejdkZdd� ZG dd� d�ZG d d
� d
�ZG dd� d�ZG 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 dd� d�Z(G dd � 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_warns� c s t �� �� fdd��}|S )z: hack in keepdims behavior into a function taking an axis c s. � | fd|i|��}|d u r d}t j||d�S )N�axisr �r )�npr
)�ar �kwargs�res��func� �E/usr/lib64/python3.9/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� ftjtjtdd�fg}|D ]`\}}}t
t|j��dg D ]@}||fd |i|��}||fd |i|��} t|t|| |d
�� q�qldS )z- Test it translates from arg<func> to <func> r ��rand� � � � )ZkthNr r )�numpy.randomr&