
    xf1                     P    d dl Zd dlmZmZmZ d dlmZ d dlm	Z
 d Zd Zd Zd Zy)	    N)assert_equalassert_allclosesuppress_warnings)_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       ?/usr/lib/python3/dist-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    c                     t        dd      t        dd      t        dd      g} t        dt        j                         t        dt        j                        t        d	t        j                        g}t               5 }|j	                  t
        d
       t        | |      D ]  \  }}t        t        |      |        	 d d d        t        dd      }t        dt        j                         }t               5 }|j	                  t
        d
       t        t        |      |       d d d        y # 1 sw Y   oxY w# 1 sw Y   y xY w)Ng-     ?   gz8   g5$qi  g"Y_g~Y9g?!z%invalid value encountered in multiplygZ     ?)
complexr   infr   filterRuntimeWarningzipr   r   r   )	sinpi_pts	sinpi_stdsuppstds        r   test_intermediate_overlowr1       s    C($%'I 0266':0"&&9/8:I 
	 +

>#JK)Y/ 	+FAsE!Hc*	++ 	S!A
)BFF7
3C		 '

>#JKa#&' '+ +' 's   <A D+6,D7+D47E 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_signr4   7   sr    dA8O8::a==c
A8O8zz!}c
A8O8zz!}}r    )numpyr   numpy.testingr   r   r   scipy.special._ufuncsr   r   r   r   r   r#   r1   r4    r    r   <module>r9      s(     J J 1 1	 	 '.r    