o
    i                     @   sH   d dl Zd dlmZ d dlmZmZ dd Zdd Zdd	 Z	d
d Z
dS )    N)pade)xp_assert_equalassert_array_almost_equalc                  C   sv   t dgd\} }t| jtdg t|jtdg t dgdd\} }t| jtdg t|jtdg d S )N      ?r   )r   r   cnpasarray)numpdenomp r   ^/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/scipy/interpolate/tests/test_pade.pytest_pade_trivial   s   r   c                  C   s  g d} t | d\}}t|jg d t|jdg t | d\}}t|jg d t|jddg t | d\}}t|jd	dg t|jg d
 t | d\}}t|jdg t|jg d t | dd\}}t|jg d t|jdg t | dd\}}t|jg d t|jddg t | dd\}}t|jd	dg t|jg d
 t | dd\}}t|jdg t|jg d t | dd\}}t|jg d t|jdg t | dd\}}t|jddg t|jddg t | dd\}}t|jdg t|jg d d S )N)r   r         ?UUUUUU?r   )r   r   r   r   r      )r   gUUUUUU?r   gUUUUUUտ   gUUUUUU?)r   gUUUUUUr      )gUUUUUUſr         r   )r   r   r   r   g      )r   r   r   )r   r   r   )anr	   r
   r   r   r   test_pade_4term_exp   sD   r   c                  C   s|   g d} g d}t dt| D ],}t dt| | D ] }t| ||\}}t|||\}}t|j|j t|j|j qqd S )N)r   r   r      )r   g       @g      @g      @r   )rangelenr   r   r   )an_intan_fltijnump_int
denomp_intnump_flt
denomp_fltr   r   r   test_pade_intsE   s   r!   c               	   C   sd  d} d| |  |    |   | d  | |   d   | d  |    d| |    d   | |   d   g}t|dd\}}t|j| |    dg t|j|   dg t|dd\}}t|j| d d|  |    dg t|j| |    dg t|dd\}}t|j| d | |     |   d  d| |     dg t|j|   d | d|     dg d S )Ny?333333?r   r   r   r   )	conjugater   r   r   )xr   r	   r
   r   r   r   test_pade_complexW   s    .4"0*r$   )numpyr   scipy.interpolater   scipy._lib._array_apir   r   r   r   r!   r$   r   r   r   r   <module>   s    
4