B
    Y                 @   sn   d dl Z d dlmZmZ d dlmZ d dlZe jdd Z	dd Z
dd	 Zd
d Zdd Zdd Zdd ZdS )    N)Bins	Histogram)ColumnDataSourcec             C   s
   t | jS )N)r   Z	auto_data)Z	test_data r   8lib/python3.7/site-packages/bkcharts/tests/test_stats.pyds	   s    r   c             C   s$   t | ddd}t|jdks td S )Nmpg   )sourcecolumnbins)r   lenr   AssertionError)r   br   r   r   test_explicit_bin_count   s    r   c             C   sJ   t | dd}t|jdkstt dddddgd d}t|jdksFtd S )Nr   )r
   r         )valuesr      )r   r   r   r   )r   r   r   r   r   test_auto_bin_count   s    r   c             C   s2   t | ddd tt| jd  dks.td S )NZcylr	   )r
   r   r   Zcyl_bin)r   r   pdZSeriesdataZdrop_duplicatesr   )r   r   r   r   test_bin_labeling   s    r   c              C   sj   t td} t| dd}t|jdks*tdd |jD dddgksHtd	d |jD ddd
gksftd S )N
   r   )r   r   c             S   s   g | ]}|j d  qS )r   )label).0r   r   r   r   
<listcomp>(   s    z-test_histogram_wo_density.<locals>.<listcomp>z[0.000000, 3.000000]z(3.000000, 6.000000]z(6.000000, 9.000000]c             S   s   g | ]}|j d  qS )r   )r   )r   r   r   r   r   r   )   s       )listranger   r   r   r   )r   hr   r   r   test_histogram_wo_density#   s
    r!   c              C   sl   t td} t| ddd}t|jdks,tdd |jD ddd	gksJtd
d |jD dddgkshtd S )Nr   r   T)r   r   Zdensityc             S   s   g | ]}|j d  qS )r   )r   )r   r   r   r   r   r   1   s    z,test_histogram_w_density.<locals>.<listcomp>z[0.000000, 3.000000]z(3.000000, 6.000000]z(6.000000, 9.000000]c             S   s   g | ]}|j d  qS )r   )r   )r   r   r   r   r   r   2   s    g?g?)r   r   r   r   r   r   )r   r    r   r   r   test_histogram_w_density,   s
    r"   c              C   sd   x^dD ]V} | | g}t |d}t|jdks.tt|jdks@ttdd |jD dkstqW d S )N)igMbPr   gMbP?   )r   r      c             S   s   g | ]
}|j qS r   )value)r   r   r   r   r   r   =   s    z3test_histogram_ill_defined_data.<locals>.<listcomp>r	   )r   r   r   r   sum)xr   r    r   r   r   test_histogram_ill_defined_data5   s    

r(   )ZpytestZbkcharts.statsr   r   Zbokeh.modelsr   Zpandasr   Zfixturer   r   r   r   r!   r"   r(   r   r   r   r   <module>   s   		