o
    ×?Hh  ã                   @   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   úa/home/air/sanwanet/gpt-api/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   gUUUUUUå¿r   é   )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