ó
ÎYc           @@  sU  d  Z  d d l m Z d d l m Z d d l Z d d l m Z y d d l Z	 e
 Z Wn e k
 rs Z e Z n Xd d l m Z m Z d d l m Z d d l m Z m Z d d	 l m Z m Z d
 d d g Z e j d „  ƒ Z e j d „  ƒ Z e j d „  ƒ Z d „  Z d „  Z  d „  Z! d „  Z" e e d ƒ d „  ƒ Z# d „  Z$ d S(   s-   This is the Bokeh charts testing interface.

i    (   t   absolute_import(   t   MockN(   t   skipIf(   t   Buildert	   XYBuilder(   t	   Dimension(   t	   ColorAttrt   DEFAULT_PALETTE(   t   Range1dt   FactorRanget   Redt   Greent   Bluec          C@  s   d t  f d „  ƒ  Y}  |  S(   Nt   TestBuilderc           B@  s<   e  Z i e ƒ  d  6Z e d ƒ Z e d ƒ Z d d g Z RS(   t   colort   xt   y(   t   __name__t
   __module__R   t   default_attributesR   R   R   t
   dimensions(    (    (    s:   lib/python2.7/site-packages/bkcharts/tests/test_builder.pyR   +   s   (   R   (   R   (    (    s:   lib/python2.7/site-packages/bkcharts/tests/test_builder.pyt   test_builder)   s    c         C@  s   |  | j  ƒ S(   N(   t   pd_data(   R   t	   test_data(    (    s:   lib/python2.7/site-packages/bkcharts/tests/test_builder.pyt   simple_builder5   s    c         C@  s   |  | j  d t ƒS(   Nt   palette(   R   t   CUSTOM_PALETTE(   R   R   (    (    s:   lib/python2.7/site-packages/bkcharts/tests/test_builder.pyt   custom_palette_builder:   s    c         C@  s7   |  ƒ  } | j  d  k s t ‚ | j d  k s3 t ‚ d  S(   N(   t   xlabelt   Nonet   AssertionErrort   ylabel(   R   t   builder(    (    s:   lib/python2.7/site-packages/bkcharts/tests/test_builder.pyt   test_empty_builder_labels?   s    	c         C@  s    |  j  d j t k s t ‚ d  S(   NR   (   t
   attributest   iterableR   R   (   R   (    (    s:   lib/python2.7/site-packages/bkcharts/tests/test_builder.pyt   test_default_colorE   s    c         C@  s    |  j  d j t k s t ‚ d  S(   NR   (   R"   R#   R   R   (   R   (    (    s:   lib/python2.7/site-packages/bkcharts/tests/test_builder.pyt   test_custom_colorI   s    c         C@  s_  t  ƒ  t  ƒ  t  ƒ  } } } d | f d | f d | f g } d t  ƒ  f d t  ƒ  f d t  ƒ  f g } i t  d d g d | ƒ d 6t  d d g d | ƒ d 6t  d d g d | ƒ d	 6} t j d  d  | | ƒ | k sİ t ‚ t j d d
 | | ƒ d | f d | f d | f g k st ‚ t j d d | | ƒ d | f d | f d | f g k s[t ‚ d  S(   Nt   col1t   col3t   col2t   columnst   seriest   itemsR   t   dasht   markert	   ascendingt
   descending(   R   R   t   _sort_legendR   R   (   R   R&   R(   R'   t   legendsR+   R"   (    (    s:   lib/python2.7/site-packages/bkcharts/tests/test_builder.pyt   test_legend_sortL   s    !*$*s   pandas not installedc         C@  sq  | j  j ƒ  } d d d d g | d <t j d d d d ƒ| d	 <t | d
 d ƒ} t | j d
 d d ƒ t ƒ sw t ‚ | j	 d k sŒ t ‚ t | d
 d ƒ} t | j d
 d d ƒ t
 ƒ sÂ t ‚ | j	 d k s× t ‚ t | d
 d ƒ} t | j d
 d d ƒ t
 ƒ st ‚ | j	 d k s"t ‚ t | d
 d	 ƒ} t | j d
 d d ƒ t ƒ sXt ‚ | j	 d k smt ‚ d  S(   Nt   at   bt   cR'   s   1950-01s   1950-05t   freqt   Mt   col4R   R&   i   i   t   linearR(   R   t   categoricali    t   datetime(   R   t   copyt   pdt
   date_rangeR   t
   isinstancet
   _get_rangeR   R   t   xscaleR	   (   R   R   R   t   builder1t   builder2t   builder3t   builder4(    (    s:   lib/python2.7/site-packages/bkcharts/tests/test_builder.pyt   test_created_range_type_   s    $$$$c         C@  sI   |  | j  d d d d ƒ}  |  j d k s0 t ‚ |  j d k sE t ‚ d  S(   Nt   legend_sort_fieldR   t   legend_sort_directionR.   (   R   RG   R   RH   (   R   R   (    (    s:   lib/python2.7/site-packages/bkcharts/tests/test_builder.pyt   test_sort_legendv   s    (%   t   __doc__t
   __future__R    t   mockR   t   pytestt   unittestR   t   pandasR=   t   Truet	   is_pandast   ImportErrort   et   Falset   bkcharts.builderR   R   t   bkcharts.propertiesR   t   bkcharts.attributesR   R   t   bokeh.modelsR   R	   R   t   fixtureR   R   R   R!   R$   R%   R2   RF   RI   (    (    (    s:   lib/python2.7/site-packages/bkcharts/tests/test_builder.pyt   <module>   s.   

				