a
    ߙfb                     @   s   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 dlmZmZmZmZmZ d dlmZmZmZ dd Zd	d
 Ze jddd Zdd Zdd Zdd Zdd Ze jddd Zdd ZdS )    N)assert_equalassert_allclose)units)	HAS_SCIPY)_lengthcircmeancircvar
circmomentcirccorrcoef)rayleightestvtestvonmisesmlec                  C   s@   t g d} d}t g dtj }t|t|| ddd d S )N)         r      r   r   g]n0a?)r   g@$   H   l   gfffff&e@D  )weights-C6?Zatol)nparrayudegr   r   )r   answerdata r   Alib/python3.9/site-packages/astropy/stats/tests/test_circstats.pytest__length   s    r!   c                  C   s8   t g dtj } dtj }t|t t| d d S )N3   C   (   m      if  gq=
ףPH@r   )r   r   r   r   r   aroundr   r   r   r   r   r    test_circmean   s    
r*   znot HAS_SCIPYc                  C   sD   dd l } tg d}| j|}tt|dtt|d d S )Nr   )g2y{?g]v>?g֠yJW?g3KDp?gǨBnLP?g^<@r   )Zscipy.statsr   r   Zstatsr   r   r(   )Zscipyr   r   r   r   r    test_circmean_against_scipy   s    r+   c                  C   s.   t g dtj } d}t|t| dd d S )Nr"   gި?r   r   )r   r   r   r   r   r   r)   r   r   r    test_circvar+   s    r,   c                  C   s   t g dtj } t g d}t t |tj d}t t| ddd dt t| ddd dt t| ddd df}t|d |d  t|d |d  t|d |d  t g d	}t|t| ddd t| ddd t| ddd fd
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   resultr   r   r    test_circmoment3   s"    r3   c                  C   sD   t g dtj } t g dtj }d}t|t| |dd d S )N)id  a         iW  $     .  iO  r9   r   U   r   iT  r8            r6   z   iI  )w         i       r4   r7   r%   i9  ^   -   /   r   rA   rB   r?      rB   rE      gKO?r   r   )r   r   r   r   r   r
   )ZalphaZbetar   r   r   r    test_circcorrcoefJ   s    rI   c                  C   sX   t g dtj } d}t| t| f}t|d |d dd t|d |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   r1   r   r   r    test_rayleightestU   s
    rK   c                  C   s.   t g dtj } d}t|t| dd d S )NrJ   g$?h㈵>r   )r   r   r   r   r   r   r)   r   r   r    
test_vtest_   s    rM   c                  C   s   t g d} d}t|d t| d dd t|d t| d dd t | tj } t dtj }tt |dt t| d d d S )	N)
gԠ
@g8+&*@g?g+}@g'E&@g$9?gH\@g^-ջ>@g }lܼ@g@)窺G6W@g˟o?r   rL   r   r   rN   r/   )	r   r   r   r   r0   r   r   r   r(   r)   r   r   r    test_vonmisesmleh   s    rO   )ZpytestZnumpyr   Znumpy.testingr   r   Zastropyr   r   Z"astropy.utils.compat.optional_depsr   Zastropy.stats.circstatsr   r   r   r	   r
   r   r   r   r!   r*   ZmarkZskipifr+   r,   r3   rI   rK   rM   rO   r   r   r   r    <module>   s"   	




