B
    •xŠ\ë  ã               @   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Zm	Z	m
Z
 ddlmZ ddlmZ dd„ Zejdd	„ ƒZd
d„ Zdd„ Zejje
ƒ oŒedddd„ ƒZej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¡ej d ¡ej d!¡d"d#„ ƒƒƒZ ej d!¡d$d%„ ƒZ!dS )&z1
Testing that we work in the downstream packages
é    N)ÚPY2ÚPY36Úis_platform_windows)Ú	DataFrame)Útestingc          
   C   s’   t r6y
t | ¡S  tk
r2   t d | ¡¡ Y qŽX nXy
t | ¡S  tk
rŒ } z.dt|ƒkrz| t|ƒkrzt d | ¡¡ ‚ W d d }~X Y nX d S )Nzskipping as {} not availablezNo module named)	r   Ú	importlibÚimport_moduleÚModuleNotFoundErrorÚpytestÚskipÚformatÚImportErrorÚstr)ÚnameÚe© r   ú;lib/python3.7/site-packages/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   Z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_xarray6   s    r   zBroken on Windows / Py2)Úreasonc               C   s   t  tjdddg¡ d S )Nz-OOz-czimport pandas)Ú
subprocessZ
check_callÚsysÚ
executabler   r   r   r   Útest_oo_optimizable=   s    r#   zignore:can't:ImportWarningc              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.apiZapiZstatsmodels.formula.apiZformulaÚdatasetsZget_rdatasetr%   ZolsÚfit)r$   ZsmZsmfr   r   r   r   Útest_statsmodelsD   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   )Úsvmr&   gü©ñÒMbP?g      Y@)ZgammaÚCéÿÿÿÿ)
r   r)   r*   r&   Zload_digitsZSVCr'   r%   ÚtargetZpredict)r   r)   r*   r&   ZdigitsZclfr   r   r   Útest_scikit_learnQ   s     r.   Úignorec              C   s&   t dƒ} |  d¡}| jdd|d d S )NÚseabornÚtipsZdayZ
total_bill)ÚxÚyr%   )r   Zload_datasetZ	stripplot)r0   r1   r   r   r   Útest_seaborn^   s    
r4   c             C   s   t dƒ}d S )NÚ
pandas_gbq)r   )r   r5   r   r   r   Útest_pandas_gbqg   s    r6   z0.7.0 pendingc              C   s   t dƒ} |  dddd¡ d S )NÚpandas_datareaderÚFZquandlz
2017-01-01z
2017-02-01)r   Z
DataReader)r7   r   r   r   Útest_pandas_datareaderl   s    r9   z$ignore:The 'warn':DeprecationWarningz%ignore:pandas.util:DeprecationWarningz"ignore:can't resolve:ImportWarningc              C   s*   t dƒ} | j d¡}|  |¡d k	s&t‚d S )NÚ	geopandasZnaturalearth_lowres)r   r&   Zget_pathZ	read_filer   )r:   Úfpr   r   r   Útest_geopandasv   s    r<   c             C   s,   t dƒ}|j | ¡}| ¡ }t || ¡ d S )NÚpyarrow)r   ZTabler   Z	to_pandasÚtmZassert_frame_equal)r   r=   ÚtableÚresultr   r   r   Útest_pyarrow   s    rA   )"Ú__doc__r   r    r!   ZnumpyZnpr
   Zpandas.compatr   r   r   Zpandasr   Zpandas.utilr   r>   r   Zfixturer   r   r   ZmarkZskipifr#   ZnetworkÚfilterwarningsr(   r.   r4   r6   Zxfailr9   r<   rA   r   r   r   r   Ú<module>   s4   	

	