
    IR-e2                     (   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
mZmZmZmZmZmZmZ d dlmZ d Zd Zej                            e d	          d
             Zd Zd Zd Zd Zej                            e d	          d             Zd ZdS )    N)assert_allcloseassert_equal)units)_lengthcirccorrcoefcircmean
circmomentcircvarrayleightestvonmisesmlevtest)	HAS_SCIPYc                      t          j        g d          } d}t          j        g d          t          j        z  }t	          |t          ||           d           d S )N)         r      r   r   g]n0a?)r   g@$   H   l   gfffff&e@D  weights-C6?atol)nparrayudegr   r   )r   answerdatas      Blib/python3.11/site-packages/astropy/stats/tests/test_circstats.pytest__lengthr$      sd     h---..GF855566>DFGD':::FFFFFF    c                      t          j        g d          t          j        z  } dt          j        z  }t	          |t          j        t          |           d                     d S )N3   C   (   m      if  gq=
ףPH@r   )r   r   r   r    r   aroundr   r"   r!   s     r#   test_circmeanr/      sT     8...//!%7DQU]F8D>>15566666r%   zrequires scipy)reasonc                      dd l } t          j        g d          }| j                            |          }t          t          j        |d          t          j        t	          |          d                     d S )Nr   )g2y{?g]v>?g֠yJW?g3KDp?gǨBnLP?g^<@r   )scipy.statsr   r   statsr   r   r-   )scipyr"   r!   s      r#   test_circmean_against_scipyr5   $   su     8NNN D [!!$''F61%%ry$'C'CDDDDDr%   c                      t          j        g d          t          j        z  } d}t	          |t          |           d           d S )Nr'   gި?r   r   )r   r   r   r    r   r
   r.   s     r#   test_circvarr7   1   sG     8...//!%7DFFGDMM555555r%   c            	      V   t          j        g d          t          j        z  } t          j        g d          }t          j        t          j        |          t          j        z  d          }t          j        t          | d          d         d          t          j        t          | d          d         d          t          j        t          | d          d         d          f}t          |d         |d                    t          |d         |d                    t          |d         |d                    t          j        g d	          }t          |t          | d          d         t          | d          d         t          | d          d         fd
           d S )Nr'   )gsѐh?gmOT6l?gԜ|@   r   )pr      r   )go?gᔹF?gN.!z?r   r   )	r   r   r   r    r-   rad2degr	   r   r   r"   r!   results      r#   test_circmomentr?   9   s    8...//!%7D X44455FYrz&))AE1155F 		*TQ'''*A..
	*TQ'''*A..
	*TQ'''*A..F F1I&&&F1I&&&F1I&&&X66677F	DA			q	!:da#8#8#8#;ZPQ=R=R=RST=UV     r%   c                      t          j        g d          t          j        z  } t          j        g d          t          j        z  }d}t	          |t          | |          d           d S )N)id  a         iW  $     .  iO  rF   r   U   r   iT  rE            rC   z   iI  )w         i       rA   rD   r*   i9  ^   -   /   r   rN   rO   rL      rO   rR      gKO?r   r   )r   r   r   r    r   r   )alphabetar!   s      r#   test_circcorrcoefrX   T   s    
 H	
 	
 	
 
 	
E 8	
 	
 	
 
 	
D FFL55DAAAAAAr%   c                     t          j        g d          t          j        z  } d}t	          |           t          |           f}t          |d         |d         d           t          |d         |d         d           d S )Ng(\g@g(\e@gfffff~c@g(\:k@gQc@)gD;z?g:m½r?r   r   r   r   )r   r   r   r    r   r   r   r=   s      r#   test_rayleightestr[   l   s|    8<<<==ED$F4  '$--0FF1Ivayt4444F1Ivayt444444r%   c                      t          j        g d          t          j        z  } d}t	          |t          |           d           d S )NrZ   g$?h㈵>r   )r   r   r   r    r   r   r.   s     r#   
test_vtestr^   v   sG     8<<<==EDFFE$KKd333333r%   c                  6   t          j        g d          } d}t          |d         t          |           d         d           t          |d         t          |           d         d           t          j        |           t
          j        z  } t          j        d          t
          j        z  }t          t          j        |d          t          j        t          |           d         d                     t          j        t           j	        d	z  t           j	        t           j	        d	z  g          } t          |           }t          j        t           j	        d	z  t           j	        g          }d	dg}t          |d         t          ||
          d         d           t          |d         t          ||
          d         d           t          j        g dg dg          } t          j        ddg          t          j        ddg          f}t          |d         t          | d          d         d           t          |d         t          | d          d         d           t          j        |           t
          j        z  } t          j        |d                   t
          j        z  |d         f}t          |d         t          | d          d         d           t          |d         t          | d          d         d           d S )N)
gԠ
@g8+&*@g?g+}@g'E&@g$9?gH\@g^-ջ>@g }lܼ@g@)窺G6W@˟o?r   r]   r   r   r`   r;   r   r   ra   )axis)
r   r   r   r   r<   r   r    r   r-   pi)r"   r!   data_to_weighr   s       r#   test_vonmisesmlere      s   
 8	
 	
 	
 D "FF1I{4003$????F1I{4003$???? :dae#DZ!!AE)F61%%ryT1B1B11Eq'I'IJJJ 8EAIEEAI	
 D FHbeai/00M!fGq	;}g>>>qA    q	;}g>>>qA   
 8    	
 D: h(+,,bh(7K.L.LMFF1I{4a888;$GGGGF1I{4a888;$GGGG :dae#Dj##ae+VAY7FF1I{4a888;$GGGGF1I{4a888;$GGGGGGr%   )numpyr   pytestnumpy.testingr   r   astropyr   r   astropy.stats.circstatsr   r   r   r	   r
   r   r   r   "astropy.utils.compat.optional_depsr   r$   r/   markskipifr5   r7   r?   rX   r[   r^   re    r%   r#   <module>ro      s        7 7 7 7 7 7 7 7      	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 9 8 8 8 8 8G G G7 7 7 	M*:;;	E 	E <;	E6 6 6  6B B B05 5 5 	M*:;;4 4 <;4PH PH PH PH PHr%   