3
���h+ � @ sj d Z ddlZddlZddlZddlZddlZddljj Z
ddl m Z ddl
jZdd� Zejdd� �Zdd � Zejjd
�d d � �Ze
jd
�e
jdd�dd� ��Zdd� Zejejjd�ejjd�ejjd�dd� ����Zejjd�dd� �Z ejejjd�d d � ��Z d d � Z ejj d d!�ejd"d#� ��Z ejjd$�d%d&� �Z!ejjd$�ejjd'�d(d)� ��Z"d*d+� Z#dS ),z1
Testing that we work in the downstream packages
� N)� DataFramec C s6 y
t j| �S tk
r0 tjd| � d�� Y nX d S )Nz skipping as z not available)� importlib�
import_module�ModuleNotFoundError�pytest�skip)�name� r �>/tmp/pip-build-5_djhm0z/pandas/pandas/tests/test_downstream.pyr s
r c C s t ddddgi�S )N�A� � � )r r r r r
�df s r c C sJ t d�}t d�}dd lj}|j| dd�}|jd k s6t�|j� d k sFt�d S )N�toolz�daskr r )Z npartitions)r Zdask.dataframeZ dataframe� from_pandasr �AssertionErrorZcompute)r r r �ddZddfr r r
� test_dask s
r z ignore:Panel class is removedc C s t d�}| j� d k st�d S )N�xarray)r Z to_xarrayr )r r r r r
� test_xarray, s r �cftimer z0.10.4c C sJ dd l } dd l}|jddd�}|j| jddd�dd�}d}||ksFt�d S ) Nr Z0001r
)Zperiodsi� r Znearest)�method)r r Z cftime_rangeZget_locZDatetimeGregorianr )r r �times�result�expectedr r r
� test_xarray_cftimeindex_nearest4 s r c C s t jtjdddg� 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 deprecatedzignore:can't:ImportWarningzGignore:Using or importing the ABCs from 'collections:DeprecationWarningc C sD t d�} dd lj}dd ljj}|jjdd�j}|jd|d�j � d S )N� statsmodelsr ZGuerryZHistDataz$Lottery ~ Literacy + np.log(Pop1831))�data)
r Zstatsmodels.api�apiZstatsmodels.formula.apiZformula�datasetsZ get_rdatasetr$ Zols�fit)r# �smZsmfr r r r
�test_statsmodelsF s
r) c C sf t d�}ddlm}m} |j� }|jddd�}|j|jd d� |jd d � � |j |jd
d � � d S ) N�sklearnr )r& �svmg����MbP?g Y@)�gamma�Cr ���r. r. )
r r* r& r+ Z load_digitsZSVCr' r$ �targetZpredict)r r* r& r+ �digitsZclfr r r
�test_scikit_learnY s r1 �ignorec C s& t d�} | jd�}| jdd|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�} | jdddd� d S )N�pandas_datareader�FZquandlz
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�} | jjd�}| j|�d k s&t�d S )N� geopandasZnaturalearth_lowres)r r& �get_path� read_filer )r? �fpr r r
�test_geopandas} s rC z4ignore:RangeIndex.* is deprecated:DeprecationWarningc C s, t d�}|jj| �}|j� }tj|| � d S )N�pyarrow)r ZTabler Z to_pandas�tmZassert_frame_equal)r rD �tabler r r r
� test_pyarrow� s rG c C s� t jjdd�} | ddg}tj|�j� }d|kr8tjd� | dddg}d | � d
�}tjtj |d ��}tj|tj
d � W d Q R X |j j j� }xdD ]}||ks�t
�q�W 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 Zraises�CalledProcessError�STDOUT�value�stdoutr )Zpyexe�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�tdZpandasr Zpandas._testingZ_testingrE r Zfixturer r �mark�filterwarningsr Z
skip_if_nor r" �networkr) r1 r8 r: Zxfailr> rC rG rZ r r r r
�<module> s<
|