U
    Ãmœdd  ã                   @   sD   d dl Zd dlmZ d dlmZ dd„ Zdd„ Zdd„ Z	d	d
„ Z
dS )é    N)Ú	sequencesc                  C   sÒ   ddgddgddgddgddgg} ddgdd	gddgd
dgddgd	d
gg}ddgdd
gddgd
dgddgd	d	gg}t  ddgddgg¡}tjt | ¡ddd� tjt ||¡ddd� tjt ||¡ddd� d S )Ngš™™™™™¹?ç      à?gš™™™™™É?gš™™™™™Ù?g333333Ó?é   é   é   é   é   é   g      @gá“©‚QÁ?g-Cëâ6?)Zatolg‚âÇ˜»–€?g/Ý$�…?)ÚnpÚarrayÚnptÚassert_allcloser   Zdiscrepancy)Zspace_0Zspace_1Zspace_2Úcorners© r   ú_/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/tools/tests/test_sequences.pyÚtest_discrepancy   s    "((r   c               
   C   sZ   t  d¡} dddddddd	d
dg
}t | |¡ t jddd�} ddddd	g}t | |¡ d S )Né
   g        r   g      Ð?g      è?g      À?g      ä?g      Ø?g      ì?g      °?g      â?r	   r   )Ústart_index)r   Zvan_der_corputr   Úassert_almost_equal)ÚsampleÚoutr   r   r   Útest_van_der_corput   s    
r   c                  C   s<   t  d¡} dddddddd	d
ddddddg}t | |¡ d S )Né2   r   r   r	   é   é   é   é   é   é   é   é   é%   é)   é+   é/   )r   Zprimes_from_2_tor   r   )Zprimesr   r   r   r   Útest_primes   s    
"r%   c                  C   s    t  ddgddgg¡} tjdd| d�}t  ddgdd	gd
dgddgddgg¡}tj||dd� tjdd| dd�}t  d
dgddgddgg¡}tj||dd� d S )Nr   r   r   r	   )ÚdimÚn_sampleÚboundsg      @g      @g      @g      @g      @gffffff@g      ô?gffffff
@g      @g333333@r   )Údecimalr   )r&   r'   r(   r   )r
   r   r   Zhaltonr   r   )r   r   r   r   r   r   Útest_halton$   s    (r*   )Únumpyr
   Znumpy.testingÚtestingr   Zstatsmodels.toolsr   r   r   r%   r*   r   r   r   r   Ú<module>   s   
