o
    tBh5                     @   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   )atolrtol)r   r   )value_and_errtabled_valueerrTolvalueerr r   z/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-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cast	CFUNCTYPE)namerestypeargtypes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"   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   r2   r   mathr   r.   r7   r   r   r   test_typical:   s   $$(zTestCtypesQuad.test_typicalc                 C   s   t ttdd d S Nr   r=   )r   r   r   r?   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 )N_sin_0_sin_1_sin_2_sin_3r          @)r)   r"   r6   c_void_pc_intPOINTER	enumerater   r4   raises
ValueErrorr   r   r   )r7   sin_0sin_1sin_2sin_3sin_4all_sigslegacy_sigslegacy_only_sigsjr:   callbackr   r   r   test_ctypes_variantsB   s:   z#TestCtypesQuad.test_ctypes_variantsN)__name__
__module____qualname__r;   r@   rB   rY   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"   r6   rJ   r)   setattr)r7   r&   r'   r%   r:   r   r   r   r;   h   s   z'TestMultivariateCtypesQuad.setup_methodc                 C   s   t t| jdtdd d S )Nr      g?ۗ?)r   r   r^   r   r?   r   r   r   r@   p   s   z'TestMultivariateCtypesQuad.test_typicalc                 C   s   t t| jdtd d S )Nr   ox?)r   r   r_   r	   r?   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 rA   )r   r`   )yr?   r   r   threadsafety|      zBTestMultivariateCtypesQuad.test_threadsafety.<locals>.threadsafetyr   r=   gtJ׵?)r   r   )r7   rh   r   r?   r   test_threadsafetyz   s   z,TestMultivariateCtypesQuad.test_threadsafetyN)rZ   r[   r\   r;   r@   rf   rj   r   r   r   r   r]   g   s
    r]   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   rb   rd   )r   r   r   r7   ro   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   rl   r   r   r   ro      rC   z(TestQuad.test_indefinite.<locals>.myfuncr   re   r   r   r	   rq   r   r   r   rf      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   rr   r   r   r   ro      s   
z&TestQuad.test_singular.<locals>.myfuncr   
   rt   ru   )pointsr=   g      g      )r   r   r   r   rq   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 Nr=   r   rl   ar   r   r   ro         z2TestQuad.test_sine_weighted_finite.<locals>.myfuncglSS%@r   r=      r   argsweightwvarii  rc   )r   r   r   r   r   )r7   ro   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   rz   r{   r   r   r   ro      s   z4TestQuad.test_sine_weighted_infinite.<locals>.myfunc      @      @r   r   r   rc   rs   r7   ro   r|   r   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   rz   r{   r   r   r   ro         z6TestQuad.test_cosine_weighted_infinite.<locals>.myfuncrt   gffffff@r   r   r   rc   rs   r   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 Nr=   rc   r   r{   r   r   r   ro      ri   z2TestQuad.test_algebraic_log_weight.<locals>.myfunc      ?r=   alg)      r   r   rc   )r   r   r   r   )r7   ro   r|   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 )NrH   r=   rc   r   r   r{   r   r   r   ro          z-TestQuad.test_cauchypv_weight.<locals>.myfuncg?g\`@?r   g]`@?r      r=   rH   rc   r   r<   cauchyr   g#fT>)r   )r   r   r   r   )r7   ro   r|   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   )rl   pqr   r   r   f   rC   z&TestQuad.test_b_less_than_a.<locals>.fr   rc      r   r   r   r   infr   maxr7   r   val_1err_1val_2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 )Nrc   rH   )r   r   r   )rl   sr   r   r   r      s   &z(TestQuad.test_b_less_than_a_2.<locals>.f)rc   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   rr   r   r   r   r         z(TestQuad.test_b_less_than_a_3.<locals>.fr   r=   r   r   r   )r   r   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   rr   r   r   r   r      r   z2TestQuad.test_b_less_than_a_full_output.<locals>.fr   r=   r   r   T)r   r   full_outputr   )r   r   r   )r7   r   res_1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   rg   rl   r   r   r   simpfunc   s   z/TestQuad.test_double_integral.<locals>.simpfuncr   rH   c                 S      | S r   r   rr   r   r   r   <lambda>       z/TestQuad.test_double_integral.<locals>.<lambda>c                 S      d|  S Nrc   r   rr   r   r   r   r          g?r   r   r   r7   r   r|   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   r:      r}   z,TestQuad.test_double_integral2.<locals>.funcc                 S   r   r   r   rr   r   r   r   r      r   z0TestQuad.test_double_integral2.<locals>.<lambda>c                 S   r   r   r   rr   r   r   r   r      r   )r=   rc   r=   rc   r   g$@r   )r7   r:   ghr   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 r   r   r   r   r   r   r   r:     r}   z,TestQuad.test_double_integral3.<locals>.funcr=   rc   g      @r   )r7   r:   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   )rn   rg   rl   tr   r   r   r   	  r}   z/TestQuad.test_triple_integral.<locals>.simpfuncr   c                 S   r   r   r   rr   r   r   r   r     r   z/TestQuad.test_triple_integral.<locals>.<lambda>c                 S   r   r   r   rr   r   r   r   r     r   c                 S   s   | | S r   r   rl   rg   r   r   r   r     r   c                 S   s   | | S r   r   r   r   r   r   r     r   )rH   gUUUUUU@r   )r   r   r   r   r   r   test_triple_integral  s   
zTestQuad.test_triple_integralN)rZ   r[   r\   r@   rf   rx   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   rk      s"    	


		rk   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 )Nrc   r   皙?      ?      ?r   r=   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 )Nrw   r   rc   r   r   r   r   r   r   r   r   
opts_basic  s   "z/TestNQuad.test_fixed_limits.<locals>.opts_basicr   r=   r   gp=
ף?g?g333333ÿT)optsr   g5m?nevalg     jA)r   r   r
   )r7   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 )Nrc   r=   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 Nrc   r=   )r   r   r   r   r   r   r   scaler   r   lim0+  s   ((z,TestNQuad.test_variable_limits.<locals>.lim0c                    s0    ||  ||   d  ||  ||   d gS ry   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 Nrc   r   r=   r   r   r   r   r   r   r   lim23  s   z,TestNQuad.test_variable_limits.<locals>.lim2c                    s     | |  d  | |  d gS ry   r   r   r   r   r   r   lim37  r   z,TestNQuad.test_variable_limits.<locals>.lim3c                 S   s   d|||   giS )Nrw   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   )r   r   g9@)r   r   )r7   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>.fr   r=   r   r   r   r   )r7   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>.fr   r=   r   r   r   )r7   r   r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)r   r=   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   r   r   )r7   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   r   r   )r7   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   rr   r   r   r   r:   {  r   z*TestNQuad.test_matching_quad.<locals>.funcr      )r   )r   r   r   )r7   r:   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  rp   z/TestNQuad.test_matching_dblquad.<locals>.func2drc   c                 S   r   )Nr   rr   r   r   r   r     r   z1TestNQuad.test_matching_dblquad.<locals>.<lambda>c                 S   r   )Nr   r   rr   r   r   r   r     r   r  r   )r  rc   )r   r   r   )r7   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   c0c1r   r   r   func3d  s   .z/TestNQuad.test_matching_tplquad.<locals>.func3dr   rc   c                 S   r   )Nr  r   rr   r   r   r   r     r   z1TestNQuad.test_matching_tplquad.<locals>.<lambda>c                 S   r   r   r   rr   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  )r   rc   )r   r   r   r   r   )r7   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   r=   epsrelg-C6?r   )r   	TypeErrorr?   r   r   r   test_dict_as_opts  s
   (zTestNQuad.test_dict_as_optsN)rZ   r[   r\   r   r   r   r   r   r   r  r  r  r  r   r   r   r   r     s    '		
r   )r   )'r/   r>   numpyr   r   r   r   r   r   r   r   r	   numpy.testingr
   r   r   r   r4   scipy.integrater   r   r   r   scipy._lib._ccallbackr   r"   ctypes.utilscipy._lib._ccallback_cr   "scipy.integrate._test_multivariate	integrate_test_multivariater!   r   r)   r+   r]   rk   r   r   r   r   r   <module>   s&    (
H 