B @`@sfdZddlZddlZddlZddlZddlZddlmm Z ddl m Z ddl mZddZejddZdd Zd d Ze d e d dddZddZejejdejdejdddZejdddZejejdddZddZejjdd ejd!d"Z ejd#d$d%Z!ejd#ejd&d'd(Z"d)d*Z#dS)+z1 Testing that we work in the downstream packages N) DataFramecCs6y t|Stk r0td|dYnXdS)Nz skipping as z not available) importlib import_moduleModuleNotFoundErrorpytestskip)namer @/tmp/pip-unpacked-wheel-q9tj5l6a/pandas/tests/test_downstream.pyrs rcCstddddgiS)NA)rr r r r dfsrcCsLtd}td}ddlm}|j|dd}|jdk s8t|dk sHtdS)Ntoolzdaskrr)Z npartitions)rZdask.dataframeZ dataframe from_pandasr AssertionErrorZcompute)rrrddZddfr r r test_dask s  rcCstd}|dk stdS)Nxarray)rZ to_xarrayr)rrr r r test_xarray,srcftimerz0.10.4cCsJddl}ddl}|jddd}|j|ddddd}d}||ksFtdS) NrZ0001r )Zperiodsir Znearest)method)rrZ cftime_rangeZget_locZDatetimeGregorianr)rrtimesresultexpectedr r r test_xarray_cftimeindex_nearest3s rcCsttjdddgdS)Nz-OOz-cz import pandas) subprocess check_callsys executabler r r r test_oo_optimizable@sr"z(ignore:pandas.util.testing is deprecatedzignore:can't:ImportWarningzGignore:Using or importing the ABCs from 'collections:DeprecationWarningcCsLtd}ddlm}ddlmm}|jddj}|jd|d dS)N statsmodelsrZGuerryZHistDataz$Lottery ~ Literacy + np.log(Pop1831))data) rZstatsmodels.apiapiZstatsmodels.formula.apiZformuladatasetsZ get_rdatasetr$Zolsfit)r#smZsmfrr r r test_statsmodelsEs  r)cCsftd}ddlm}m}|}|jddd}||jdd|jdd| |jdddS)Nsklearnr)r&svmgMbP?gY@)gammaC) rr*r&r+Z load_digitsZSVCr'r$targetZpredict)rr*r&r+digitsZclfr r r test_scikit_learnXs  r1ignorecCs&td}|d}|jdd|ddS)NseaborntipsdayZ total_bill)xyr$)rZ load_datasetZ stripplot)r3r4r r r test_seabornes r8cCs td}dS)N pandas_gbq)r)rr9r r r test_pandas_gbqnsr:z10.8.1 tries to import urlencode from pd.io.common)reasoncCstd}|dddddS)Npandas_datareaderFZquandlz 2017-01-01z 2017-02-01)rZ DataReader)r<r r r test_pandas_datareaderssr>z"ignore:can't resolve:ImportWarningcCs*td}|jd}||dk s&tdS)N geopandasZnaturalearth_lowres)rr&get_path read_filer)r?fpr r r test_geopandas|s rCz4ignore:RangeIndex.* is deprecated:DeprecationWarningcCs,td}|j|}|}t||dS)Npyarrow)rZTablerZ to_pandastmZassert_frame_equal)rrDtablerr r r test_pyarrows rGc Cstjdd}|ddg}t|}d|kr8td|dddg}d |d }tjtj |d }tj|tj d WdQRX|j j }xd D]}||kst qWdS)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)numpyZpytzZdateutil)r r!replacer check_outputdecoderrZraisesCalledProcessErrorSTDOUTvaluestdoutr)Zpyexecalloutputmsgexcrr r r test_missing_required_dependencys      rX)$__doc__rrr rLnprZpandas.util._test_decoratorsutilZ_test_decoratorstdZpandasrZpandas._testingZ_testingrErZfixturerrrZ skip_if_norr"networkmarkfilterwarningsr)r1r8r:Zxfailr>rCrGrXr r r r s<