o
    Eb5                     @   s   d dl Z d dlZd dlZd dlmZmZmZmZmZm	Z	m
Z
mZ d dlmZmZmZmZ d dlZd dlmZmZmZmZ d dlmZ d dlZd dlZd dlmZ d dlm  mZ  ddd	Z!d
d Z"G dd dZ#G dd dZ$G dd dZ%G dd dZ&dS )    N)sqrtcossinarctanexplogpiInf)assert_assert_allcloseassert_array_lessassert_almost_equal)quaddblquadtplquadnquad)LowLevelCallable)sine_ctypes+i)+P>c                 C   s2   | \}}t |||dd |d urt|| d S d S )Nr   )atolZrtol)r   r   )Zvalue_and_errZtabled_valueerrTolvalueerr r   E/usr/lib/python3/dist-packages/scipy/integrate/tests/test_quadpack.pyassert_quad   s
   r   c                 G   s$   t t| }t|tj|g|R  S N)getattr	clib_testctypescastZ	CFUNCTYPE)namerestypeargtypesZptrr   r   r   get_clib_test_routine   s   
r$   c                   @   ,   e Zd Zdd Zdd Zdd Zdd Zd	S )
TestCtypesQuadc              	   C   s   t jdkr	dg}nt jdkrdg}nddg}|D ]}z	t|| _W  n ty,   Y qw td tj}tjf}dD ]}t	| j|}||_
||_q;d S )	Nwin32zapi-ms-win-crt-math-l1-1-0.dlldarwinz
libm.dylibzlibm.soz	libm.so.6zCtypes can't import libm.so)r   r   tan)sysplatformr   ZCDLLlibOSErrorpytestskipc_doubler   r"   r#   )selffilesfiler"   r#   r!   funcr   r   r   setup_method    s(   


zTestCtypesQuad.setup_methodc                 C   sp   t t| jjddttjddd  t t| jjddttjddd  t t| jjddttjddd  d S )Nr         )r   r   r,   r   mathr   r)   r1   r   r   r   test_typical:   s   $$(zTestCtypesQuad.test_typicalc                 C   s   t ttdd d S Nr   r7   )r   r   r   r9   r   r   r   test_ctypes_sine?      zTestCtypesQuad.test_ctypes_sinec                 C   s   t dtjtjtj}t dtjtjttjtj}t dtjtj}t dtjtjttj}t dtjtjtj}|||||g}||g}|g}t|D ]!\}	}
t|
}|
|v r`t	t
t|dt qJtt|dtd d qJt|D ]\}	}
|
|v rtt|
dtd d qpt	t
t|
dt qpd S )NZ_sin_0Z_sin_1Z_sin_2Z_sin_3r          @)r$   r   r0   Zc_void_pc_intZPOINTER	enumerater   r.   Zraises
ValueErrorr   r   r   )r1   Zsin_0Zsin_1Zsin_2Zsin_3Zsin_4Zall_sigsZlegacy_sigsZlegacy_only_sigsjr4   callbackr   r   r   test_ctypes_variantsB   s:   z#TestCtypesQuad.test_ctypes_variantsN)__name__
__module____qualname__r5   r:   r<   rD   r   r   r   r   r&      s
    r&   c                   @   r%   )
TestMultivariateCtypesQuadc                 C   s>   t j}t jt jf}dD ]}t||g|R  }t| || qd S )N)_multivariate_typical_multivariate_indefinite_multivariate_sin)r   r0   r?   r$   setattr)r1   r"   r#   r!   r4   r   r   r   r5   h   s   z'TestMultivariateCtypesQuad.setup_methodc                 C   s   t t| jdtdd d S )Nr      g?ۗ?)r   r   rI   r   r9   r   r   r   r:   p   s   z'TestMultivariateCtypesQuad.test_typicalc                 C   s   t t| jdtd d S )Nr   ox?)r   r   rJ   r	   r9   r   r   r   test_indefiniteu   s   z*TestMultivariateCtypesQuad.test_indefinitec                    s"    fdd}t t|ddd d S )Nc                    s   | t  jddd  S r;   )r   rK   )yr9   r   r   threadsafety|      zBTestMultivariateCtypesQuad.test_threadsafety.<locals>.threadsafetyr   r7   gtJ׵?)r   r   )r1   rS   r   r9   r   test_threadsafetyz   s   z,TestMultivariateCtypesQuad.test_threadsafetyN)rE   rF   rG   r5   r:   rQ   rU   r   r   r   r   rH   g   s
    rH   c                   @   s   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd  Zd!S )"TestQuadc                 C   s    dd }t t|dtdd d S )Nc                 S   s   t ||  |t|   t S r   )r   r   r   )xnzr   r   r   myfunc      z%TestQuad.test_typical.<locals>.myfuncr   rM   rO   )r   r   r   r1   rZ   r   r   r   r:      s   zTestQuad.test_typicalc                 C   s   dd }t t|dtd d S )Nc                 S   s   t |   t|  S r   )r   r   rW   r   r   r   rZ      r=   z(TestQuad.test_indefinite.<locals>.myfuncr   rP   r   r   r	   r\   r   r   r   rQ      s   zTestQuad.test_indefinitec                 C   s>   dd }t t|ddddgddtd td	 td
  d S )Nc                 S   sD   d|   k rdk rt | S  d|   krdkr n dS t|  S dS )Nr         @      @g        )r   r   r]   r   r   r   rZ      s   
z&TestQuad.test_singular.<locals>.myfuncr   
   r_   r`   )pointsr7   g      g      )r   r   r   r   r\   r   r   r   test_singular   s   zTestQuad.test_singularc              	   C   sV   dd }d}t t|dddd|ddt| |t|  |td	  d
|d    d S )Nc                 S   s   t || d  S Nr7   r   rW   ar   r   r   rZ         z2TestQuad.test_sine_weighted_finite.<locals>.myfuncglSS%@r   r7      r   argsweightwvarii  rN   )r   r   r   r   r   )r1   rZ   omer   r   r   test_sine_weighted_finite   s
   .z"TestQuad.test_sine_weighted_finitec              	   C   s>   dd }d}d}t t|dt|d|d||d |d    d S )	Nc                 S   s   t |  | S r   re   rf   r   r   r   rZ      s   z4TestQuad.test_sine_weighted_infinite.<locals>.myfunc      @      @r   r   rj   rN   r^   r1   rZ   rg   rn   r   r   r   test_sine_weighted_infinite   s   z$TestQuad.test_sine_weighted_infinitec              	   C   s@   dd }d}d}t t|t d|d|d||d |d    d S )	Nc                 S   s   t | | S r   re   rf   r   r   r   rZ         z6TestQuad.test_cosine_weighted_infinite.<locals>.myfuncr_   gffffff@r   r   rj   rN   r^   rr   r   r   r   test_cosine_weighted_infinite   s   z&TestQuad.test_cosine_weighted_infinitec              	   C   sD   dd }d}t t|dd|dddttdd	|   d	 d   d S )
Nc                 S   s   dd|  d|    S Nr7   rN   r   rf   r   r   r   rZ      rT   z2TestQuad.test_algebraic_log_weight.<locals>.myfunc      ?r7   alg)      rz   rj   rN   )r   r   r   r   )r1   rZ   rg   r   r   r   test_algebraic_log_weight   s   z"TestQuad.test_algebraic_log_weightc              	   C   s   dd }d}dt d dt d|  d d|  d	    td
|d   td
|  d|  d	  }tt|ddddd
d|dd d S )Nc                 S   s    d|  | d d d|    S )Nr>   r7   rN   rp   r   rf   r   r   r   rZ          z-TestQuad.test_cauchypv_weight.<locals>.myfuncg?g\`@?rw   g]`@?rp      r7   r>   rN   r   r6   Zcauchyrj   g#fT>)r   )r   r   r   r   )r1   rZ   rg   ZtabledValuer   r   r   test_cauchypv_weight   s   
"

zTestQuad.test_cauchypv_weightc                 C   sN   dd }t |dtjdd\}}t |tjddd\}}t|| t||d d S )Nc                 S   s   |t | |   S r   )npr   )rW   pqr   r   r   f   r=   z&TestQuad.test_b_less_than_a.<locals>.fr   rN      rk   r   r   r   infr   maxr1   r   Zval_1Zerr_1Zval_2Zerr_2r   r   r   test_b_less_than_a      zTestQuad.test_b_less_than_ac                 C   sV   dd }t |tj tjdd\}}t |tjtj dd\}}t|| t||d d S )Nc                 S   s&   t | d  d | t d|  S )NrN   r>   )r   r   r   )rW   sr   r   r   r      s   &z(TestQuad.test_b_less_than_a_2.<locals>.f)rN   r   r   r   r   r   r   r   test_b_less_than_a_2   s   zTestQuad.test_b_less_than_a_2c                 C   sN   dd }t |ddddd\}}t |ddddd\}}t|| t||d d S )	Nc                 S      dS N      ?r   r]   r   r   r   r         z(TestQuad.test_b_less_than_a_3.<locals>.fr   r7   ry   r   r   )rl   rm   r   )r   r   r   r   r   r   r   test_b_less_than_a_3   r   zTestQuad.test_b_less_than_a_3c                 C   s^   dd }t |dddddd}t |dddddd}t|d |d }t|d |d  |d	 d S )
Nc                 S   r   r   r   r]   r   r   r   r      r   z2TestQuad.test_b_less_than_a_full_output.<locals>.fr   r7   ry   r   T)rl   rm   full_outputr   )r   r   r   )r1   r   Zres_1Zres_2r   r   r   r   test_b_less_than_a_full_output   s
   z'TestQuad.test_b_less_than_a_full_outputc                 C   sB   dd }d\}}t t|||dd dd d|d |d    d S )	Nc                 S   s   ||  S r   r   rR   rW   r   r   r   simpfunc   s   z/TestQuad.test_double_integral.<locals>.simpfuncr   r>   c                 S      | S r   r   r]   r   r   r   <lambda>       z/TestQuad.test_double_integral.<locals>.<lambda>c                 S      d|  S NrN   r   r]   r   r   r   r          g?rq   r   r   r1   r   rg   br   r   r   test_double_integral   s
   zTestQuad.test_double_integralc              	   C   s:   dd }dd }dd }d}t t|dd|||d	d
 d S )Nc                 S   s   | | | | S r   r   )x0x1t0t1r   r   r   r4      rh   z,TestQuad.test_double_integral2.<locals>.funcc                 S   r   r   r   r]   r   r   r   r      r   z0TestQuad.test_double_integral2.<locals>.<lambda>c                 S   r   r   r   r]   r   r   r   r      r   )r7   rN   r7   rN   r   g$@r   )r1   r4   ghrk   r   r   r   test_double_integral2   s
   zTestQuad.test_double_integral2c                 C   s"   dd }t t|ddddd d S )Nc                 S   s   | | d d S rv   r   r   r   r   r   r   r4     rh   z,TestQuad.test_double_integral3.<locals>.funcr7   rN   g      @r   )r1   r4   r   r   r   test_double_integral3  s   zTestQuad.test_double_integral3c              
   C   sP   dd }d\}}t t|||dd dd dd dd d	d
|d |d    d S )Nc                 S   s   || |  | S r   r   )rY   rR   rW   tr   r   r   r   	  rh   z/TestQuad.test_triple_integral.<locals>.simpfuncr   c                 S   r   r   r   r]   r   r   r   r     r   z/TestQuad.test_triple_integral.<locals>.<lambda>c                 S   r   r   r   r]   r   r   r   r     r   c                 S   s   | | S r   r   rW   rR   r   r   r   r     r   c                 S   s   | | S r   r   r   r   r   r   r     r   )r>   gUUUUUU@rp   )r   r   r   r   r   r   test_triple_integral  s   
zTestQuad.test_triple_integralN)rE   rF   rG   r:   rQ   rc   ro   rs   ru   r{   r~   r   r   r   r   r   r   r   r   r   r   r   r   rV      s"    	


		rV   c                   @   s\   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd ZdS )	TestNQuadc                 C   sz   dd }dd }t |ddgddgdd	gd
dgg|i i i gdd}t|d d d t|d d dko9|d d dk  d S )Nc                 S   sP   | d ||  |d  t |  | d|  d d|  dkr#d }|S d }|S )NrN   r   皙?      ?      ?r   r7   r   r   )r   r   x2x3valr   r   r   func1  s    z*TestNQuad.test_fixed_limits.<locals>.func1c                  W   s"   dd| d  d d| d   giS )Nrb   r   rN   r   r   r   r   r   r   r   r   
opts_basic  s   "z/TestNQuad.test_fixed_limits.<locals>.opts_basicr   r7   rx   gp=
ף?g?g333333ÿT)optsr   g5m?Znevalg     jA)r   r   r
   )r1   r   r   resr   r   r   test_fixed_limits  s   (zTestNQuad.test_fixed_limitsc                    s   d dd } fdd} fdd} fdd	} fd
d}dd }dd }dd }dd }	t |||||gd||||	gd}
t|
d d S )Ng?c                 S   sD   | | |d  t | d | ||  | dkrd }|S d }|S )NrN   r7   r   r   )r   r   r   r   r   r   r   r   r   r   func2&  s   z-TestNQuad.test_variable_limits.<locals>.func2c                    sT    | d | t || |  d  d  | d | t || |  d  d gS NrN   r7   )r   r   r   r   r   r   r   Zscaler   r   lim0+  s   ((z,TestNQuad.test_variable_limits.<locals>.lim0c                    s0    ||  ||   d  ||  ||   d gS rd   r   r   r   r   r   r   r   r   lim1/  s   z,TestNQuad.test_variable_limits.<locals>.lim1c                    s8    | |d |d    d  | |d |d    d gS NrN   r   r7   r   r   r   r   r   r   r   lim23  s   z,TestNQuad.test_variable_limits.<locals>.lim2c                    s     | |  d  | |  d gS rd   r   r   r   r   r   r   lim37  r|   z,TestNQuad.test_variable_limits.<locals>.lim3c                 S   s   d|||   giS )Nrb   r   r   r   r   r   opts0:  s   z-TestNQuad.test_variable_limits.<locals>.opts0c                 S      i S r   r   r   r   r   r   opts1=  r   z-TestNQuad.test_variable_limits.<locals>.opts1c                 S   r   r   r   r   r   r   r   opts2@  r   z-TestNQuad.test_variable_limits.<locals>.opts2c                 S   r   r   r   r   r   r   r   opts3C  r   z-TestNQuad.test_variable_limits.<locals>.opts3r   )rk   r   g9@)r   r   )r1   r   r   r   r   r   r   r   r   r   r   r   r   r   test_variable_limits#  s   
zTestNQuad.test_variable_limitsc                 C   s0   dd }t t|ddgddggi i gdd d S )Nc                 S   r   r   r   r   r   r   r   r   K  r   z9TestNQuad.test_square_separate_ranges_and_opts.<locals>.frx   r7   r   rp   r   r   )r1   r   r   r   r   $test_square_separate_ranges_and_optsJ  s   (z.TestNQuad.test_square_separate_ranges_and_optsc                 C   s4   dd }ddg}i }t t|||g||gdd d S )Nc                 S   r   r   r   r   r   r   r   r   Q  r   z8TestNQuad.test_square_aliased_ranges_and_opts.<locals>.frx   r7   r   rp   r   )r1   r   rZoptr   r   r   #test_square_aliased_ranges_and_optsP  s    z-TestNQuad.test_square_aliased_ranges_and_optsc                 C   sP   dd }dd }dd }dd }d	d
 }||g}||g}t t|||dd d S )Nc                 S   r   r   r   r   r   r   r   r   Y  r   z<TestNQuad.test_square_separate_fn_ranges_and_opts.<locals>.fc                  W   r   N)rx   r7   r   r   r   r   r   	fn_range0\  r   zDTestNQuad.test_square_separate_fn_ranges_and_opts.<locals>.fn_range0c                  W   r   r   r   r   r   r   r   	fn_range1_  r   zDTestNQuad.test_square_separate_fn_ranges_and_opts.<locals>.fn_range1c                  W   r   r   r   r   r   r   r   fn_opt0b  r   zBTestNQuad.test_square_separate_fn_ranges_and_opts.<locals>.fn_opt0c                  W   r   r   r   r   r   r   r   fn_opt1e  r   zBTestNQuad.test_square_separate_fn_ranges_and_opts.<locals>.fn_opt1r   rp   r   )r1   r   r   r   r   r   rangesr   r   r   r   'test_square_separate_fn_ranges_and_optsX  s   z1TestNQuad.test_square_separate_fn_ranges_and_optsc                 C   s@   dd }dd }dd }||g}||g}t t|||dd d S )	Nc                 S   r   r   r   r   r   r   r   r   m  r   z;TestNQuad.test_square_aliased_fn_ranges_and_opts.<locals>.fc                  W   r   r   r   r   r   r   r   fn_rangep  r   zBTestNQuad.test_square_aliased_fn_ranges_and_opts.<locals>.fn_rangec                  W   r   r   r   r   r   r   r   fn_opts  r   z@TestNQuad.test_square_aliased_fn_ranges_and_opts.<locals>.fn_optr   rp   r   )r1   r   r   r   r   r   r   r   r   &test_square_aliased_fn_ranges_and_optsl  s   z0TestNQuad.test_square_aliased_fn_ranges_and_optsc                 C   sF   dd }t |dd\}}t|ddggd\}}t|| t|| d S )Nc                 S   s   | d d S r   r   r]   r   r   r   r4   {  rt   z*TestNQuad.test_matching_quad.<locals>.funcr      )r   )r   r   r   )r1   r4   r   reserrres2reserr2r   r   r   test_matching_quadz  s
   
zTestNQuad.test_matching_quadc                 C   sR   dd }t |dddd dd \}}t|dd	gd
g\}}t|| t|| d S )Nc                 S   s   | d |d  | |  d S r   r   r   r   r   r   func2d  r[   z/TestNQuad.test_matching_dblquad.<locals>.func2drN   c                 S   r   )Nr   r]   r   r   r   r     r   z1TestNQuad.test_matching_dblquad.<locals>.<lambda>c                 S   r   )Nr   r   r]   r   r   r   r     r   r   r   )r   rN   )r   r   r   )r1   r   r   r   r   r   r   r   r   test_matching_dblquad  s
   
zTestNQuad.test_matching_dblquadc              
   C   s`   dd }t |dddd dd dd d	d d
d}t|tj tjgddgdgd
d}t|| d S )Nc                 S   s.   | d ||d   | |  d |t |  S r   r   )r   r   r   Zc0Zc1r   r   r   func3d  s   .z/TestNQuad.test_matching_tplquad.<locals>.func3drx   rN   c                 S   r   )Nr   r   r]   r   r   r   r     r   z1TestNQuad.test_matching_tplquad.<locals>.<lambda>c                 S   r   r   r   r]   r   r   r   r     r   c                 S   s   t j S r   r   r   r   r   r   r   r     r   c                 S   s   t jS r   r   r   r   r   r   r     s    r   r   r   )rx   rN   )r   r   r   r   r   )r1   r   r   r   r   r   r   test_matching_tplquad  s   "zTestNQuad.test_matching_tplquadc                 C   s<   zt dd ddgddggddid W d S  ty   J w )Nc                 S   s   | | S r   r   r   r   r   r   r     r   z-TestNQuad.test_dict_as_opts.<locals>.<lambda>r   r7   Zepsrelg-C6?r   )r   	TypeErrorr9   r   r   r   test_dict_as_opts  s
   (zTestNQuad.test_dict_as_optsN)rE   rF   rG   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r     s    '		
r   )r   )'r*   r8   Znumpyr   r   r   r   r   r   r   r   r	   Znumpy.testingr
   r   r   r   r.   Zscipy.integrater   r   r   r   Zscipy._lib._ccallbackr   r   Zctypes.utilZscipy._lib._ccallback_cr   Z"scipy.integrate._test_multivariateZ	integrateZ_test_multivariater   r   r$   r&   rH   rV   r   r   r   r   r   <module>   s&    (
H 