ó
šxŠ\c           @   sÆ  d  Z  d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z m	 Z	 m
 Z
 d d l m Z d d l m Z d „  Z e j d „  ƒ Z d „  Z d	 „  Z e j j e
 ƒ  oÀ e d
 d ƒd „  ƒ Z e j e j j d ƒ d „  ƒ ƒ Z e j j d ƒ d „  ƒ Z e j e j j d ƒ d „  ƒ ƒ Z d „  Z e j j d
 d ƒ e j d „  ƒ ƒ Z e j j d ƒ e j j d ƒ e j j d ƒ d „  ƒ ƒ ƒ Z  e j j d ƒ d „  ƒ Z! d S(   s1   
Testing that we work in the downstream packages
iÿÿÿÿN(   t   PY2t   PY36t   is_platform_windows(   t	   DataFrame(   t   testingc         C   s¯   t  rD y t j |  ƒ SWq« t k
 r@ t j d j |  ƒ ƒ q« Xng y t j |  ƒ SWnS t k
 rª } d t | ƒ k r¤ |  t | ƒ k r¤ t j d j |  ƒ ƒ n  ‚  n Xd  S(   Ns   skipping as {} not availables   No module named(	   R   t	   importlibt   import_modulet   ModuleNotFoundErrort   pytestt   skipt   formatt   ImportErrort   str(   t   namet   e(    (    s;   lib/python2.7/site-packages/pandas/tests/test_downstream.pyR      s    $c           C   s   t  i d d d g d 6ƒ S(   Ni   i   i   t   A(   R   (    (    (    s;   lib/python2.7/site-packages/pandas/tests/test_downstream.pyt   df%   s    c         C   sm   t  d ƒ } t  d ƒ } d d  l j } | j |  d d ƒ} | j d  k	 sQ t ‚ | j ƒ  d  k	 si t ‚ d  S(   Nt   toolzt   daskiÿÿÿÿt   npartitionsi   (   R   t   dask.dataframet	   dataframet   from_pandasR   t   Nonet   AssertionErrort   compute(   R   R   R   t   ddt   ddf(    (    s;   lib/python2.7/site-packages/pandas/tests/test_downstream.pyt	   test_dask*   s    c         C   s(   t  d ƒ } |  j ƒ  d  k	 s$ t ‚ d  S(   Nt   xarray(   R   t	   to_xarrayR   R   (   R   R   (    (    s;   lib/python2.7/site-packages/pandas/tests/test_downstream.pyt   test_xarray6   s    t   reasons   Broken on Windows / Py2c           C   s    t  j t j d d d g ƒ d  S(   Ns   -OOs   -cs   import pandas(   t
   subprocesst
   check_callt   syst
   executable(    (    (    s;   lib/python2.7/site-packages/pandas/tests/test_downstream.pyt   test_oo_optimizable=   s    s   ignore:can't:ImportWarningc          C   sb   t  d ƒ }  d d  l j } d d  l j j } | j j d d ƒ j } | j d d | ƒj	 ƒ  d  S(   Nt   statsmodelsiÿÿÿÿt   Guerryt   HistDatas$   Lottery ~ Literacy + np.log(Pop1831)t   data(
   R   t   statsmodels.apit   apit   statsmodels.formula.apit   formulat   datasetst   get_rdatasetR)   t   olst   fit(   R&   t   smt   smfR   (    (    s;   lib/python2.7/site-packages/pandas/tests/test_downstream.pyt   test_statsmodelsD   s
    c         C   s|   t  d ƒ } d d l m } m } | j ƒ  } | j d d d d ƒ } | j | j d  | j d  ƒ | j	 | j d ƒ d  S(   Nt   sklearniÿÿÿÿ(   t   svmR.   t   gammagü©ñÒMbP?t   Cg      Y@(
   R   R5   R6   R.   t   load_digitst   SVCR1   R)   t   targett   predict(   R   R5   R6   R.   t   digitst   clf(    (    s;   lib/python2.7/site-packages/pandas/tests/test_downstream.pyt   test_scikit_learnQ   s    t   ignorec          C   s;   t  d ƒ }  |  j d ƒ } |  j d d d d d | ƒ d  S(   Nt   seabornt   tipst   xt   dayt   yt
   total_billR)   (   R   t   load_datasett	   stripplot(   RA   RB   (    (    s;   lib/python2.7/site-packages/pandas/tests/test_downstream.pyt   test_seaborn^   s    c         C   s   t  d ƒ } d  S(   Nt
   pandas_gbq(   R   (   R   RJ   (    (    s;   lib/python2.7/site-packages/pandas/tests/test_downstream.pyt   test_pandas_gbqg   s    s   0.7.0 pendingc          C   s&   t  d ƒ }  |  j d d d d ƒ d  S(   Nt   pandas_datareadert   Ft   quandls
   2017-01-01s
   2017-02-01(   R   t
   DataReader(   RL   (    (    s;   lib/python2.7/site-packages/pandas/tests/test_downstream.pyt   test_pandas_datareaderl   s    s$   ignore:The 'warn':DeprecationWarnings%   ignore:pandas.util:DeprecationWarnings"   ignore:can't resolve:ImportWarningc          C   s=   t  d ƒ }  |  j j d ƒ } |  j | ƒ d  k	 s9 t ‚ d  S(   Nt	   geopandast   naturalearth_lowres(   R   R.   t   get_patht	   read_fileR   R   (   RQ   t   fp(    (    s;   lib/python2.7/site-packages/pandas/tests/test_downstream.pyt   test_geopandasv   s    c         C   s>   t  d ƒ } | j j |  ƒ } | j ƒ  } t j | |  ƒ d  S(   Nt   pyarrow(   R   t   TableR   t	   to_pandast   tmt   assert_frame_equal(   R   RW   t   tablet   result(    (    s;   lib/python2.7/site-packages/pandas/tests/test_downstream.pyt   test_pyarrow   s    ("   t   __doc__R   R!   R#   t   numpyt   npR   t   pandas.compatR    R   R   t   pandasR   t   pandas.utilR   RZ   R   t   fixtureR   R   R   t   markt   skipifR%   t   networkt   filterwarningsR4   R?   RI   RK   t   xfailRP   RV   R^   (    (    (    s;   lib/python2.7/site-packages/pandas/tests/test_downstream.pyt   <module>   s4   					!	