ó šxŠ\c@sÃdZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl mZde fd„ƒYZdefd „ƒYZd „ZdS( s­Test extension array for storing nested data in a pandas container. The JSONArray stores lists of dictionaries. The storage mechanism is a list, not an ndarray. Note: We currently store lists of UserDicts (Py3 only). Pandas has a few places internally that specifically check for dicts, and does non-scalar things in that case. We *want* the dictionaries to be treated as scalars, so we hack around pandas by using UserDicts. iÿÿÿÿN(tExtensionDtype(tcompat(tExtensionArrayt JSONDtypecBs_eZejZdZyejƒZWne k r>iZnXe d„ƒZ e d„ƒZ RS(tjsoncCstS(sgReturn the array type associated with this dtype Returns ------- type (t JSONArray(tcls((s@lib/python2.7/site-packages/pandas/tests/extension/json/array.pytconstruct_array_type&scCs2||jkr|ƒStdj||ƒƒ‚dS(Ns!Cannot construct a '{}' from '{}'(tnamet TypeErrortformat(Rtstring((s@lib/python2.7/site-packages/pandas/tests/extension/json/array.pytconstruct_from_string0s ( t__name__t __module__RtMappingttypeRt collectionstUserDicttna_valuetAttributeErrort classmethodRR (((s@lib/python2.7/site-packages/pandas/tests/extension/json/array.pyRs    RcBsÎeZeƒZdZded„Zeded„ƒZ ed„ƒZ d„Z d„Z d„Z ed„ƒZd„Zedd „Zed „Zed „Zd „Zed „ƒZd„Zd„ZRS(iècCsexB|D]:}t||jjƒstdt|jjƒƒ‚qqW||_|j|_|_dS(NsAll values must be of type (t isinstancetdtypeRR tstrtdatat_itemst_data(tselftvaluesRtcopytval((s@lib/python2.7/site-packages/pandas/tests/extension/json/array.pyt__init__=s   cCs ||ƒS(N((RtscalarsRR((s@lib/python2.7/site-packages/pandas/tests/extension/json/array.pyt_from_sequenceKscCs2|g|D]!}|dkr tj|ƒ^q ƒS(N((RR(RRtoriginaltx((s@lib/python2.7/site-packages/pandas/tests/extension/json/array.pyt_from_factorizedOscCsÊt|tjƒr|j|St|tjƒrs|jdkrs|jgt||ƒD]\}}|rT|^qTƒSt|t j ƒr¯t |ƒg|D]}|j|^q•ƒSt |ƒ|j|ƒSdS(Ntbool( RtnumberstIntegralRtnptndarrayRR"tzipRtIterableR(RtitemR$tmti((s@lib/python2.7/site-packages/pandas/tests/extension/json/array.pyt __getitem__Ss !5*cCs"t|tjƒr"||j|¬s (RtlistRRO(RR$((s@lib/python2.7/site-packages/pandas/tests/extension/json/array.pytunique¨s cCs8ttjjg|D]}|j^qƒƒ}||ƒS(N(RUR2tchaint from_iterableR(Rt to_concatR$R((s@lib/python2.7/site-packages/pandas/tests/extension/json/array.pyt_concat_same_type¯s.cCs7|jƒ}t|ƒdkr-|jƒ}n|dfS(Ni((t_values_for_argsortR;travel(Rtfrozen((s@lib/python2.7/site-packages/pandas/tests/extension/json/array.pyt_values_for_factorize´s cCsCdgg|D]}t|jƒƒ^q }tj|dtƒdS(NRi((RQRRR)R@tobject(RR$R]((s@lib/python2.7/site-packages/pandas/tests/extension/json/array.pyR[»s,N(R RRRt__array_priority__RCtFalseR RR"R%R0R:R<tpropertyR?RARMRtTrueRPRVRZR^R[(((s@lib/python2.7/site-packages/pandas/tests/extension/json/array.pyR9s$       c CsigtdƒD]X}tjgttjddƒƒD]*}tjtjƒtjddƒf^q2ƒ^q S(Nidii (trangeRRtrandomtrandinttchoiceR t ascii_letters(t_((s@lib/python2.7/site-packages/pandas/tests/extension/json/array.pyt make_dataÃs(t__doc__RR2R'ReR R=tnumpyR)tpandas.core.dtypes.baseRtpandasRtpandas.core.arraysRRRRj(((s@lib/python2.7/site-packages/pandas/tests/extension/json/array.pyt s       Š