U
|
¬ý°d} ã @ sb d Z ddlZddlZddlZddlZddlZddlmZ ddl m
|
m Z ddl Zddl mZmZ ddlmZ dd Zejdd Zd d
|
Zd d Ze d ¡dd Zdd Ze d¡e dd¡dd Z dd Z dd Z ej j ej dd Z!d d Z"ej j ej d d Z#d!d" Z$ej j ej ej j%e&d#d$d%d& Z'd'd( Z(d)d* Z)d+d, Z*d-d. Z+dS )/z1
|
Testing that we work in the downstream packages
|
é N)ÚIntCastingNaNError)Ú DataFrameÚSeriesc C s8 z t | ¡W S tk
|
r2 t d| d¡ Y nX d S )Nz skipping as z not available)Ú importlibÚ import_moduleÚModuleNotFoundErrorÚpytestÚskip)Úname© r úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/test_downstream.pyr s r c C s t ddddgiS )NÚAé é é )r r r r r Údf s r c
|
C sj t d¡}zLtd}td}dd lm} |j| dd}|jd k sDt| ¡ d k sTtW 5 t d|¡ X d S )Núcompute.use_numexprÚtoolzÚdaskr r ©Z npartitions)
|
ÚpdÚ
|
get_optionÚ
|
set_optionr Údask.dataframeÚ dataframeÚ from_pandasr ÚAssertionErrorÚcompute)r Úolduser r ÚddZddfr r r Ú test_dask% s
|
r c
|
C s t d¡} zftd}dd lm} dd lm} tddddg}|j |dd }|
|
|¡ ¡ }t
|
|¡}t ||¡ W 5 t d| ¡ X d S )
|
Nr r r g ø?gffffff@g @g @r r )r r r r Ú
|
dask.arrayÚarrayr r r r Zfixr ÚnpÚtmÚassert_series_equal)r r Údar ÚsZdsÚresultÚexpectedr r r Útest_dask_ufunc7 s
|
|
r* r c C sÊ dd l m} t dddg¡}| |¡}t|}t|}t ||¡ d}tj t
|
|d t|dd W 5 Q R X d }tj |d
|
< tj t |d t|dd W 5 Q R X tj t |d t|dd W 5 Q R X d S ) Nr r g @r z)Trying to coerce float values to integers©ÚmatchÚi8)Zdtypez9Cannot convert non-finite values \(NA or inf\) to integerr ) r r r# r" Z
|
from_arrayr r$ r% r ÚraisesÚ
|
ValueErrorÚnanr )r ZarrÚdarrÚresr) Úmsgr r r Ú7test_construct_dask_float_array_int_dtype_match_ndarrayK s
|
|
r4 c C s t d}| ¡ d k std S )NÚxarray)r Z to_xarrayr )r r5 r r r Ú test_xarrayf s r6 Úcftimer5 z0.21.0c C sP dd l } dd l}|jddd}| ddd¡}|j|gdd}d}||ksLtd S ) Nr Z0001r )ZperiodsiÐ r Znearest)Úmethod)r7 r5 Z cftime_rangeZDatetimeGregorianZ get_indexerr )r7 r5 ÚtimesÚkeyr( r) r r r Ú test_xarray_cftimeindex_nearestl s r; c C s t tjdddg¡ d S )Nú-OOú-cú import pandas©Ú
|
subprocessÚ
|
check_callÚsysÚ
|
executabler r r r Útest_oo_optimizablez s rD c C s t tjdddg¡ d S )Nr< r= z_import pandas as pd, pickle; pickle.loads(pickle.dumps(pd.date_range('2021-01-01', periods=1)))r? r r r r Ú)test_oo_optimized_datetime_index_unpickle s ûÿrE c C sL t d} dd lm} dd lm m} |j dd¡j}|jd|d ¡ d S )NÚ statsmodelsr ZGuerryZHistDataz$Lottery ~ Literacy + np.log(Pop1831))Údata)
|
r Zstatsmodels.apiÚapiZstatsmodels.formula.apiZformulaÚdatasetsZ get_rdatasetrG ZolsÚfit)rF ÚsmZsmfr r r r Útest_statsmodels s
|
rL c C sf t d} ddlm}m} | ¡ }|jddd}| |jd d
|jd d
¡ | |jdd
¡ d S )NÚsklearnr )rI Úsvmgü©ñÒMbP?g Y@)ÚgammaÚCéÿÿÿÿ)
|
r rM rI rN Z load_digitsZSVCrJ rG ÚtargetZpredict)rM rI rN ÚdigitsZclfr r r Útest_scikit_learn s rT c C s& t d} | d¡}| jdd|d d S )NÚseabornÚtipsÚdayZ
|
total_bill)ÚxÚyrG )r Z load_datasetZ stripplot)rU rV r r r Ú test_seaborn¦ s
|
rZ c C s t jddd td} d S )NÚ
|
pandas_gbqz0.10.0)Z
|
minversion)r Z importorskipr )r[ r r r Útest_pandas_gbq® s r\ z|The Quandl API key must be provided either through the api_key variable or through the environmental variable QUANDL_API_KEY)r. Úreasonc C s t d} | dddd¡ d S )NÚpandas_datareaderÚFZquandlz
|
2017-01-01z
|
2017-02-01)r Z
|
DataReader)r^ r r r Útest_pandas_datareader´ s r` c C s, t d}|j | ¡}| ¡ }t || ¡ d S )NÚpyarrow)r ZTabler Z to_pandasr$ Úassert_frame_equal)r ra Útabler( r r r Ú test_pyarrowÀ s rd c C sN t d}| | ¡}|j||jd}t | |¡ |j||jd}t | |¡ d S )NÚyaml)ÚLoader)r ÚdumpÚloadrf r$ rb Z UnsafeLoader)r re ZdumpedZloadedZloaded2r r r Útest_yaml_dumpÇ s
|
ri c C s¢ t j dd¡} | ddg}t |¡ ¡ }d|kr8t d¡ | dddg}d | d
|
}tjtj |d }tj|tj
|
d W 5 Q R X |j j ¡ }d D ]}||kst 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.r+ )Ústderr)ÚnumpyZpytzZdateutil)rB rC Úreplacer@ Ú check_outputÚdecoder r r. ÚCalledProcessErrorÚSTDOUTÚvalueÚstdoutr )ZpyexeÚcallÚoutputr3 Úexcr
|
r r r Ú test_missing_required_dependencyÔ s
|
|
|
ÿ rx c
|
C s² t d¡} ztd}dd lm} | ddg¡}tdddgi}||d< ||d< d |jd
|
d gdf< |jdgd d
f }tdgd gdgd dgd }t ||¡ W 5 t d| ¡ X d S )Nr r r r r ÚaÚbÚcéd FT)ry rz r{ )Úindex)
|
r r r r r! r" r Úlocr$ rb )r r r& Zddar r( r) r r r Ú*test_frame_setitem_dask_array_into_new_colø s
|
r ),Ú__doc__r r@ rB rm r# r Z pandas.errorsr Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr r r Zpandas._testingZ_testingr$ r Zfixturer r r* Z
|
skip_if_nor4 r6 r; rD rE ÚmarkÚnetworkrL rT rZ r\ Zxfailr/ r` rd ri rx r r r r r Ú<module> sT
|
|
|
|
þ $
|
