a
    ߙfb
                     @   sP   d dl mZ d dlmZmZ d dlmZmZ dd Zdd Zdd	 Z	d
d Z
dS )    )assert_allclose)bayesian_info_criterionbayesian_info_criterion_lsq)akaike_info_criterionakaike_info_criterion_lsqc                  C   sN   d} d}d}d}t | d |d |}t | d |d |}t||| dd d S )	N)gfg     e)      d   g(\@r      g?Zatol)r   r   )lnLn_params	n_samplesanswerZbic_gZbic_t r   Clib/python3.9/site-packages/astropy/stats/tests/test_info_theory.pytest_bayesian_info_criterion   s    r   c                  C   sN   d} d}d}d}t |d |d | }t |d |d | }t||| dd d S )	Ny   )gRQgGz)      gffffff?r   r
   {Gz?r   )r   r   )r   r   r   r   Zaic_1Zaic_2r   r   r   test_akaike_info_criterion   s    r   c                  C   sz   d} d}d}d}t |d t|d |d | dd t |d t|d |d | dd t |d	 t|d	 |d	 | dd d S )
Nr	   )   r   r   )g      9@g      :@g      ;@)gQF`gQ`gQ+_r   r   r   r
   r   )r   r   r   r   Zssrr   r   r   r   test_akaike_info_criterion_lsq   s     r   c                  C   sz   d} d}d}d}t |d t|d |d | dd t |d t|d |d | dd t |d	 t|d	 |d	 | dd d
S )a  This test is from:
    http://www.statoek.wiso.uni-goettingen.de/veranstaltungen/non_semi_models/
    AkaikeLsg.pdf
    Note that in there, they compute a "normalized BIC". Therefore, the
    answers presented here are recalculated versions based on their values.
       )r
   r   r
   )i?  i   i\  )g;Oh@g;O6g@gQKg@r   r   r   r
   r   N)r   r   r   r   r   r    test_bayesian_info_criterion_lsq0   s,    r   N)Znumpy.testingr   Zastropy.stats.info_theoryr   r   r   r   r   r   r   r   r   r   r   r   <module>   s   