a
    ߙfb                     @   sx   d dl Z d dlZd dlmZmZ d dlmZmZ d dl	m
Z
 dd Ze jddd	 Ze jdd
d Zdd ZdS )    N)assert_equalassert_allclose)jackknife_resamplingjackknife_stats)	HAS_SCIPYc                  C   sB   t g d} t g dg dg dg dg}t|t|  d S )N)            )r   r	   r
   )r   r	   r
   )r   r   r
   )r   r   r	   )nparrayr   r   dataanswer r   Alib/python3.9/site-packages/astropy/stats/tests/test_jackknife.pytest_jackknife_resampling   s    "r   znot HAS_SCIPYc                  C   s0   t d} d}t|t| t jdd dd d S )N)	s            i  i  i     i  )gz,C'p@g        g^2!I@r   r	   -C6?Zatol)r   r   r   r   meanr   r   r   r   test_jackknife_stats   s    
r   c                  C   s|   t g d} t | d} | d d df } d}dd }t|t| |ddd	 d
d t d}t|t| |dd	 d
d d S )N)40   *   $   !            '   r   &   r   r   r       r   r      r    )   +   -   "      r&         r   r%   r   r*      r"   r*   r'   r
          r$         /      r'   r'   r1   r.      r+      r.   r'   (   )r   r   )g_Qr\@gnlg9D6@c                 S   s*   t | t |  | t |   t|  S )N)r   sumr   len)xr   r   r   mle_var*       z3test_jackknife_stats_conf_interval.<locals>.mle_vargffffff?r   r	   r   r   )gޓZQ@gKc@)r   r   Zreshaper   r   )r   r   r<   r   r   r   "test_jackknife_stats_conf_interval   s    
r>   c                   C   sD   t t& ttdtjdd W d    n1 s60    Y  d S )Nr   r   )Zconfidence_level)pytestZraises
ValueErrorr   r   Zaranger   r   r   r   r   test_jackknife_stats_exceptions4   s    rA   )r?   Znumpyr   Znumpy.testingr   r   Zastropy.stats.jackknifer   r   Z"astropy.utils.compat.optional_depsr   r   ZmarkZskipifr   r>   rA   r   r   r   r   <module>   s   

	

