ó
ĪYc           @   s   d  d l  Z  d  d l m Z m Z d  d l m Z d  d l Z e  j d    Z	 d   Z
 d   Z d   Z d   Z d	   Z d
   Z d S(   i’’’’N(   t   Binst	   Histogram(   t   ColumnDataSourcec         C   s   t  |  j  S(   N(   R   t	   auto_data(   t	   test_data(    (    s8   lib/python2.7/site-packages/bkcharts/tests/test_stats.pyt   ds	   s    c         C   s:   t  d |  d d d d  } t | j  d k s6 t  d  S(   Nt   sourcet   columnt   mpgt   binsi   (   R    t   lenR	   t   AssertionError(   R   t   b(    (    s8   lib/python2.7/site-packages/bkcharts/tests/test_stats.pyt   test_explicit_bin_count   s    c         C   ss   t  d |  d d  } t | j  d k s0 t  t  d d d d d d g d d   } t | j  d k so t  d  S(	   NR   R   R   i   t   valuesi   R	   i   (   R    R
   R	   R   t   None(   R   R   (    (    s8   lib/python2.7/site-packages/bkcharts/tests/test_stats.pyt   test_auto_bin_count   s    $c         C   sK   t  d |  d d d d  t t j |  j d  j    d k sG t  d  S(   NR   R   t   cylR	   i   t   cyl_bin(   R    R
   t   pdt   Seriest   datat   drop_duplicatesR   (   R   (    (    s8   lib/python2.7/site-packages/bkcharts/tests/test_stats.pyt   test_bin_labeling   s    c          C   s¶   t  t d   }  t d |  d d  } t | j  d k sB t  g  | j D] } | j d ^ qL d d d g k sz t  g  | j D] } | j d ^ q d d d	 g k s² t  d  S(
   Ni
   R   R	   i   i    s   [0.000000, 3.000000]s   (3.000000, 6.000000]s   (6.000000, 9.000000]i   (   t   listt   rangeR   R
   R	   R   t   labelR   (   R   t   hR   (    (    s8   lib/python2.7/site-packages/bkcharts/tests/test_stats.pyt   test_histogram_wo_density#   s
    8c          C   s¼   t  t d   }  t d |  d d d t  } t | j  d k sH t  g  | j D] } | j d ^ qR d d d	 g k s t  g  | j D] } | j d ^ q d
 d
 d g k sø t  d  S(   Ni
   R   R	   i   t   densityi    s   [0.000000, 3.000000]s   (3.000000, 6.000000]s   (6.000000, 9.000000]g¹?gĮ?(	   R   R   R   t   TrueR
   R	   R   R   R   (   R   R   R   (    (    s8   lib/python2.7/site-packages/bkcharts/tests/test_stats.pyt   test_histogram_w_density,   s
    8c          C   s   x d
 D] }  |  |  g } t  d |  } t | j  d k sC t  t | j  d k s^ t  t g  | j D] } | j ^ qk  d	 k s t  q Wd  S(   Niė’’’gü©ńŅMbPæi    gü©ńŅMbP?i   R   i   i   i   (   iė’’’gü©ńŅMbPæi    gü©ńŅMbP?i   (   R   R
   R	   R   t   sumt   value(   t   xR   R   R   (    (    s8   lib/python2.7/site-packages/bkcharts/tests/test_stats.pyt   test_histogram_ill_defined_data5   s    (   t   pytestt   bkcharts.statsR    R   t   bokeh.modelsR   t   pandasR   t   fixtureR   R   R   R   R   R   R#   (    (    (    s8   lib/python2.7/site-packages/bkcharts/tests/test_stats.pyt   <module>   s   							