
    bi`	                         d dl Z d dlZd dlZd dlmZmZ d dlmZ	 d dlm
Z d Zd Zej                  j                  d      d        Zd	 Zy)
    N)assert_equalassert_allclose)_sinpi)_cospic                  z   t        j                  dd      } t        j                  t        j                  ddd       t        j                  ddd      f      }t        j                  | |      \  } }| d|z  z   }t        |      }t        |j                  d       t        |      }t        |j                  d       y )Ne   6  
                 ?        )
nparangehstacklinspacemeshgridsinpir   realcospiimagxyzress       X/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/scipy/special/tests/test_trig.pytest_integer_real_partr      s    
		$A
		BKKS"--r{{3R/HIJA;;q!DAq	BqDA
(C3
(C3    c                     t        j                  dd      dz   } t        j                  t        j                  ddd       t        j                  ddd      f      }t        j                  | |      \  } }| d|z  z   }t        |      }t        |j                  d       t        |      }t        |j                  d       y )	Nr   r	         ?r
   r   r   r   r   )
r   r   r   r   r   r   r   r   r   r   r   s       r   test_half_integer_real_partr"      s    
		$s"A
		BKKS"--r{{3R/HIJA;;q!DAq	BqDA
(C3
(C3r   z/Temporary skip while gh-19526 is being resolvedc                     t        dd      t        dd      t        dd      g} t        dt        j                         t        dt        j                        t        d	t        j                        g}t        j                         5  t        j
                  d
dt               t        | |      D ]P  \  }}t        |      }t        |j                  |j                         t        |j                  |j                         R 	 d d d        t        dd      }t        dt        j                         }t        j                         5  t        j
                  d
dt               t        |      }t        |j                  |j                         t        |j                  |j                         d d d        y # 1 sw Y   xY w# 1 sw Y   y xY w)Ng-     ?   gz8   g5$qi  g"Y_g~Y9g?!ignorez%invalid value encountered in multiplygZ     ?)complexr   infwarningscatch_warningsfilterwarningsRuntimeWarningzipr   r   r   r   r   )	sinpi_pts	sinpi_stdpstdr   s        r   test_intermediate_overlowr2   #   sm    C($%'I 0266':0"&&9/8:I 
	 	 	" 0=~	O)Y/ 	0FAs(CCHHchh/CHHchh/	00 	S!A
)BFF7
3C		 	 	" ,=~	OAh#((+#((+, ,0 0, ,s   A;F5A'G5F>G
c                      t        d      } | dk(  sJ t        j                  |       sJ t        d      } | dk(  sJ t        j                  |       rJ t        d      } | dk(  sJ t        j                  |       rJ y )Ng       r   r!   )r   r   signbitr   )r   s    r   test_zero_signr5   A   sr    dA8O8::a==c
A8O8zz!}c
A8O8zz!}}r   )r)   pytestnumpyr   numpy.testingr   r   scipy.special._ufuncsr   r   r   r   r   r"   markskipr2   r5    r   r   <module>r=      sK       7 1 1	 	  CD, E,:r   