3
���h[! � @ s� d dl Z d dlZd dlmZ d dlZd dlZd dlmZmZ d dl j
Z G dd� d�Z G dd� de �Z
dd � Zd
d � Zd d
� ZG dd� de �ZG dd� de �ZdS )� N)�List)�api�compatc @ s e Zd Zddd�ZdS )�BaseNc C sJ t dd� t|�D ��}|d k r2t t|�t|� �}t |�}tj||� d S )Nc s s | ]}|j d �s|V qdS )�__N)�
startswith)�.0�f� r
�;/tmp/pip-build-5_djhm0z/pandas/pandas/tests/api/test_api.py� <genexpr> s z Base.check.<locals>.<genexpr>)�sorted�dir�set�tmZassert_almost_equal)�self� namespace�expected�ignored�resultr
r
r �check
s
z
Base.check)N)�__name__�
__module__� __qualname__r r
r
r
r r s r c ' @ s� e Zd ZU dddgZdddddd d
d d d
dddg
ZddgZee dddgZddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<g'Z g Z
ee
d=gZ ee e j
s�e jd>d?d@g� g Zee dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^g Zd_d`dadbdcddgZdedfdgdhdidjdkdldmdndodpdqdrdsdtdudvdwgZdxgZdydzd{d|gZg Zee g Zee d}d~dd�d�d�d�d�d�d�d�d�g Zd�d�� Zd�d�� Zd�S )�� TestPDApi�tests�localeZconftestr Zarraysr �core�errors�pandasZplotting�test�testingZtseries�util�options�io�np�datetimeZ
IndexSliceZNaTZNAZ CategoricalZCategoricalIndexZ DataFrameZ
DateOffsetZ
DatetimeIndexZ ExcelFileZ ExcelWriterZ Float64IndexZGrouperZHDFStore�IndexZ
Int64IndexZ
MultiIndexZPeriodZ PeriodIndexZ
RangeIndexZ UInt64IndexZSeriesZ SparseDtypeZ StringDtypeZ TimedeltaZTimedeltaIndexZ TimestampZIntervalZ
IntervalIndexZCategoricalDtypeZ PeriodDtypeZ
IntervalDtypeZDatetimeTZDtypeZ BooleanDtypeZ Int8DtypeZ
Int16DtypeZ
Int32DtypeZ
Int64DtypeZ
UInt8DtypeZ UInt16DtypeZ UInt32DtypeZ UInt64DtypeZNamedAgg� SparseArrayZPanelZ SparseSeriesZSparseDataFrame�arrayZ bdate_range�concatZcrosstabZcutZ
date_rangeZinterval_range�evalZ factorizeZ get_dummiesZ
infer_freqZisnaZisnullZlreshapeZmeltZnotnaZnotnull�offsets�mergeZ
merge_orderedZ
merge_asofZ period_rangeZpivotZ pivot_tableZqcutZ
show_versionsZtimedelta_range�uniqueZ value_countsZ wide_to_longZ reset_optionZdescribe_option�
get_optionZoption_contextZ
set_optionZset_eng_float_formatZread_clipboardZread_csvZ
read_excelZread_fwfZread_gbqZread_hdfZ read_htmlZ read_jsonZ read_pickleZread_sasZread_sqlZread_sql_queryZread_sql_tableZ
read_stataZ
read_tableZ read_featherZ read_parquetZread_orcZ read_spssZjson_normalizeZ to_datetimeZ
to_numericZ to_pickleZ to_timedeltaZ_configZ
_hashtableZ_libZ_libsZ_np_version_under1p16Z_np_version_under1p17Z_np_version_under1p18Z
_is_numpy_dev�_testingZ_tslibZ_typing�_versionc C sz | j | j | j | j | j | j | j | j | j | j }t
j sf|j | j
| j | j | j | j � | jt|| j� d S )N)�lib�misc�modules�classes�funcs� funcs_option�
funcs_read�
funcs_json�funcs_to�private_modulesr �PY37�extend�deprecated_modules�deprecated_classes� deprecated_classes_in_future�deprecated_funcs_in_future�deprecated_funcsr �pdr )r Z
checktheser
r
r �test_api� s
< zTestPDApi.test_apic C s� | j | j | j | j | j }xv|D ]n}tjt��Z tt |�}t
j s�|dkrd|j t
t jj�d � n$|dkrv|g � n|j t
|�d � W d Q R X q$W d S )Nr& � r( ���rF )r> r? r@ rB rA r �assert_produces_warning�
FutureWarning�getattrrC r r< � __getattr__r r& )r Zdeprecated_listZdeprZ
deprecatedr
r
r � test_depr� s
zTestPDApi.test_deprN)r r r r r2 r> r �strr3 r5 r? r@ r r< r= r4 r6 r7 r8 r9 r: rA rB r; rD rK r
r
r
r r s
r c C s| ddl m } dd l}|j� �V |jdt� | ddddd�tj ddddd�ksPt�ttj ddddd�tj �snt�W d Q R X d S )Nr )r& �ignorei� rE � )r& �warnings�catch_warnings� simplefilterrH rC �AssertionError�
isinstance)r& rO r
r
r �
test_datetime� s
&rT c C sL dd l } | j� �2 | jdt� ttjdddgdd�tj�s>t�W d Q R X d S )Nr rM rE rN � ZSparse)Zdtype) rO rP rQ rH rS rC r) r( rR )rO r
r
r �test_sparsearray s
rV c C sV dd l } dd l}| j� �4 | jdt� tjjdd�|jdd�kj� sHt �W d Q R X d S )Nr rM �
)
rO ZnumpyrP rQ rH rC r% Zarange�allrR )rO r% r
r
r �test_np
s
rY c @ s e Zd ZdddgZdd� ZdS )�TestApi�types�
extensionsZindexersc C s | j t| j� d S )N)r r �allowed)r r
r
r rD s zTestApi.test_apiN)r r r r] rD r
r
r
r rZ s
rZ c @ s8 e Zd ZddddgZdd� Zdd� Zd d
� Zd d � Zd
S )� TestTestingZassert_frame_equal�assert_series_equalZassert_index_equalZ assert_extension_array_equalc C s ddl m} | j|| j� d S )Nr )r! )r r! r r6 )r r! r
r
r � test_testing& s zTestTesting.test_testingc C s\ t jjdd � tjt��}dd l}W d Q R X dt|d j�ksBt �dt|d j�ksXt �d S )Nzpandas.util.testingr z!pandas.util.testing is deprecatedzpandas.testing instead)
�sysr4 �popr rG rH �pandas.util.testingrL �messagerR )r �mr r
r
r � test_util_testing_deprecated+ s
z(TestTesting.test_util_testing_deprecatedc C s` t jjdd � tjt��}ddlm} W d Q R X dt|d j �ksFt
�dt|d j �ks\t
�d S )Nzpandas.util.testingr )r_ z!pandas.util.testing is deprecatedzpandas.testing instead) ra r4 rb r rG rH rc r_ rL rd rR )r re r_ r
r
r �#test_util_testing_deprecated_direct5 s
z/TestTesting.test_util_testing_deprecated_directc C sN t jtjddgt jd�j� }d|ks(t�tjt dd�� t
j j W d Q R X d S )Nz-cz6import pandas; pandas.util.testing.assert_series_equal)�stderrz!pandas.util.testing is deprecated�foo)�match)
�
subprocess� check_outputra �
executable�STDOUT�decoderR �pytestZraises�AttributeErrorrC r" ri )r �outr
r
r �test_util_in_top_level> s z"TestTesting.test_util_in_top_levelN)r r r r6 r` rf rg rs r
r
r
r r^ s
r^ )rk ra Ztypingr rp r rC r r Zpandas._testingr0 r r r rT rV rY rZ r^ r
r
r
r �<module> s
a
|