← Back
Editing: test_polyint.cpython-311.pyc
� d�cTv � �f � d dl Z d dlZd dlZd dlmZmZmZmZm Z m Z d dlmZ d dlZd dlmZmZmZmZmZmZmZmZmZmZmZmZ dd i fd�Zg d�Zd� Zd� Zd � Zd � Z G d� d� � Z! G d � d� � Z" G d� d� � Z# G d� d� � Z$ G d� d� � Z%d� Z&d� Z'd� Z( G d� d� � Z)dS )� N)�assert_almost_equal�assert_array_equal�assert_array_almost_equal�assert_allclose�assert_equal�assert_)�raises)�KroghInterpolator�krogh_interpolate�BarycentricInterpolator�barycentric_interpolate�approximate_taylor_polynomial�CubicHermiteSpline�pchip�PchipInterpolator�pchip_interpolate�Akima1DInterpolator�CubicSpline�make_interp_splinec �H � t j � d� � g d�}t t dt |� � dz � � � � }|� |t |� � dz z d� � t j j d|z � � |� � }t j |� � } | t u r@t j j d|z � � |� � } | ||| fd|i|��| � � }n | ||fd|i|��| � � }|pd|j d |� z |z |j |d � dd � z }t |j |� � |r�|j dk r�| t u rK | t |� � t |� � t | � � fd|i|��t | � � � � n< | t |� � t |� � fd|i|��t | � � � � | j dk r�|��|j d |� dt |� � z z |j |d � dd � z } |t d � � f||j z z dz }|� | � � }t j ||� � \ }}t% ||� � d S d S d S ) N�� )���r � � � � r r �� �axis� �r )�np�random�seed�list�range�len�insert�rand� transpose�zerosr �shaper �size�slice�ndim�reshape�broadcast_arraysr )�interpolator_cls�x_shape�y_shape�deriv_shaper � extra_args�x�s�y�xi�dydx�yi�target_shape�bs_shape�yvs �F/usr/lib/python3/dist-packages/scipy/interpolate/tests/test_polyint.py�check_shaperA s� � ��I�N�N�4�������A��U�1�c�'�l�l�1�n� %� %�&�&�A��H�H�T�S��\�\�!�^� $�a�(�(�(� � �����)�3�3�A�6�6�A� ��'� � �B��-�-�-��y�~��w��0�:�:�1�=�=�� B� � �a��D� B� B�t� B�z� B� B�2� F� F��� <� � �a�� <� <�� <�� <� <�R� @� @�� �&�B�!�'�%�4�%�.�8��� !������q�r�r� 2�3�L����<�(�(�(� � 5�1�6�A�:�:��1�1�1� +���T�!�W�W�d�1�g�g�t�D�z�z� +� +�� +�)� +� +�+/��8�8� 5� 5� 5� 5� +���T�!�W�W�d�1�g�g� +� +�D� +�)� +� +�+/��8�8� 5� 5� 5� �w��{�{�{�*��7�5�D�5�>�D��W���$5�5�������q�r�r�8J�J�� ��t��������/�4�7� 8�� �Z�Z�� !� !���#�B��+�+���A���A������ �{�*�*� )r �r r! )r r � c �\ � d� } t t t t t t | fD ]�}t D ]v}t D ]l}t t |� � t |� � � � D ]>}|t k rt |||d |� � �!dD ]}d|i}t |||d ||� � ��?�m�w��d S )Nc �&