a PfF@sddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl mZmZmZejjddZejddZejdd Zejd d Zejd d ZejddZejddZejddZGddde jZddZddZddZ dS)N)base) JSONArray JSONDtype make_dataZ UnhashablereasoncCstSN)rr r ^/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/extension/json/test_json.pydtypesr cCs.t}t|dt|dkr&t}qt|S)z*Length-100 PeriodArray for semantics test.r)rlenr)datar r r rsrcCstiddigS)zLength 2 array with [NA, Valid]a rr r r r data_missing,srcCstddiddidddgS)Nbr crrrr r r r data_for_sorting2srcCstddiiddigS)Nrr rrrr r r r data_missing_for_sorting7srcCstjSr)operatoreqr r r r na_cmp<src Cs4tddiddiiiddddddddiddigS)Nrr rrrrrr r r r data_for_groupingAsrcseZdZejjddfddZejjddfddZejjddfd d Zejjddfd d Z ejjd dfddZ ejjddfddZ ejjddfddZ ejjddfddZ ejjddfddZejddgdgdfdgdgdfdgdgd fdgd!gd"fd#gdgd$fd#gdgd$fd#gdgdfd#gd!gdfgfd%d&Zefd'd(Zefd)d*Zefd+d,Zejd-d.d/gfd0d1Zejd-d.d/gfd2d3Zejjd4dfd5d6Zejjd7d/ed8fd9d:Zejjd;dfddfd?d@Zejjd>dfdAdBZejdCfdDdEZfdFdGZejjdHdfdIdJZ efdKdLZ!efdMdNZ"efdOdPZ#efdQdRZ$fdSdTZ%fdUdVZ&ejjdWdfdXdYZ'ejjdWdfdZd[Z(ejjdWdfd\d]Z)ejjdWdfd^d_Z*ejjd`e+,gdae-j,gdadbdce-j,d.d.d.e-j.e-j.gdbdcggdddefdfdgZ/fdhdiZ0ejjdjdfdkdlZ1ejjdmgdne-j,gdndodce+,gdnggdpdefdqdrZ2ejjdsdejjdtdudvdwe-j.gd/fej3dudvdwe-j.gd.ejjdxddye-j,dudvdwe-j.gdodcd/fe-j,dudvdwe-j.gdodcd/fggdzdefd{d|Z4ejjd}dfd~dZ5fddZ6ejjddfddZ7ejjdjdejdddgfddZ8ejjddfddZ9ejjddfddZ:ejjddfddZ;ejjddfddZejjddejdddgfddZ?Z@S) TestJSONArrayz:comparison method not implemented for JSONArray (GH-37867)rcst|dSr)super test_containsselfr __class__r r r!RszTestJSONArray.test_containsz¬ implemented constructor from dtypecst|dSr)r test_from_dtyper"r$r r r&YszTestJSONArray.test_from_dtypezRecursionError, GH-33900c s@t}z&tdt||Wt|n t|0dSNd)sysgetrecursionlimitsetrecursionlimitr *test_series_constructor_no_data_with_indexr#r na_value rec_limitr$r r r,^s  z8TestJSONArray.test_series_constructor_no_data_with_indexc s@t}z&tdt||Wt|n t|0dSr')r)r*r+r ,test_series_constructor_scalar_na_with_indexr-r$r r r0is  z:TestJSONArray.test_series_constructor_scalar_na_with_indexzcollection as scalar, GH-33901c s@t}z&tdt||Wt|n t|0dSr')r)r*r+r )test_series_constructor_scalar_with_index)r#rr r/r$r r r1ts  z7TestJSONArray.test_series_constructor_scalar_with_indexzDifferent definitions of NAcstdS)z The test does .astype(object).stack(future_stack=True). 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)r test_stackr#r$r r r2szTestJSONArray.test_stackz dict for NAcst||Sr)r test_unstack)r#rindexr$r r r4szTestJSONArray.test_unstackzSetting a dict as a scalarcstdSz;We treat dictionaries as a mapping in fillna, not a scalar.N)r test_fillna_seriesr3r$r r r7sz TestJSONArray.test_fillna_seriescstdSr6)r test_fillna_framer3r$r r r8szTestJSONArray.test_fillna_framez'limit_area, input_ilocs, expected_ilocsZoutside)r rrrr )r rr rr )rr r r r)rr r r r )rr rr r)rr rr r Zinside)r r r r r csHd}tjt|d"t||||Wdn1s:0YdS)Nz'JSONArray does not implement limit_areamatch)pytestraisesNotImplementedErrorr test_ffill_limit_area)r#rZ limit_areaZ input_ilocsZexpected_ilocsmsgr$r r r>s z#TestJSONArray.test_ffill_limit_areacst||dSr)r test_value_counts)r#Zall_dataZdropnar$r r r@szTestJSONArray.test_value_countscst|dSr)r test_value_counts_with_normalizer"r$r r rAsz.TestJSONArray.test_value_counts_with_normalizecstdSr)r test_sort_values_framer3r$r r rBsz$TestJSONArray.test_sort_values_frame ascendingTFcst|||dSr)r test_sort_values)r#rrC sort_by_keyr$r r rDszTestJSONArray.test_sort_valuescst|||dSr)r test_sort_values_missing)r#rrCrEr$r r rFsz&TestJSONArray.test_sort_values_missingz#combine for JSONArray not supportedcst|dSr)r test_combine_le)r#Z data_repeatedr$r r rGszTestJSONArray.test_combine_lezGcombine for JSONArray not supported - may pass depending on random data)rstrictr<cst|dSr)r test_combine_firstr"r$r r rIsz TestJSONArray.test_combine_firstzbroadcasting errorcst||dSr)r test_where_series)r#rr.r$r r rJszTestJSONArray.test_where_serieszCan't compare dicts.cst|dSr)r test_searchsorted)r#rr$r r rKszTestJSONArray.test_searchsortedcst|||dSr)r test_equals)r#rr.Z as_seriesr$r r rLszTestJSONArray.test_equalsz-fill-value is interpreted as a dict of valuescst|dSr)r test_fillna_copy_frame)r#rr$r r rMsz$TestJSONArray.test_fillna_copy_framecs,|rtjjdd}||t|dS)NzFails with CoWr)r;markxfail applymarkerr &test_equals_same_data_different_object)r#rZusing_copy_on_writerequestrNr$r r rQs z4TestJSONArray.test_equals_same_data_different_objectz$failing on np.array(self, dtype=str)cstdS)zThis currently fails in NumPy on np.array(self, dtype=str) with *** ValueError: setting an array element with a sequence N)r test_astype_strr3r$r r rSszTestJSONArray.test_astype_strcstdS)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)r test_groupby_extension_transformr3r$r r rTsz.TestJSONArray.test_groupby_extension_transformcstdS)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)r test_groupby_extension_applyr3r$r r rUs z*TestJSONArray.test_groupby_extension_applycstdSz This fails when we get to tm.assert_series_equal when left.index contains dictionaries, which are not hashable. N)r test_groupby_extension_aggr3r$r r rWsz(TestJSONArray.test_groupby_extension_aggcstdSrV)r test_groupby_extension_no_sortr3r$r r rXsz,TestJSONArray.test_groupby_extension_no_sortcs:t|ddkr(tjjdd}||t||dS)Nrr zraises in coercing to Seriesr)r r;rNrOrPr test_arith_frame_with_scalar)r#rZall_arithmetic_operatorsrRrNr$r r rYs z*TestJSONArray.test_arith_frame_with_scalarcs4|jdvr"tjjdd}||t||dS)N)rnez"Comparison methods not implementedr)__name__r;rNrOrPr test_compare_array)r#rZ comparison_oprRrNr$r r r\%s  z TestJSONArray.test_compare_arrayz.ValueError: Must have equal len keys and valuecst|dSr)r test_setitem_loc_scalar_mixedr"r$r r r]+sz+TestJSONArray.test_setitem_loc_scalar_mixedcst|dSr)r ,test_setitem_loc_scalar_multiple_homogoneousr"r$r r r^/sz:TestJSONArray.test_setitem_loc_scalar_multiple_homogoneouscst|dSr)r test_setitem_iloc_scalar_mixedr"r$r r r_3sz,TestJSONArray.test_setitem_iloc_scalar_mixedcst|dSr)r -test_setitem_iloc_scalar_multiple_homogoneousr"r$r r r`7sz;TestJSONArray.test_setitem_iloc_scalar_multiple_homogoneousmask)TTTFFboolean)r ) numpy-arrayz boolean-arrayzboolean-array-na)ZidscsV|rtjjdd}||n$t|tjsBtjjdd}||t|||dS)N unhashabler@rArBrDrFrGAssertionErrorrIrJrKrLskiprMrQrSrTrUrWrXrYr\r]r^r_r`rfarraypdZNArgrjrkroparamrprqrurvryrzr{r|r}r~rr __classcell__r r r$r rQs                            $           rcOsz|jjdkr\|j|jksJtjt|jt|j|jd}tjt|jt|j|jd}t j ||g|Ri|dS)Njson)r5rt) r rtrSeriesrvaluesZastypeobjectr5tmZassert_series_equal)leftrightargskwargsr r r custom_assert_series_equals rc Os|dd}tj|j|j|dd|dd|dd|d d|d d |jd kj}|D]$}t||||g|Ri|q^|j|d }|j|d }tj||g|Ri|dS)NobjZ DataFrameZcheck_column_typeequiv check_namesT check_exactFcheck_categoricalz.columns)exactrrrrr)columns) getrZassert_index_equalrZdtypesr5rZdropZassert_frame_equal)rrrrobj_typeZjsonscolr r r custom_assert_frame_equals       "  rcCsttdditdditddig}t|}t||t||t|gd}d}t j t |d t||Wdn1s0Yt j t |d "t||Wdn1s0YdS) Nrr rrrr)rrr zSeries are differentr9) r collectionsUserDictrrrrZto_frameZtaker;r<r)rrrr?r r r test_custom_assertss     (r)!rrr)numpyrfr;ZpandasrZpandas._testingZ_testingrZpandas.tests.extensionrZ!pandas.tests.extension.json.arrayrrrrNrOrZfixturer rrrrrrZExtensionTestsrrrrr r r r s:         _