o
    Eb                     @   s@   d dl mZmZ d dlmZ dd Zdd Zdd Zd	d
 ZdS )    )assert_array_equalassert_array_almost_equal)padec                  C   s^   t dgd\} }t| jdg t|jdg t dgdd\} }t| jdg t|jdg d S )N      ?r   )r   r   c)numpdenomp r	   C/usr/lib/python3/dist-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   )Zan_intZan_fltijZnump_intZ
denomp_intZnump_fltZ
denomp_fltr	   r	   r
   test_pade_intsB   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_complexT   s   .4":*r   N)	Znumpy.testingr   r   Zscipy.interpolater   r   r   r   r   r	   r	   r	   r
   <module>   s    
4