U
    f/es                     @   sl  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  m	Z
 ddlmZ ddlmZ dd Zejdd Ze
jdd	 Zd
d Ze
de
dddd Zdd Zejejdejdejddd Zejddd Zejejddd Zdd Zejj dd ejd!d" Z!ejd#d$d% Z"ejd#ejd&d'd( Z#d)d* Z$dS )+z1
Testing that we work in the downstream packages
    N	DataFramec                 C   s8   zt | W S  tk
r2   td|  d Y nX d S )Nzskipping as z not available)	importlibimport_moduleModuleNotFoundErrorpytestskip)name r
   @/tmp/pip-unpacked-wheel-tiezk1ph/pandas/tests/test_downstream.pyr      s    r   c                   C   s   t ddddgiS )NA         r   r
   r
   r
   r   df   s    r   c                 C   sL   t d}t d}dd lm} |j| dd}|jd k	s8t| d k	sHtd S )Ntoolzdaskr   r   )Znpartitions)r   Zdask.dataframeZ	dataframefrom_pandasr   AssertionErrorZcompute)r   r   r   ddZddfr
   r
   r   	test_dask"   s    r   c                 C   s   t d}|  d k	std S )Nxarray)r   Z	to_xarrayr   )r   r   r
   r
   r   test_xarray/   s    r   cftimer   z0.10.4c                  C   sJ   dd l } dd l}|jddd}|j| ddddd}d}||ksFtd S )	Nr   Z0001r   )Zperiodsi  r   Znearest)method)r   r   Zcftime_rangeZget_locZDatetimeGregorianr   )r   r   timesresultexpectedr
   r
   r   test_xarray_cftimeindex_nearest6   s    r   c                   C   s   t tjdddg d S )Nz-OO-cimport pandas)
subprocess
check_callsys
executabler
   r
   r
   r   test_oo_optimizableC   s    r%   z(ignore:pandas.util.testing is deprecatedzignore:can't:ImportWarningzGignore:Using or importing the ABCs from 'collections:DeprecationWarningc                  C   sL   t d} dd lm} dd lm  m} |jddj}|jd|d	  d S )Nstatsmodelsr   ZGuerryZHistDataz$Lottery ~ Literacy + np.log(Pop1831))data)
r   Zstatsmodels.apiapiZstatsmodels.formula.apiZformuladatasetsZget_rdatasetr'   Zolsfit)r&   smZsmfr   r
   r
   r   test_statsmodelsH   s
    
r,   c                 C   sf   t d}ddlm}m} | }|jddd}||jd d |jd d  |	|jdd   d S )Nsklearnr   )r)   svmgMbP?g      Y@)gammaC)
r   r-   r)   r.   Zload_digitsZSVCr*   r'   targetZpredict)r   r-   r)   r.   digitsZclfr
   r
   r   test_scikit_learn[   s     r4   ignorec                  C   s&   t d} | d}| jdd|d d S )NseaborntipsdayZ
total_bill)xyr'   )r   Zload_datasetZ	stripplot)r6   r7   r
   r
   r   test_seabornk   s    
r;   c                 C   s   t d}d S )N
pandas_gbq)r   )r   r<   r
   r
   r   test_pandas_gbqt   s    r=   z10.8.1 tries to import urlencode from pd.io.common)reasonc                  C   s   t d} | dddd d S )Npandas_datareaderFZquandlz
2017-01-01z
2017-02-01)r   Z
DataReader)r?   r
   r
   r   test_pandas_datareadery   s    rA   z"ignore:can't resolve:ImportWarningc                  C   s*   t d} | jd}| |d k	s&td S )N	geopandasZnaturalearth_lowres)r   r)   get_path	read_filer   )rB   fpr
   r
   r   test_geopandas   s    rF   z4ignore:RangeIndex.* is deprecated:DeprecationWarningc                 C   s,   t d}|j| }| }t||  d S )Npyarrow)r   ZTabler   Z	to_pandastmZassert_frame_equal)r   rG   tabler   r
   r
   r   test_pyarrow   s    rJ   c               	   C   s   t jdd} | ddg}t| }d|kr8td | dddg}d	|  d
}tjtj	|d}tj|tj
d W 5 Q R X |jj }dD ]}||kstqd S )N\/r   z$import pandas;print(pandas.__file__)zsite-packagesz pandas installed as site packagez-sSEr    zCommand '\['zD', '-sSE', '-c', 'import pandas'\]' returned non-zero exit status 1.)match)stderr)numpyZpytzZdateutil)r#   r$   replacer!   check_outputdecoder   r   ZraisesCalledProcessErrorSTDOUTvaluestdoutr   )Zpyexecalloutputmsgexcr	   r
   r
   r    test_missing_required_dependency   s    


r[   )%__doc__r   r!   r#   rO   npr   Zpandas.util._test_decoratorsutilZ_test_decoratorstdZpandasr   Zpandas._testingZ_testingrH   r   Zfixturer   Z&skip_array_manager_not_yet_implementedr   r   Z
skip_if_nor   r%   networkmarkfilterwarningsr,   r4   r;   r=   ZxfailrA   rF   rJ   r[   r
   r
   r
   r   <module>   sP   












