ó
¦–Õ\c           @   s§   d  d l  j j Z d  d l Z d  d l Z d  d l Z d  d l j	 Z
 d  d l m Z d „  Z d „  Z d „  Z e j j e j j d „  ƒ ƒ Z d „  Z d „  Z d S(	   iÿÿÿÿN(   t	   assert_eqc       
   C   s‚  t  j j d d i t d 6t d 6t d 6d d d d	 ƒ}  |  j d
 t j d d d	 ƒk sa t	 ‚ |  j d t j d d d	 ƒk s‰ t	 ‚ t
 j |  j t j d d d g ƒ ƒ |  d j ƒ  j t k sÍ t	 ‚ |  d j ƒ  j t k sì t	 ‚ |  d j ƒ  j t k st	 ‚ |  j j d k s#t	 ‚ |  j ƒ  j j |  j j k sGt	 ‚ |  j t t j d d d d d d	 ƒ ƒ k szt	 ‚ t
 j |  j ƒ  |  j ƒ  ƒ t  j j d d i t d 6t d 6t d 6d d d d	 d d ƒ} t  j j d d i t d 6t d 6t d 6d d d d	 d d ƒ} t  j j d d i t d 6t d 6t d 6d d d d	 d d ƒ} t  j j d d i t d 6t d 6t d 6d d d d d d ƒ} t  j j d d i t d 6t d 6t d 6d d d d	 d d ƒ} t
 j | j ƒ  | j ƒ  ƒ | j d ƒ | j d ƒ k j ƒ  j ƒ  st	 ‚ | j | j k s6t	 ‚ | j | j k sNt	 ‚ | j | j k sft	 ‚ | j | j k s~t	 ‚ d  S(   Nt   2000t   2015t   At   Bt   Ct   freqt   2Dt   partition_freqt   6Mi    s
   2000-01-31iÿÿÿÿs
   2014-07-31t	   timestampt   startt   endt   seedi{   iÈ  t   3Mt   1Di
   (   t   ddt   demot   make_timeseriest   floatt   intt   strt	   divisionst   pdt	   Timestampt   AssertionErrort   tmt   assert_index_equalt   columnst   Indext   headt   dtypet   objectt   indext   namet   tuplet
   date_ranget   assert_frame_equalt   allt   _name(   t   dft   at   bt   ct   dt   e(    (    s@   lib/python2.7/site-packages/dask/dataframe/io/tests/test_demo.pyt   test_make_timeseries
   s8    *((%$3*****1c          C   st   t  j j ƒ  }  d |  j k  o) d k  n s4 t ‚ t |  j ƒ d k sO t ‚ t t |  j ƒ ƒ d k sp t ‚ d  S(   Ni   iè  (	   R   R   R   t   npartitionsR   t   lenR   t   sett   dtypes(   R(   (    (    s@   lib/python2.7/site-packages/dask/dataframe/io/tests/test_demo.pyt   test_make_timeseries_no_args,   s    %c              s^   t  j j d d i t d 6d d d d ƒ‰  t ‡  f d †  t ˆ  j d	 ƒ Dƒ ƒ sZ t ‚ d  S(
   NR   t   2001R   R   t   3HR   R   c         3   sO   |  ]E } ˆ  j  | ƒ j j ƒ  j ƒ  ˆ  j  | d  ƒ j j ƒ  j ƒ  k  Vq d S(   i   N(   t   get_partitionR!   t   maxt   computet   min(   t   .0t   i(   R(   (    s@   lib/python2.7/site-packages/dask/dataframe/io/tests/test_demo.pys	   <genexpr>7   s   i   (   R   R   R   R   R&   t   rangeR/   R   (    (    (   R(   s@   lib/python2.7/site-packages/dask/dataframe/io/tests/test_demo.pyt   test_no_overlaps3   s    c          C   s   t  j d ƒ t j j d d d d d d d ƒ}  t |  t j ƒ sI t ‚ d	 |  j k  oc d
 k  n sn t ‚ t	 |  |  ƒ d  S(   Nt   pandas_datareadert   GOOGR   s
   2010-01-01t   stops
   2010-01-30R   t   1hi
   i   (
   t   pytestt   importorskipR   R   t   daily_stockt
   isinstancet	   DataFrameR   R/   R    (   R(   (    (    s@   lib/python2.7/site-packages/dask/dataframe/io/tests/test_demo.pyt   test_daily_stock<   s
    $%c          C   sÉ   t  j j d d i t d 6t d 6t d 6d d d d	 d
 d d d ƒ}  |  j j ƒ  } |  j j ƒ  } t j	 | | d d ƒ\ } } d | k  o˜ d k n s£ t
 ‚ d | k  oº d k n sÅ t
 ‚ d  S(   NR   R4   R   R   R   R   R   R   R	   t   A_lami@B t   B_lami   t	   schedulers   single-threadedid   i€–˜ i   (   R   R   R   R   R   R   t   nuniqueR   t   daskR8   R   (   R(   t   a_cardinalityt   b_cardinalityt   aat   bb(    (    s@   lib/python2.7/site-packages/dask/dataframe/io/tests/test_demo.pyt   test_make_timeseries_keywordsF   s    *"c          C   sÉ   t  j j d d i t d 6t d 6t d 6d d d d	 d
 d d d ƒ}  |  j j ƒ  } |  j j ƒ  } t j	 | | d d ƒ\ } } d | k  o˜ d k n s£ t
 ‚ d | k  oº d k n sÅ t
 ‚ d  S(   NR   R4   t   A_Bt   B_R   R   R   R   R	   t   A_B_lami@B t   B__lami   RJ   s   single-threadedid   i€–˜ i   (   R   R   R   R   R   RR   RK   RS   RL   R8   R   (   R(   RM   RN   RO   RP   (    (    s@   lib/python2.7/site-packages/dask/dataframe/io/tests/test_demo.pyt#   test_make_timeseries_fancy_keywordsR   s    	"(   t   pandas.util.testingt   utilt   testingR   t   pandasR   RB   RL   t   dask.dataframet	   dataframeR   t   dask.dataframe.utilsR    R.   R3   R=   t   markt   xfailt   networkRG   RQ   RV   (    (    (    s@   lib/python2.7/site-packages/dask/dataframe/io/tests/test_demo.pyt   <module>   s   	"						