File: //lib/python3/dist-packages/numpy/lib/tests/__pycache__/test_stride_tricks.cpython-310.pyc
o
6��aAY � @ sB d dl Zd dlmZ d dlmZmZmZmZm Z m
Z
d dlmZm
Z
mZmZmZmZ d dlZdd� Zdd� Zd8d
d�Zdd
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d!� Z d"d#� Z!d$d%� Z"d&d'� Z#G d(d)� d)�Z$d*d+� Z%G d,d-� d-ej&�Z'G d.d/� d/e'�Z(d0d1� Z)d2d3� Z*d4d5� Z+d6d7� Z,dS )9� N��rational)�assert_equal�assert_array_equal�
assert_raises�assert_�assert_raises_regex�assert_warns)�
as_strided�broadcast_arrays�_broadcast_shape�broadcast_to�broadcast_shapes�sliding_window_viewc C s@ dd� | D �}t |� }dd� |D �}|gt|� }t||� d S )Nc S � g | ]}t �|��qS � ��np�zeros��.0�sr r �D/usr/lib/python3/dist-packages/numpy/lib/tests/test_stride_tricks.py�
<listcomp> � z)assert_shapes_correct.<locals>.<listcomp>c S s g | ]}|j �qS r )�shape)r �ar r r r s )r �lenr )�input_shapes�expected_shape�inarrays� outarrays� outshapes�expectedr r r �assert_shapes_correct s
r$ c C s$ dd� | D �}t ttg|�R � d S )Nc S r r r r r r r r r z4assert_incompatible_shapes_raise.<locals>.<listcomp>)r �
ValueErrorr )r r r r r � assert_incompatible_shapes_raise s r&