U
|
ý°d»3 ã @ sú d dl Z d dlZd dlZd dlZd dlZd dlmZ d dl m
|
Z
|
d dl m Z m Z mZ ejdd Zejdd Zejdd Zejd
|
d Zejd d Zejdd Zejdd Zejdd ZG dd dZG dd dee
|
jZG dd dee
|
jZ G dd dee
|
j Z G d d d ee
|
j Z G d d d ee
|
j!Z"G d d! d!ee
|
j#Z$G d"d# d#ee
|
j%Z&ej'j(d$d%Z)G d&d' d'e
|
j*Z+G d(d) d)ee
|
j,Z-G d*d+ d+ee
|
j.Z/G d,d- d-ee
|
j0Z1G d.d/ d/ee
|
j2Z3G d0d1 d1ee
|
j4Z5G d2d3 d3ee
|
j6Z7dS )4é N)Úbase)Ú JSONArrayÚ JSONDtypeÚ make_datac C s t S ©N)r © r r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/json/test_json.pyÚdtype s r c C s. t } t| d t| d kr&t } qt| S )z*Length-100 PeriodArray for semantics test.r é )r Úlenr )Údatar r r r s r c C s t i ddigS )z Length 2 array with [NA, Valid]Úaé
|
©r r r r r Ú data_missing' s r c C s t ddiddidddgS )NÚbr
|
Úcé é é ©r r r r r r r Údata_for_sorting- s r c C s t ddii ddigS )Nr r
|
r r r r r r r Údata_missing_for_sorting2 s r c C s | j S r )Úna_value)r r r r r 7 s r c C s t jS r )ÚoperatorÚeqr r r r Úna_cmp< s r c
|
C s4 t ddiddii i ddddddddiddigS )Nr r
|
r r r r r r r r r Údata_for_groupingA s øÿr c @ s$ e Zd Zedd Zedd ZdS )ÚBaseJSONc O st |j jdkr\|j |j ks ttjt|j t¡|j |jd}tjt|j t¡|j |jd}t
|
j ||f|| d S )NÚjson)ÚindexÚname) r r! ÚAssertionErrorÚpdÚSeriesr ÚvaluesZastypeÚobjectr ÚtmÚassert_series_equal)ÚclsÚleftÚrightÚargsÚkwargsr r r r( V s ÿýz BaseJSON.assert_series_equalc
|
O s° | dd¡}tj|j|j| dd¡| dd¡| dd¡| d d¡| d
|
d |jd kj}|D ] }| j|| || f|| q^|j|d }|j|d }tj||f|| d S )NÚobjZ DataFrameZcheck_column_typeÚequivÚ check_namesTÚ check_exactFÚcheck_categoricalz.columns)Úexactr0 r1 r2 r. r )Úcolumns) Úgetr' Zassert_index_equalr4 Zdtypesr r( ZdropÚassert_frame_equal)r) r* r+ r, r- Zobj_typeZjsonsÚcolr r r r6 d s
|
|
|
|
ù
|
zBaseJSON.assert_frame_equalN)Ú__name__Ú
|
__module__Ú __qualname__Ú classmethodr( r6 r r r r r Q s
|
r c @ s e Zd ZdS )Ú TestDtypeN©r8 r9 r: r r r r r< { s r< c s2 e Zd Zdd Zejjdd fddZ ZS )Ú TestInterfacec C sÈ t t ddi¡t ddi¡t ddi¡g}t |¡}| ||¡ | | ¡ | ¡ ¡ t | dddg¡¡}d}t j
|
t |d | ||¡ W 5 Q R X t j
|
t |d | | ¡ | ¡ ¡ W 5 Q R X d S )
|
Nr r
|
r r r r r zSeries are different©Úmatch) r Ú collectionsÚUserDictr# r$ r( r6 Zto_frameZtakeÚpytestÚraisesr" )Úselfr r r Úmsgr r r Útest_custom_asserts s ýÿ
|
z!TestInterface.test_custom_assertsz:comparison method not implemented for JSONArray (GH-37867)©Úreasonc s t |¡ d S r )ÚsuperÚ test_contains©rE r ©Ú __class__r r rK s zTestInterface.test_contains) r8 r9 r: rG rC ÚmarkÚxfailrK Ú __classcell__r r rM r r> s
|
ÿr> c sx e Zd Zejjdd fddZejjdd fddZejjdd fdd Zejjd
|
d fd d Z Z
|
S ) ÚTestConstructorsz¬ implemented constructor from dtyperH c s t |¡ d S r )rJ Útest_from_dtyperL rM r r rS s z TestConstructors.test_from_dtypezRecursionError, GH-33900c s6 t ¡ }z t d¡ t ||¡ W 5 t |¡ X d S ©Néd )ÚsysÚgetrecursionlimitÚsetrecursionlimitrJ Ú*test_series_constructor_no_data_with_index©rE r r Ú rec_limitrM r r rY ¤ s
|
|
z;TestConstructors.test_series_constructor_no_data_with_indexc s6 t ¡ }z t d¡ t ||¡ W 5 t |¡ X d S rT )rV rW rX rJ Ú,test_series_constructor_scalar_na_with_indexrZ rM r r r\ ¯ s
|
|
z=TestConstructors.test_series_constructor_scalar_na_with_indexz collection as scalar, GH-33901c s6 t ¡ }z t d¡ t ||¡ W 5 t |¡ X d S rT )rV rW rX rJ Ú)test_series_constructor_scalar_with_index)rE r r r[ rM r r r] º s
|
|
z:TestConstructors.test_series_constructor_scalar_with_index) r8 r9 r: rC rO rP rS rY r\ r] rQ r r rM r rR s
|
|
rR c sD e Zd Zejjdd fddZejjdd fddZ ZS )Ú TestReshapingzDifferent definitions of NArH c s t ¡ dS )zÛ
|
The test does .astype(object).stack(). If we happen to have
|
any missing values in `data`, then we'll end up with different
|
rows since we consider `{}` NA, but `.astype(object)` doesn't.
|
N)rJ Ú
|
test_stack©rE rM r r r_ Ç s zTestReshaping.test_stackz dict for NAc s t ||¡S r )rJ Ú test_unstack)rE r r rM r r ra Ð s zTestReshaping.test_unstack) r8 r9 r: rC rO rP r_ ra rQ r r rM r r^ Æ s r^ c @ s e Zd ZdS )Ú TestGetitemNr= r r r r rb × s rb c @ s e Zd ZdS )Ú TestIndexNr= r r r r rc Û s rc c sD e Zd Zejjdd fddZejjdd fddZ ZS )Ú TestMissingzSetting a dict as a scalarrH c s t ¡ dS ©z;We treat dictionaries as a mapping in fillna, not a scalar.N)rJ Útest_fillna_seriesr` rM r r rf à s z TestMissing.test_fillna_seriesc s t ¡ dS re )rJ Útest_fillna_framer` rM r r rg å s z TestMissing.test_fillna_frame) r8 r9 r: rC rO rP rf rg rQ r r rM r rd ß s rd Z
|
UnhashablerH c @ s e Zd ZdS )Ú
|
TestReduceNr= r r r r rh î s rh c sD e Zd Ze fddZe fddZe fddZej ddd g¡ fd
|
d Z
|
ej ddd g¡ fd d Z ejj dd fddZ ejj dd fddZejj dd ed fddZe fddZejj dd fdd Zejj d d fd d Zejj d d fd d!Zej d"¡ fd#d$Z ZS )%Ú TestMethodsc s t ||¡ d S r )rJ Útest_value_counts)rE Zall_dataZdropnarM r r rj ó s z TestMethods.test_value_countsc s t |¡ d S r )rJ Ú test_value_counts_with_normalizerL rM r r rk ÷ s z,TestMethods.test_value_counts_with_normalizec s t ¡ d S r )rJ Útest_sort_values_framer` rM r r rl û s z"TestMethods.test_sort_values_frameÚ ascendingTFc s t |||¡ d S r )rJ Útest_sort_values)rE r rm Ú sort_by_keyrM r r rn s z TestMethods.test_sort_valuesc s t |||¡ d S r )rJ Útest_sort_values_missing)rE r rm ro rM r r rp s
|
ÿz$TestMethods.test_sort_values_missingz#combine for JSONArray not supportedrH c s t |¡ d S r )rJ Útest_combine_le©rE Z data_repeatedrM r r rq s zTestMethods.test_combine_lec s t |¡ d S r )rJ Útest_combine_addrr rM r r rs s z TestMethods.test_combine_addzGcombine for JSONArray not supported - may pass depending on random data)rI ÚstrictrD c s t |¡ d S r )rJ Útest_combine_firstrL rM r r ru s z TestMethods.test_combine_firstc s t ||¡ d S r )rJ Ú test_hash_pandas_object_works)rE r ÚkindrM r r rv s z)TestMethods.test_hash_pandas_object_workszbroadcasting errorc s t ||¡ d S r )rJ Útest_where_series)rE r r rM r r rx ! s z TestMethods.test_where_serieszCan't compare dicts.c s t |¡ d S r )rJ Útest_searchsorted)rE r rM r r ry ( s z TestMethods.test_searchsortedc s t |||¡ d S r )rJ Ú test_equals)rE r r Z as_seriesrM r r rz , s zTestMethods.test_equalsz-fill-value is interpreted as a dict of valuesc s t |¡ d S r )rJ Útest_fillna_copy_frame)rE r rM r r r{ 0 s z"TestMethods.test_fillna_copy_frame)r8 r9 r: Ú
|
unhashablerj rk rl rC rO Z parametrizern rp rP rq rs r" ru rv rx ry rz Úskipr{ rQ r r rM r ri ò s< ü
|
ri c s* e Zd Zejjdd fddZ ZS )Ú TestCastingz$failing on np.array(self, dtype=str)rH c s t ¡ dS )zThis currently fails in NumPy on np.array(self, dtype=str) with
|
|
*** ValueError: setting an array element with a sequence
|
N)rJ Útest_astype_strr` rM r r r 6 s zTestCasting.test_astype_str)r8 r9 r: rC rO rP r rQ r r rM r r~ 5 s r~ c sj e Zd Ze fddZe fddZe fddZe fddZej j
|
d d
|
fd d Z Z S ) Ú TestGroupbyc s t ¡ dS )a
|
This currently fails in Series.name.setter, since the
|
name must be hashable, but the value is a dictionary.
|
I think this is what we want, i.e. `.name` should be the original
|
values, and not the values for factorization.
|
N)rJ Ú test_groupby_extension_transformr` rM r r r D s z,TestGroupby.test_groupby_extension_transformc s t ¡ dS )zó
|
This fails in Index._do_unique_check with
|
|
> hash(val)
|
E TypeError: unhashable type: 'UserDict' with
|
|
I suspect that once we support Index[ExtensionArray],
|
we'll be able to dispatch unique.
|
N)rJ Ú test_groupby_extension_applyr` rM r r r N s z(TestGroupby.test_groupby_extension_applyc s t ¡ dS ©z
|
This fails when we get to tm.assert_series_equal when left.index
|
contains dictionaries, which are not hashable.
|
N)rJ Útest_groupby_extension_aggr` rM r r r [ s z&TestGroupby.test_groupby_extension_aggc s t ¡ dS r )rJ Ú test_groupby_extension_no_sortr` rM r r r
c s z*TestGroupby.test_groupby_extension_no_sortz'GH#39098: Converts agg result to objectrH c s t |¡ d S r )rJ Útest_groupby_agg_extension)rE r rM r r r k s z&TestGroupby.test_groupby_agg_extension) r8 r9 r: r| r r r r
rC rO rP r rQ r r rM r r C s r c sN e Zd Z fddZdd Zejjdd fddZe f fd d
|
Z
|
Z S ) ÚTestArithmeticOpsc s< t |d dkr*tjjdd}|j |¡ t ||¡ d S )Nr r
|
z raises in coercing to SeriesrH )r rC rO rP ÚnodeZ
|
add_markerrJ Ú test_arith_frame_with_scalar)rE r Zall_arithmetic_operatorsÚrequestrO rM r r r q s z.TestArithmeticOps.test_arith_frame_with_scalarc C s0 t |¡}tjtdd || W 5 Q R X d S )NÚ unsupportedr? )r# r$ rC rD Ú TypeError)rE r Zserr r r Ú$test_add_series_with_extension_arrayw s
|
z6TestArithmeticOps.test_add_series_with_extension_arrayznot implementedrH c s t ¡ d S r )rJ Útest_divmod_series_arrayr` rM r r r | s z*TestArithmeticOps.test_divmod_series_arrayc s t j|||tdS )N)Úexc)rJ Ú_check_divmod_opr )rE ÚsÚopÚotherr rM r r r s z"TestArithmeticOps._check_divmod_op) r8 r9 r: r r rC rO rP r ÚNotImplementedErrorr rQ r r rM r r p s
|
r c @ s e Zd ZdS )ÚTestComparisonOpsNr= r r r r r s r c @ s e Zd ZdS )Ú TestPrintingNr= r r r r r s r )8rA r rV rC Zpandasr# Zpandas._testingZ_testingr' Zpandas.tests.extensionr Z!pandas.tests.extension.json.arrayr r r Zfixturer r r r r r r r r ZBaseDtypeTestsr< ZBaseInterfaceTestsr> ZBaseConstructorsTestsrR ZBaseReshapingTestsr^ ZBaseGetitemTestsrb ZBaseIndexTestsrc ZBaseMissingTestsrd rO rP r| ZBaseNoReduceTestsrh ZBaseMethodsTestsri ZBaseCastingTestsr~ ZBaseGroupbyTestsr ZBaseArithmeticOpsTestsr ZBaseComparisonOpsTestsr ZBasePrintingTestsr r r r r Ú<module> sN
|
|
|
|
|
|
|
|
* ( C-
|
