a
    Pf$                     @  s   d dl mZ d dlZd dlZd dlmZ d dlmZ d dl	m
ZmZmZmZmZ G dd dZG dd deZG d	d
 d
eZG dd deZdd ZdS )    )annotationsN)api)
extensionsindexersinterchangetypestypingc                   @  s   e Zd ZdddZdS )BaseNc                 C  sJ   t dd t|D }|d ur2t t|t| }t |}t|| d S )Nc                 s  s$   | ]}| d s|dkr|V  qdS )__r   N)
startswith).0f r   R/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/api/test_api.py	<genexpr>   s   zBase.check.<locals>.<genexpr>)sorteddirsettmZassert_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 Zd ZU g dZg dZg dZg dZg dZg Zde	d< g Z
de	d< g d	Zg d
Zg dZdgZg dZg Zde	d< g Zde	d< g dZejsed dd Zdd Zdd ZdS )	TestPDApi)ZtestslocaleZconftestZ_version_meson)	r   arraysoptionstesttestingerrorsZplottingioZtseries)compatcorepandasutil_built_with_meson)Z
IndexSliceZNaTZNA)(Z
ArrowDtypeZCategoricalZCategoricalIndex	DataFrameZ
DateOffsetZDatetimeIndexZ	ExcelFileZExcelWriterFlagsZGrouperZHDFStoreIndexZ
MultiIndexZPeriodZPeriodIndexZ
RangeIndexZSeriesZSparseDtypeZStringDtypeZ	TimedeltaZTimedeltaIndex	TimestampZIntervalZIntervalIndexCategoricalDtypePeriodDtypeIntervalDtypeDatetimeTZDtypeZBooleanDtypeZ	Int8DtypeZ
Int16DtypeZ
Int32DtypeZ
Int64DtypeZ
UInt8DtypeZUInt16DtypeZUInt32DtypeZUInt64DtypeZFloat32DtypeZFloat64DtypeZNamedAggz	list[str]deprecated_classesmodules)arrayZbdate_rangeconcatZcrosstabcutZ
date_rangeZinterval_rangeevalZ	factorizeZget_dummiesZfrom_dummiesZ
infer_freqZisnaZisnullZlreshapeZmeltZnotnaZnotnulloffsetsmergeZmerge_orderedZ
merge_asofZperiod_rangeZpivotZpivot_tableZqcutZshow_versionsZtimedelta_rangeuniqueZvalue_countsZwide_to_long)Zreset_optionZdescribe_option
get_optionZoption_contextZ
set_optionZset_eng_float_format)Zread_clipboardZread_csvZ
read_excelZread_fwfZread_gbqZread_hdfZ	read_htmlZread_xml	read_jsonZread_pickleZread_sasZread_sqlZread_sql_queryZread_sql_tableZ
read_stataZ
read_tableZread_featherZread_parquetZread_orcZ	read_spssZjson_normalize)to_datetimeZ
to_numericZ	to_pickleZto_timedeltadeprecated_funcs_in_futuredeprecated_funcs)_configZ_libsZ_is_numpy_devZ_pandas_datetime_CAPIZ_pandas_parser_CAPI_testingZ_typing_versionc                 C  sX   | j | j | j | j | j | j | j | j | j | j	 | j
 }| jt|| jd d S )N)r   r   r   )
public_libprivate_libmiscr4   classesfuncsfuncs_option
funcs_read
funcs_jsonfuncs_toprivate_modulesr   pdr   )r   Z
checktheser   r   r   test_api   s.    	
zTestPDApi.test_apic                 C  sr   t | j| j | j | j | j | j | j | j | j	 t | j
 }t tj}|| }|r^J || }|rnJ d S N)r   rD   rF   r4   rG   rH   rI   rJ   rK   rL   r3   rN   __all__)r   r   actualZ
extraneousmissingr   r   r   test_api_all   s4    

zTestPDApi.test_api_allc              	   C  sT   | j | j | j }|D ]8}tt tt|}W d    q1 sD0    Y  qd S rP   )r3   r@   r?   r   assert_produces_warningFutureWarninggetattrrN   )r   Zdeprecated_listZdepr_r   r   r   	test_depr   s    zTestPDApi.test_deprN)r   r   r   r   rD   rE   rF   rG   r3   __annotations__r4   rH   rI   rJ   rK   rL   r?   r@   rM   rN   r*   appendrO   rT   rY   r   r   r   r   r   !   s(   
,#
	
r   c                   @  sl   e Zd Zg dZg dZg dZddgZg dZg dZdd	 Z	d
d Z
dd Zdd Zdd Zdd ZdS )TestApi)r   r   r   r   r   )ZDataFrameGroupByZDatetimeIndexResamplerGroupbyZ	ExpandingZExpandingGroupbyZExponentialMovingWindowZExponentialMovingWindowGroupbyZ
JsonReaderZNaTTypeZNATypeZPeriodIndexResamplerGroupbyZ	ResamplerZRollingZRollingGroupbyZSeriesGroupByZStataReaderZTimedeltaIndexResamplerGroupbyZTimeGrouperZWindow).Zis_any_real_numeric_dtypeZis_array_likeZis_boolZis_bool_dtypeZis_categorical_dtypeZ
is_complexZis_complex_dtypeZis_datetime64_any_dtypeZis_datetime64_dtypeZis_datetime64_ns_dtypeZis_datetime64tz_dtypeZis_dict_likeZis_dtype_equalZis_extension_array_dtypeZis_file_likeZis_floatZis_float_dtypeZis_hashableZis_int64_dtype
is_integerZis_integer_dtypeZis_intervalZis_interval_dtypeZis_iteratorZis_list_likeZis_named_tupleZ	is_numberZis_numeric_dtypeZis_object_dtypeZis_period_dtypeZis_reZis_re_compilableZ	is_scalarZis_signed_integer_dtypeZ	is_sparseZis_string_dtypeZis_timedelta64_dtypeZis_timedelta64_ns_dtypeZis_unsigned_integer_dtypeZpandas_dtypeZinfer_dtypeZunion_categoricalsr/   r2   r1   r0   Zfrom_dataframer+   )Zcheck_array_indexerZBaseIndexerZFixedForwardWindowIndexerZVariableOffsetWindowIndexer)	Z
no_defaultZExtensionDtypeZregister_extension_dtypeZregister_dataframe_accessorZregister_index_accessorZregister_series_accessorZtakeZExtensionArrayZExtensionScalarOpsMixinc                 C  s   |  t| j d S rP   )r   r   allowed_api_dirsr   r   r   r   rO   U  s    zTestApi.test_apic                 C  s   |  t| j d S rP   )r   
api_typingallowed_typingr_   r   r   r   test_api_typingX  s    zTestApi.test_api_typingc                 C  s   |  t| j d S rP   )r   	api_typesallowed_api_typesr_   r   r   r   test_api_types[  s    zTestApi.test_api_typesc                 C  s   |  t| j d S rP   )r   api_interchangeallowed_api_interchanger_   r   r   r   test_api_interchange^  s    zTestApi.test_api_interchangec                 C  s   |  t| j d S rP   )r   api_indexersallowed_api_indexersr_   r   r   r   test_api_indexersa  s    zTestApi.test_api_indexersc                 C  s   |  t| j d S rP   )r   api_extensionsallowed_api_extensionsr_   r   r   r   test_api_extensionsd  s    zTestApi.test_api_extensionsN)r   r   r   r^   ra   rd   rg   rj   rm   rO   rb   re   rh   rk   rn   r   r   r   r   r\      s   0r\   c                   @  s$   e Zd Zg dZdd Zdd ZdS )TestTesting)Zassert_frame_equalZassert_series_equalZassert_index_equalZassert_extension_array_equalc                 C  s   ddl m} | || j d S )Nr   )r#   )r(   r#   r   rH   )r   r#   r   r   r   test_testingp  s    zTestTesting.test_testingc                 C  s:   t jtdd tjj W d    n1 s,0    Y  d S )Nfoomatch)pytestZraisesAttributeErrorrN   r)   rq   r_   r   r   r   test_util_in_top_levelu  s    z"TestTesting.test_util_in_top_levelN)r   r   r   rH   rp   rv   r   r   r   r   ro   h  s   ro   c                  C  sN   d} t jt| d tjj}W d    n1 s00    Y  |tjju sJJ d S )Nz0PandasArray has been renamed NumpyExtensionArrayrr   )r   rU   rV   rN   r    ZPandasArrayZNumpyExtensionArray)msgresr   r   r   test_pandas_array_aliasz  s    &ry   )
__future__r   rt   r(   rN   r   Zpandas._testingrB   r   Z
pandas.apir   rl   r   ri   r   rf   r   rc   r   r`   r	   r   r\   ro   ry   r   r   r   r   <module>   s   	 Vr