o
    `^h-                     @   s  d dl Z d dlmZ d dlZd dlZd dlZd dlm	Z	m
Z
 dd Zdd Zdd Zd	d
 Zdd Zejdg ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<ej ejd=  d>fd?ejejd=  d>fd@dA Ze jjdB dC dD ZedEkoejdFkZedGkoejdHkZejjepe dIdJejdg dKdLdM ZdS )N    N)assert_equalassert_allclosec                   C   s(   t tddtjd d d dd d S )N         -q=rtol)r   sczetanppi r   r   [/home/air/shanriGPT/back/venv/lib/python3.10/site-packages/scipy/special/tests/test_zeta.py	test_zeta
   s   (r   c                  C   s(   g d} g d}t t| |dd d S )N)g g?gH.?	   2   K   )g^ңge
gp#	6gj-s`?gXk   <g    @;r   r   )r   r
   zetac)xdesiredr   r   r   
test_zetac   s   r   c                   C   sj   t tjdks
J tt tj sJ t ddksJ t dtjks)J tt g dd d S )Nr   g      g      ?)ii)r
   r   r   infisnanr   r   r   r   r   test_zetac_special_cases   s
   r   c                   C   s   t tt jsJ tt jdksJ tddksJ ttg dd ttdt jd d dd ttd	t jd	 d
 dd d S )Nr   r   g      )r   iiir   r   r   r      Z   )	r   r   r
   r   nanr   r   r   r   r   r   r   r   test_riemann_zeta_special_cases&   s   "r!   c                  C   s"   d} d}t t| |ddd d S )Ng    @pgkr   gvIh%,=)atolr	   r   r
   r   )sr   r   r   r    test_riemann_zeta_avoid_overflow2   s   r%   zz, desired, rtol)y܍	@ޓq@yӻQN?%Kh?vIh%<=)yz&E@ayf|~?xL{r&   )ynN9 S;@yV(?eyr&   )yK"Bnyy[r6nr&   )y?e5@MtN@y̌q]?AaO?r&   )yRp{fӰ@|/PyIU ?ž?r&   )y g
S@yϜ<@]@r&   )y@0Ry
)اAr&   )yv_)?0PZ?ygPq i?{^A޿r&   )y2?cB@y]'??r&   )y}d?wOP@y+j<?.ѿr&   )yvQ_?m5@y]hN?>ä?r&   )yi6?{V@@ypjXۦ`4?r&   )yv_)?0PZygPq i?{^A?r&   )y2?cBy]'?ƿr&   )y}d?wOPy+j<?.?r&   )yvQ_?m5y]hN?>är&   )yi6?{V@ypjXۦ`4ǿr&   )y       V瞯<ygOkWe9>/fHz>)y      V瞯<yqU-$yhb<r'   )y      V瞯<y|XZ_45[r'   )y       V瞯<ycQ*9G9,c<r'   )y       V瞯ҼyOkWe9>/f<r'   )y      V瞯ҼyQqU-$yhbr'   )y      V瞯ҼyrXZ_45[<r'   )y       V瞯ҼyZQ*9G9,cr'   )y      V瞯<yʿ)Cp r'   )y        V瞯<y       ۍмr'   )y?V瞯<yHG|Hd5ڼr'   )y      ?V瞯<y;	5r&   )y      V瞯Ҽyʿ)Cp <r'   )y       V瞯Ҽy      `ۍ<r'   )y?V瞯ҼyHG|0t5<r'   )y      ?V瞯Ҽy;	3=r&   )y;ziH@ESJ@y     ?,莕e<r&   )y8-CnGI@ESJ@y     ?ՌRP<r&   )y;ziH@)|QD@y?	Idr&   )y8-CnGI@)|QD@y?(r&   )y;ziH@yuU@y?}6 Gwr&   )y8-CnGI@yuU@y?r&   )y      ?    .Ayz?ی^p@r'   )y_+_hA@    .Ayf    ?
lR=r&   )y8-CnGI@    .Ay     ?Z <r&   )y      ?    .yz?ی^pr'   )y_+_hA@    .yf    ?
lRr&   )y8-CnGI@    .y     ?Z ļr&   yX\mrh`<+@y              ?r&   yX\mr%0c                 C      t t| ||d d S Nr   r#   zr   r	   r   r   r   test_riemann_zeta_complex8   s    Rr,   	Compilerszc++namegcclinuxclangdarwinz/Underflow may not be avoided on other platforms)reason))yF5,kh`<+@yqdݻIZ-a=)ypmh`<+@y:q#Nxm Vkr&   )yF5,k%0y61;Jjr4   )ypm%0yC@Ҷ*Ѣ˺dr4   c                 C   r(   r)   r#   r*   r   r   r   )test_riemann_zeta_complex_avoid_underflow  s   r5   )scipyscipy.specialspecialr
   sysnumpyr   pytestnumpy.testingr   r   r   r   r   r!   r%   markparametrizer   r,   
__config__CONFIGcpp_compilerplatform	gcc_linuxclang_macOSskipifr5   r   r   r   r   <module>   s   	^adgknqtx{~                               !  "  #  $  %  &  '  (  *  -  0  3  6  9  =  >  A  D  E  H  K  L  
Q