d02ddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z mZejdZejdZejdZejdZejdZejd Zejd Zejd ZGd d ZGddee jZGddee jZGddee jZGddee jZ Gddee j!Z"Gddee j#Z$Gddee j%Z&ej'(dZ)Gdde j*Z+Gd d!ee j,Z-Gd"d#ee j.Z/Gd$d%ee j0Z1Gd&d'ee j2Z3Gd(d)ee j4Z5Gd*d+ee j6Z7dS),N)base) JSONArray JSONDtype make_datactSN)ra/var/www/html/t/fyr/venv311/lib/python3.11/site-packages/pandas/tests/extension/json/test_json.pydtyper s ;;r ct}t|dt|dkr:t}t|dt|dk:t|S)z*Length-100 PeriodArray for semantics test.r)rlenr)datas r rrsf ;;D d1g,,#d1g,, & &{{ d1g,,#d1g,, & & T??r c(tiddigS)zLength 2 array with [NA, Valid]a rr r r data_missingr's b3)_ % %%r c4tddiddidddgS)Nbrcrrrr r r data_for_sortingr-s' sAha*:*:; < <r ctjSr)operatoreqr r r na_cmpr%<s ;r c LtddiddiiiddddddddiddigS)Nrrrrrrrr r r data_for_groupingr'AsQ  !H !H  !  !   !H !H    r c:eZdZedZedZdS)BaseJSONc|jjdkr|j|jksJtjt |jt|j|j}tjt |jt|j|j}tj ||g|Ri|dS)Njson)indexname) r r-pdSeriesrvaluesastypeobjectr,tmassert_series_equal)clsleftrightargskwargss r r4zBaseJSON.assert_series_equalVs :?f $ $:,,,,9$+,,V4455TZdiDI%,--f5566kZE tU 5 5 5 + +  $ $Q * * * + + + + + + + + + + + + + + +]> 5 5 5 @ @  # #AJJLL!**,, ? ? ? @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @s$9DD #D ;F  FFz:comparison method not implemented for JSONArray (GH-37867)reasoncJt|dSr)super test_containsr`r __class__s r rgzTestInterface.test_containss# d#####r ) rLrMrNrbr]markxfailrg __classcell__ris@r rUrUss@@@, [K$$$$$$$$$r rUcDeZdZejdfdZejdfdZejdfdZejdfdZ xZ S) TestConstructorsz¬ implemented constructor from dtyperccJt|dSr)rftest_from_dtyperhs r rqz TestConstructors.test_from_dtypes# %%%%%r zRecursionError, GH-33900ctj} tjdt||tj|dS#tj|wxYwNd)sysgetrecursionlimitsetrecursionlimitrf*test_series_constructor_no_data_with_indexr`r r! rec_limitris r rxz;TestConstructors.test_series_constructor_no_data_with_indexsq)++  -  !# & & & GG > >uh O O O  !) , , , , ,C !) , , , , 6A""A8ctj} tjdt||tj|dS#tj|wxYwrs)rurvrwrf,test_series_constructor_scalar_na_with_indexrys r r}z=TestConstructors.test_series_constructor_scalar_na_with_indexsq)++  -  !# & & & GG @ @ Q Q Q  !) , , , , ,C !) , , , ,r{zcollection as scalar, GH-33901ctj} tjdt||tj|dS#tj|wxYwrs)rurvrwrf)test_series_constructor_scalar_with_index)r`rr rzris r rz:TestConstructors.test_series_constructor_scalar_with_indexsq)++  -  !# & & & GG = =dE J J J  !) , , , , ,C !) , , , ,r{) rLrMrNr]rjrkrqrxr}rrlrms@r roros [FGG&&&&HG& [899----:9- [899----:9- [>??----@?-----r roceZdZejdfdZejdfdZxZS) TestReshapingzDifferent definitions of NArccHtdS)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)rf test_stackr`ris r rzTestReshaping.test_stacks! r z dict for NAcHt||Sr)rf test_unstack)r`rr,ris r rzTestReshaping.test_unstacksww##D%000r ) rLrMrNr]rjrkrrrlrms@r rrs [;<<=< [m,,1111-,11111r rceZdZdS) TestGetitemNrRr r r rrrSr rceZdZdS) TestIndexNrRr r r rrrSr rceZdZejdfdZejdfdZxZS) TestMissingzSetting a dict as a scalarrccHtdSz;We treat dictionaries as a mapping in fillna, not a scalar.N)rftest_fillna_seriesrs r rzTestMissing.test_fillna_seriess! ""$$$$$r cHtdSr)rftest_fillna_framers r rzTestMissing.test_fillna_frames! !!#####r ) rLrMrNr]rjrkrrrlrms@r rrs [:;;%%%%<;% [:;;$$$$<;$$$$$r r UnhashablercceZdZdS) TestReduceNrRr r r rrrSr rc2eZdZefdZefdZefdZej dddgfdZ ej dddgfdZ ej d fd Z ej d fd Zej d defdZefdZej d fdZej d fdZej d fdZejdfdZxZS) TestMethodscLt||dSr)rftest_value_counts)r`all_datadropnaris r rzTestMethods.test_value_countss# !!(F33333r cJt|dSr)rf test_value_counts_with_normalizerhs r rz,TestMethods.test_value_counts_with_normalizes! 0066666r cHtdSr)rftest_sort_values_framers r rz"TestMethods.test_sort_values_frames! &&(((((r ascendingTFcNt|||dSr)rftest_sort_values)r`rr sort_by_keyris r rzTestMethods.test_sort_valuess&   !19kJJJJJr cNt|||dSr)rftest_sort_values_missing)r`rrrris r rz$TestMethods.test_sort_values_missings2 (( $i     r z#combine for JSONArray not supportedrccJt|dSr)rftest_combine_ler` data_repeatedris r rzTestMethods.test_combine_le s!  .....r cJt|dSr)rftest_combine_addrs r rzTestMethods.test_combine_adds!   /////r zGcombine for JSONArray not supported - may pass depending on random data)rdstrictr^cJt|dSr)rftest_combine_firstrhs r rzTestMethods.test_combine_firsts# ""4(((((r cLt||dSr)rftest_hash_pandas_object_works)r`rkindris r rz)TestMethods.test_hash_pandas_object_workss# --dD99999r zbroadcasting errorcLt||dSr)rftest_where_series)r`rr!ris r rzTestMethods.test_where_series!s% !!$11111r zCan't compare dicts.cJt|dSr)rftest_searchsorted)r`rris r rzTestMethods.test_searchsorted(s" !!"233333r cNt|||dSr)rf test_equals)r`rr! as_seriesris r rzTestMethods.test_equals,s% D(I66666r z-fill-value is interpreted as a dict of valuescJt|dSr)rftest_fillna_copy_frame)r`rris r rz"TestMethods.test_fillna_copy_frame0s! &&|44444r )rLrMrN unhashablerrrr]rj parametrizerrrkrrr_rrrrrskiprrlrms@r rrs4444Z47777Z7))))Z) [[4-88KKKK98K [[4-88    98  [CDD////ED/ [CDD0000ED0 [,  ))))  )::::Z: [2332222432  [4554444654 [4557777657 [EFF5555GF55555r rc`eZdZejdfdZxZS) TestCastingz$failing on np.array(self, dtype=str)rccHtdS)zThis currently fails in NumPy on np.array(self, dtype=str) with *** ValueError: setting an array element with a sequence N)rftest_astype_strrs r rzTestCasting.test_astype_str6s! !!!!!r )rLrMrNr]rjrkrrlrms@r rr5sX [DEE""""FE"""""r rceZdZefdZefdZefdZefdZej dfdZ xZ S) TestGroupbycHtdS)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)rf test_groupby_extension_transformrs r rz,TestGroupby.test_groupby_extension_transformDs! 0022222r cHtdS)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)rftest_groupby_extension_applyrs r rz(TestGroupby.test_groupby_extension_applyNs! ,,.....r cHtdSz This fails when we get to tm.assert_series_equal when left.index contains dictionaries, which are not hashable. N)rftest_groupby_extension_aggrs r rz&TestGroupby.test_groupby_extension_agg[s! **,,,,,r cHtdSr)rftest_groupby_extension_no_sortrs r rz*TestGroupby.test_groupby_extension_no_sortcs! ..00000r z'GH#39098: Converts agg result to objectrccJt|dSr)rftest_groupby_agg_extension)r`r'ris r rz&TestGroupby.test_groupby_agg_extensionks" **+<=====r ) rLrMrNrrrrrr]rjrkrrlrms@r rrCs3333Z3 / / / /Z /----Z-1111Z1 [GHH>>>>IH>>>>>r rc~eZdZfdZdZejdfdZe ffd Z xZ S)TestArithmeticOpsct|ddkr:tjd}|j|t ||dS)Nrrzraises in coercing to Seriesrc)rr]rjrknode add_markerrftest_arith_frame_with_scalar)r`rall_arithmetic_operatorsrequestrjris r rz.TestArithmeticOps.test_arith_frame_with_scalarqsh tAw<<1  ;$$,J$KKD L # #D ) ) ) ,,T3KLLLLLr ctj|}tjtd5||zddddS#1swxYwYdS)N unsupportedrW)r.r/r]r^ TypeError)r`rsers r $test_add_series_with_extension_arrayz6TestArithmeticOps.test_add_series_with_extension_arraywsioo ]9M : : :   $JJ                  sAA Aznot implementedrccHtdSr)rftest_divmod_series_arrayrs r rz*TestArithmeticOps.test_divmod_series_array|s! ((*****r cXt|||tS)N)exc)rf_check_divmod_opr)r`sopotherrris r rz"TestArithmeticOps._check_divmod_ops$ww''2u)'DDDr ) rLrMrNrrr]rjrkrNotImplementedErrorrrlrms@r rrpsMMMMM   [/00++++10+ 2EEEEEEEEEEEr rceZdZdS)TestComparisonOpsNrRr r r rrrSr rceZdZdS) TestPrintingNrRr r r rrrSr r)8rYr#rur]pandasr.pandas._testing_testingr3pandas.tests.extensionr!pandas.tests.extension.json.arrayrrrfixturer rrrrr!r%r'r)BaseDtypeTestsrQBaseInterfaceTestsrUBaseConstructorsTestsroBaseReshapingTestsrBaseGetitemTestsrBaseIndexTestsrBaseMissingTestsrrjrkrBaseNoReduceTestsrBaseMethodsTestsrBaseCastingTestsrBaseGroupbyTestsrBaseArithmeticOpsTestsrBaseComparisonOpsTestsrBasePrintingTestsrr r r rsf ''''''    &&& ===///   '<'<'<'<'<'<'<'%-%-%-%-%-x!;%-%-%-P11111Hd5111"     (D1        $-    $ $ $ $ $(D1 $ $ $[  l  3 3      '   @5@5@5@5@5(D1@5@5@5F"""""(D1"""*>*>*>*>*>(D1*>*>*>ZEEEEE$"=EEE,     $"=        8T3     r