3
pd+                 @   sj  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jj	Z
ddlmZ ddljZdd Zejdd Zdd	 Zejjd
dd Ze
jde
jdddd Zdd Zejejjdejjdejjddd Zejjddd Zejejjddd Zdd Zejjd d!ejd"d# Z ejjd$d%d& Z!ejjd$ejjd'd(d) Z"d*d+ Z#dS ),z1
Testing that we work in the downstream packages
    N)	DataFramec             C   s6   y
t j| S  tk
r0   tjd|  d Y nX d S )Nzskipping as z not available)	importlibimport_moduleModuleNotFoundErrorpytestskip)name r	   >/tmp/pip-build-7vycvbft/pandas/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   sJ   t d}t d}dd lj}|j| dd}|jd k	s6t|j d k	sFt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   zignore:Panel class is removedc             C   s   t d}| j 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| 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_nearest4   s    r   c               C   s   t jtjdddg d S )Nz-OOz-czimport pandas)
subprocess
check_callsys
executabler	   r	   r	   r
   test_oo_optimizableA   s    r"   z(ignore:pandas.util.testing is deprecatedzignore:can't:ImportWarningzGignore:Using or importing the ABCs from 'collections:DeprecationWarningc              C   sD   t d} dd lj}dd ljj}|jjddj}|jd|dj	  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_statsmodelsF   s
    

r)   c             C   sf   t d}ddlm}m} |j }|jddd}|j|jd d |jd d	  |j	|jd
d   d S )Nsklearnr   )r&   svmgMbP?g      Y@)gammaCr   r.   r.   )
r   r*   r&   r+   Zload_digitsZSVCr'   r$   targetZpredict)r   r*   r&   r+   digitsZclfr	   r	   r
   test_scikit_learnY   s     r1   ignorec              C   s&   t d} | jd}| jdd|d d S )NseaborntipsdayZ
total_bill)xyr$   )r   Zload_datasetZ	stripplot)r3   r4   r	   r	   r
   test_seabornf   s    
r8   c             C   s   t d}d S )N
pandas_gbq)r   )r   r9   r	   r	   r
   test_pandas_gbqo   s    r:   z10.8.1 tries to import urlencode from pd.io.common)reasonc              C   s   t d} | j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_datareadert   s    r>   z"ignore:can't resolve:ImportWarningc              C   s*   t d} | jjd}| j|d k	s&td S )N	geopandasZnaturalearth_lowres)r   r&   get_path	read_filer   )r?   fpr	   r	   r
   test_geopandas}   s    rC   z4ignore:RangeIndex.* is deprecated:DeprecationWarningc             C   s,   t d}|jj| }|j }tj||  d S )Npyarrow)r   ZTabler   Z	to_pandastmZassert_frame_equal)r   rD   tabler   r	   r	   r
   test_pyarrow   s    rG   c              C   s   t jjdd} | ddg}tj|j }d|kr8tjd | dddg}d	|  d
}tjtj	|d}tj|tj
d W d Q R X |jjj }xdD ]}||kstqW d S )N\/z-cz$import pandas;print(pandas.__file__)zsite-packagesz pandas installed as site packagez-sSEzimport pandaszCommand '\['zD', '-sSE', '-c', 'import pandas'\]' returned non-zero exit status 1.)match)stderrnumpypytzdateutil)rL   rM   rN   )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    


rZ   )$__doc__r   r   r    rL   npr   Zpandas.util._test_decoratorsutilZ_test_decoratorstdZpandasr   Zpandas._testingZ_testingrE   r   Zfixturer   r   markfilterwarningsr   Z
skip_if_nor   r"   networkr)   r1   r8   r:   Zxfailr>   rC   rG   rZ   r	   r	   r	   r
   <module>   s<   



	
	