U
|
ý°d ã @ s
|
d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z
|
d dl Z d dl Z d dl m mZ d dlmZ d dlmZmZmZmZmZmZmZmZmZ d dl m Z dd Z e j ddddd gd
|
d d Z!G d d dZ"G dd dZ#G dd dZ$dS )é N)ÚIS64) Ú DataFrameÚ DatetimeIndexÚIndexÚNaTÚ PeriodIndexÚSeriesÚ TimedeltaÚ TimestampÚ
|
date_rangec C s dd | ¡ D S )zÉ
|
Sanitize dictionary for JSON by converting all keys to strings.
|
|
Parameters
|
----------
|
d : dict
|
The dictionary to convert.
|
|
Returns
|
-------
|
cleaned_dict : dict
|
c S s i | ]\}}t ||qS © )Ústr)Ú.0ÚkÚvr r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/json/test_ujson.pyÚ
|
<dictcomp>- s z _clean_dict.<locals>.<dictcomp>)Úitems©Údr r r Ú _clean_dict s r ÚsplitÚrecordsÚvaluesÚindex)Úparamsc C s | j S ©N)Úparam)Úrequestr r r Úorient0 s r c @ s e Zd Zejje dddd Zej dddg¡dd Z ej d
|
d d d g¡dd Z
|
dd Z dd Z dd Z ej dddg¡dd Zej dejej g¡d d Zd d Zd d! Zd"d# Zd$d% Zej d&d'd(d)d*g¡d+d, Zd-d. Zej d/d0d1g¡d2d3 Zd4d5 Zd6d7 Zd8d9 Zd:d; Z d<d= Z d>d? Z ej d@dAdBdCg¡dDdE Z dFdG Z dHdI Z!ej dJd*ddg¡dKdL Z"dMdN Z#dOdP Z$ej dQe% &¡ e% &dRdSdT¡e% &dUdVdWdX¡g¡dYdZ Z'd[d\ Z(d]d^ Z)ej d_e*e+ ,d`¡e+j-e+j.e+j. g¡dadb Z/dcdd Z0dedf Z1dgdh Z2didj Z3dkdl Z4ej dmdndodpdqg¡drds Z5ej dtdndog¡dudv Z6ej dwdxdydzd{d|g¡d}d~ Z7ej ddddg¡dd Z8ej d
dddg¡dd Z9ej dddg¡dd Z:dd Z;dd Z<dd Z=ej dddg¡dd Z>ej dddg¡dd Z?ej d d¡d¢d£d¤d¥g¡d¦d§ Z@d¨d© ZAej dªd«d¬dd®g¡d¯d° ZBd±d² ZCd³d´ ZDdµd¶ ZEd·d¸ ZFd¹dº ZGd*S )»ÚTestUltraJSONTestsz$not compliant on 32-bit, xref #15865)Úreasonc C sØ t d¡}tj|dd}t |¡}|dks.tt d¡}tj|dd}|dksRtt |¡}|dkshtt d ¡}tj|dd}|d
|
kstt |¡}|d ks¢tt d ¡}tj|dd}|d ksÆtt |¡}|dksÜtt d¡}tj|dd}|dkstt |¡}|dkstt d¡}tj|dd}|dks@tt |¡}|dksXtt d¡}tj|dd}|dks~tt |¡}|dkstt d¡}tj|dd}|dks¼tt |¡}|dksÔtd S )Nz 1337.1337é ©Údouble_precisiongç §èä@z0.95é ú1.0ç ð?z0.94z0.9gÍÌÌÌÌÌì?z1.95z2.0g @z-1.95z-2.0g Àz0.995é z0.9995é z0.99999999999999944)ÚdecimalÚDecimalÚujsonÚencodeÚdecodeÚAssertionError©ÚselfÚsutÚencodedÚdecodedr r r Útest_encode_decimal8 sN
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
z&TestUltraJSONTests.test_encode_decimalÚ ensure_asciiTFc s> dd}d} fdd}|| ||dd ||dd d S ) Nz"A string \ /
|
</script> &z,"A string \\ \/ \b \f \n \r \t <\/script> &"z;"A string \\ \/ \b \f \n \r \t \u003c\/script\u003e \u0026"c sJ t jfd i|}|| ks"tt |¡ks4tt |¡ksFtd S )Nr6 )r, r- r/ ÚjsonÚloadsr. )Zexpected_outputÚ encode_kwargsÚoutput©r6 Ú string_inputr r Úhelperx s ÿÿÿ z@TestUltraJSONTests.test_encode_string_conversion.<locals>.helperF)Zencode_html_charsTr )r1 r6 Znot_html_encodedZ html_encodedr= r r; r Ú test_encode_string_conversionp s ÿ
|
z0TestUltraJSONTests.test_encode_string_conversionÚ long_numberlüÿÿÿZ:`¡7{ g#å_çt¦ÂgÔpAª¿Ác C s0 d|i}t j|dd}t |¡}||ks,td S )NÚar" r# ©r, r- r. r/ )r1 r? r2 r3 r4 r r r Útest_double_long_numbers s
|
z+TestUltraJSONTests.test_double_long_numbersc
|
C sj t j}dD ]Z}t ||¡r
|
t ||¡4 t t d¡¡dks@ttjddddksVtW 5 Q R X qfq
|
d S )N)z it_IT.UTF-8Z Italian_ItalygòìÄûËLz4.78T©Z precise_floatg
ëQ¸ @) ÚlocaleÚ
|
LC_NUMERICÚtmZcan_set_localeZ
|
set_localer, r8 Údumpsr/ )r1 Z lc_categoryZ
|
new_localer r r Útest_encode_non_c_locale s z+TestUltraJSONTests.test_encode_non_c_localec C s0 ddi}t |¡}t j|dd}||ks,td S )Nr@ g=
|
×£p=@TrC rA r0 r r r Ú test_decimal_decode_test_precise s
|
z3TestUltraJSONTests.test_decimal_decode_test_precisec C sx d}|t t |¡¡ks td}|t t |¡¡ks8td}|t t |¡¡ksTtd}t |t t |¡¡¡sttd S )NgWw'&l¡7g0ä.ÿ++g¹j7Ö¶gó¡?>¬úÓ¡)r, r. r- r/ ÚnpZallclose)r1 Únumr r r Ú#test_encode_double_tiny_exponential¥ s z6TestUltraJSONTests.test_encode_double_tiny_exponentialÚ unicode_keyÚkey1u بÙc C s$ |di}|t t |¡¡ks td S )NZvalue1©r, r. r- r/ )r1 rM Z unicode_dictr r r Ú"test_encode_dict_with_unicode_keys¯ s z5TestUltraJSONTests.test_encode_dict_with_unicode_keysÚ double_inputc C sJ t |¡}t|dtt |¡dks(tt|dtt |¡dksFtd S ©Né )r, r- Úroundr7 r8 r/ r. )r1 rQ r: r r r Ú test_encode_double_conversion´ s
|
z0TestUltraJSONTests.test_encode_double_conversionc C s d}t |¡}|dkstd S )Nr' r& ©r, r- r/ )r1 Z decimal_inputr: r r r Útest_encode_with_decimal¼ s
|
z+TestUltraJSONTests.test_encode_with_decimalc C s@ g gggd }t |¡}|t |¡ks*t|t |¡ks<td S )Né ©r, r- r7 r8 r/ r. )r1 Z nested_inputr: r r r Ú"test_encode_array_of_nested_arrays s
|
z5TestUltraJSONTests.test_encode_array_of_nested_arraysc C sB ddddgd }t |¡}|t |¡ks,t|t |¡ks>td S )Ngc ATÞ@é
|
rY )r1 Z doubles_inputr: r r r Ú test_encode_array_of_doublesÉ s
|
z/TestUltraJSONTests.test_encode_array_of_doublesc C s d}t j|dd}|t |¡ks$t|t |¡ks6tdD ]@}t j||d}t||}|t |¡ksht|t |¡ks:tq:d S )Ng² )>@r" r# )r) é )r, r- r7 r8 r/ r. rT )r1 rQ r: r$ Z rounded_inputr r r Útest_double_precisionÐ s
|
z(TestUltraJSONTests.test_double_precisionÚ invalid_valrX éÿÿÿÿÚ9Nc C sF d}t |trtnt}d}tj||d tj||d W 5 Q R X d S )NgöF7Ý >@zInvalid value '.*' for option 'double_precision', max is '15'|an integer is required \(got type |object cannot be interpreted as an integer©Úmatchr# )Ú
|
isinstanceÚintÚ
|
ValueErrorÚ TypeErrorÚpytestÚraisesr, r- )r1 r_ rQ Úexpected_exceptionÚmsgr r r Ú test_invalid_double_precisionÞ s
|
ÿz0TestUltraJSONTests.test_invalid_double_precisionc C sB d}t |¡}|t |¡ks t|t |¡ks2t|dks>td S )NzA string \ /
|
z "A string \\ \/ \b \f \n \r \t"rY )r1 r< r: r r r Ú test_encode_string_conversion2ò s
|
|
z1TestUltraJSONTests.test_encode_string_conversion2Ú unicode_inputu@ RäksmörgÃ¥s اساÙ
Ø© ب٠Ù
ØÙ
د ب٠عÙض ب٠ÙادÙõ
|
æÂÂ¥ÃÂc C s< t |¡}t |¡}|t |¡ks&t|t |¡ks8td S r ©r, r- r. r7 rG r/ r8 )r1 rn ÚencÚdecr r r Ú test_encode_unicode_conversionú s
|
|
z1TestUltraJSONTests.test_encode_unicode_conversionc C s: d}t |¡}t |¡}||ks$t|t |¡ks6td S )Nú)r, r- r. r/ r7 rG )r1 Z escaped_inputrq rr r r r Ú test_encode_control_escaping s
|
|
|
z/TestUltraJSONTests.test_encode_control_escapingc C s@ d}t |¡}t |¡}|t |¡ks*t|t |¡ks<td S )Nu ðÂÂÂrp )r1 Zsurrogate_inputrq rr r r r Ú"test_encode_unicode_surrogate_pair s
|
|
|
z5TestUltraJSONTests.test_encode_unicode_surrogate_pairc C s@ d}t |¡}t |¡}|t |¡ks*t|t |¡ks<td S )Nu ð°TRAILINGNORMALrp ©r1 Zfour_bytes_inputrq rr r r r Ú test_encode_unicode_4bytes_utf8 s
|
|
|
z2TestUltraJSONTests.test_encode_unicode_4bytes_utf8c C s@ d}t |¡}t |¡}|t |¡ks*t|t |¡ks<td S )Nu ó¿¿¿TRAILINGNORMALrp rw r r r Ú&test_encode_unicode_4bytes_utf8highest s
|
|
|
z9TestUltraJSONTests.test_encode_unicode_4bytes_utf8highestc C s2 d}d}t jt|d t |g¡ W 5 Q R X d S )Nu 'í«'zT'utf-8' codec can't encode character '\\udac0' in position 1: surrogates not allowedrb )rh ri ÚUnicodeEncodeErrorr, rG )r1 Ústringrk r r r Útest_encode_unicode_error& s
|
ÿz,TestUltraJSONTests.test_encode_unicode_errorc C sN g ggg}t |¡}|t |¡ks&t|t |¡ks8t|t |¡ksJtd S r ©r, r- r7 r8 r/ rG r. )r1 Zarr_in_arr_inputr: r r r Útest_encode_array_in_array/ s
|
|
|
z-TestUltraJSONTests.test_encode_array_in_arrayÚ num_inputéiz i
ÿÿìûÿÿÿ c C sD t |¡}|t |¡ks t|t |¡ks.t|t |¡ks@td S r r} )r1 r r: r r r Útest_encode_num_conversion7 s
|
z-TestUltraJSONTests.test_encode_num_conversionc C s> ddddg}t |¡}|t |¡ks(t|t |¡ks:td S )Nr% r( r) é rY )r1 Z
|
list_inputr: r r r Útest_encode_list_conversionE s
|
z.TestUltraJSONTests.test_encode_list_conversionc C s@ ddddd}t |¡}|t |¡ks*t|t |¡ks<td S )Nr% r( r) r )Zk1Zk2Zk3Zk4rY )r1 Z
|
dict_inputr: r r r Útest_encode_dict_conversionL s
|
z.TestUltraJSONTests.test_encode_dict_conversionÚ builtin_valuec C sD t |¡}|t |¡ks t|t |¡ks.t|t |¡ks@td S r r} )r1 r r: r r r Ú%test_encode_builtin_values_conversionS s
|
z8TestUltraJSONTests.test_encode_builtin_values_conversionc C s\ t j t ¡ ¡}tj|dd}t | ¡ ¡}t|t
|
|¡ksBt t|t |¡ksXt d S )NÚs©Ú date_unit) ÚdatetimeÚ fromtimestampÚtimer, r- ÚcalendarÚtimegmÚ utctimetuplere r7 r8 r/ r. )r1 Zdatetime_inputr: Úexpectedr r r Ú test_encode_datetime_conversionZ s
|
z2TestUltraJSONTests.test_encode_datetime_conversionc C sn t j t ¡ ¡}tj|dd}|j|j|jdddf}t
|
|¡}t |t |¡ksTtt |t |¡ksjtd S )Nr r r )r Údater r r, r- ÚyearÚmonthÚdayr r re r7 r8 r/ r. )r1 Z
|
date_inputr: Útupr r r r Útest_encode_date_conversionb s
|
z.TestUltraJSONTests.test_encode_date_conversionÚtestr% r( r) r[ é r" éË< c C s* t |¡}d| ¡ d}||ks&td S )Nú"©r, r- Ú isoformatr/ ©r1 r r: r r r r Ú!test_encode_time_conversion_basicl s
|
z4TestUltraJSONTests.test_encode_time_conversion_basicc C s> t ddddtj¡}t |¡}d| ¡ d}||ks:td S ©Nr[ r r" r r )r r ÚpytzÚutcr, r- r r/ r r r r Ú test_encode_time_conversion_pytzu s
|
z3TestUltraJSONTests.test_encode_time_conversion_pytzc C sB t ddddtj ¡ ¡}t |¡}d| ¡ d}||ks>td S r¡ ) r r ÚdateutilÚtzZtzutcr, r- r r/ r r r r Ú$test_encode_time_conversion_dateutil| s
|
z7TestUltraJSONTests.test_encode_time_conversion_dateutilÚ decoded_inputr c C s t |¡dkstdd S )NÚnullz Expected nullrV )r1 r¨ r r r Útest_encode_as_null s z&TestUltraJSONTests.test_encode_as_nullc C sè t ddddddd¡}t| d¡}t tj|dd ¡}||jd
|
ksJtt tj|d d ¡}||jd ksptt tj|d d ¡}||jd kstt tj|dd ¡}||jks¸td}tj t
|
|d tj|dd W 5 Q R X d S )NéÝ é é é r iÐI Únsr r i Ê;Úmsi@B Úuséè z*Invalid value 'foo' for option 'date_unit'rb Úfoo) r r
|
Zas_unitr, r. r- Ú_valuer/ rh ri rf )r1 ÚvalZstampZ roundtriprk r r r Útest_datetime_units s z&TestUltraJSONTests.test_datetime_unitsc C sH d}t j|dd}t |¡}|tj|ddks2t|t |¡ksDtd S )Nro F)r6 rp )r1 Ú unencodedrq rr r r r Útest_encode_to_utf8 s
|
|
z&TestUltraJSONTests.test_encode_to_utf8c C s, d}t |¡}t t|¡}||ks(td S )Nz{"obj": 31337}©r, r. r r/ )r1 rn Zdec1Zdec2r r r Útest_decode_from_unicode¦ s
|
z+TestUltraJSONTests.test_decode_from_unicodec C sZ G dd d}G dd d}| }| |_ ||j _ tjtdd t |¡ W 5 Q R X d S )Nc @ s e Zd ZdZdS )z8TestUltraJSONTests.test_encode_recursion_max.<locals>.O2r N©Ú__name__Ú
|
__module__Ú __qualname__Úmemberr r r r ÚO2± s rÀ c @ s e Zd ZdZdS )z8TestUltraJSONTests.test_encode_recursion_max.<locals>.O1r Nr» r r r r ÚO1´ s rÁ ú Maximum recursion level reachedrb )r¿ rh ri Ú OverflowErrorr, r- )r1 rÀ rÁ r¨ r r r Útest_encode_recursion_max® s z,TestUltraJSONTests.test_encode_recursion_maxc C s0 d}d}t jt|d t |¡ W 5 Q R X d S )Nzfdsa sda v9sa fdsaz0Unexpected character found when decoding 'false'rb ©rh ri rf r, r. )r1 Z jibberishrk r r r Útest_decode_jibberish¾ s z(TestUltraJSONTests.test_decode_jibberishÚ broken_jsonú[Ú{ú]Ú}c C s, d}t jt|d t |¡ W 5 Q R X d S )NzExpected object or valuerb rÅ )r1 rÇ rk r r r Útest_decode_broken_jsonÄ s
|
z*TestUltraJSONTests.test_decode_broken_jsonÚ too_big_charc C s, t jtdd t |d ¡ W 5 Q R X d S )Nz#Reached object decoding depth limitrb i rÅ )r1 rÍ r r r Útest_decode_depth_too_bigÒ s z,TestUltraJSONTests.test_decode_depth_too_bigÚ
|
bad_stringz"TESTINGz
|
"TESTING\"ZtruÚfaÚnc C s, d}t jt|d t |¡ W 5 Q R X d S )NzSUnexpected character found when decoding|Unmatched ''"' when when decoding 'string'rb rÅ )r1 rÏ rk r r r Útest_decode_bad_string× s ÿz)TestUltraJSONTests.test_decode_bad_stringzbroken_json, err_msg)z {{1337:""}}z:Key name of object must be 'string' when decoding 'object')z
|
{{"key":"}z*Unmatched ''"' when when decoding 'string')z[[[truez8Unexpected character found when decoding array value (2)c
|
C s< t dD ].}tjtt |¡d t |¡ W 5 Q R X qd S )Nr² rb )Úrangerh ri rf ÚreÚescaper, r. )r1 rÇ Úerr_msgÚ_r r r Ú test_decode_broken_json_leaké s z/TestUltraJSONTests.test_decode_broken_json_leakÚ invalid_dictz {{{{31337}}}}z{{{{"key":}}}}z {{{{"key"}}}}c C s, d}t jt|d t |¡ W 5 Q R X d S )Nz{Key name of object must be 'string' when decoding 'object'|No ':' found when decoding object value|Expected object or valuerb rÅ )r1 rÙ rk r r r Útest_decode_invalid_dictù s
|
ÿz+TestUltraJSONTests.test_decode_invalid_dictÚnumeric_int_as_strZ31337z-31337c C s t |t |¡kstd S r )re r, r. r/ )r1 rÛ r r r Útest_decode_numeric_int
|
s z*TestUltraJSONTests.test_decode_numeric_intc C s d}t |¡}|t |¡ks t|t |¡ks2t|t |¡ksDtd}t |¡}|t |¡ksdt|t |¡ksvt|t |¡kstdt d¡kstd S )Nz 31337 1337ú z" \u0000\r\n "z
|
r} )r1 Ú wrapped_inputr: Z alone_inputr r r Útest_encode_null_character s
|
|
z-TestUltraJSONTests.test_encode_null_characterc C s d}t |¡t |¡ks td S )Nz"31337 \u0000 31337"©r, r. r7 r8 r/ )r1 rÞ r r r Útest_decode_null_character s z-TestUltraJSONTests.test_decode_null_characterc C sB ddddddg}t |¡}|t |¡ks,t|t |¡ks>td S )Nì ÿÿÿÿ rY ©r1 Ú
|
long_inputr: r r r Ú test_encode_list_long_conversion$ s ú
|
z3TestUltraJSONTests.test_encode_list_long_conversionrä râ l ÿÿÿÿ c C sD t |¡}|t |¡ks t|t |¡ks.t|t |¡ks@td S r r} rã r r r Útest_encode_long_conversion2 s
|
z.TestUltraJSONTests.test_encode_long_conversionÚbigNumì ìûÿÿÿ c C sJ t |¡}t||ksttjtdd t |¡|ks<tW 5 Q R X d S ©Nz#Value is too big|Value is too smallrb )r, r- r r/ rh ri rf r8 )r1 rç Úencodingr r r Ú#test_dumps_ints_larger_than_maxsize: s
|
þz6TestUltraJSONTests.test_dumps_ints_larger_than_maxsizeÚint_expZ1337E40z1.337E40z1337E+9z 1.337e+40z1.337E-4c C s t |¡t |¡kstd S r rà )r1 rí r r r Útest_decode_numeric_int_expE s z.TestUltraJSONTests.test_decode_numeric_int_expc C s, d}t jt|d t d ¡ W 5 Q R X d S )NzExpected 'str' or 'bytes'rb )rh ri rg r, r8 ©r1 rk r r r Ú test_loads_non_str_bytes_raisesK s z2TestUltraJSONTests.test_loads_non_str_bytes_raisesrµ l £Mö+ l l l ÿÿ c C s& d| d}t |¡d |ks"td S )Nz{"id": rË Úid©r, r. r/ )r1 rµ Údocr r r Ú&test_decode_number_with_32bit_sign_bitP s z9TestUltraJSONTests.test_decode_number_with_32bit_sign_bitc C s4 t dD ]&}d ¡ }|d d d }t |¡ qd S )Nr[ õ Ã¥é r( )rÓ r- r, )r1 r× ÚbaseÚ escape_inputr r r Útest_encode_big_escapeW s z)TestUltraJSONTests.test_encode_big_escapec C s@ t dD ]2}d ¡ }d}||d d d | }t |¡ qd S )Nr[ rõ ó "rö r( )rÓ r- r, r. )r1 r× r÷ Úquoterø r r r Útest_decode_big_escape^ s
|
z)TestUltraJSONTests.test_decode_big_escapec sD ddi G fddd}| }t |¡}t |¡}| ks@td S )NÚkeyr c s e Zd Z fddZdS )z1TestUltraJSONTests.test_to_dict.<locals>.DictTestc s S r r ©r1 r r r ÚtoDictk s z8TestUltraJSONTests.test_to_dict.<locals>.DictTest.toDictN)r¼ r½ r¾ rÿ r r r r ÚDictTestj s r rA )r1 r Úor: rr r r r Ú test_to_dictg s
|
|
z TestUltraJSONTests.test_to_dictc sX G fddd d}t jt|d t d¡ W 5 Q R X dtj dtdksXtdd }d
|
tj d|dksztd d }t jtd d tj d|d W 5 Q R X dd }t tj d|d¡dksÖtdd }t t t ddd¡¡¡t tj d|d¡kst d dg}t
|
t
|
j |td¡t tj|td¡ksTtd S )Nc s8 e Zd ZddddZe fddZedddZdS ) z<TestUltraJSONTests.test_default_handler.<locals>._TestObjectN©Úreturnc S s
|
|| _ d S r )rµ )r1 rµ r r r Ú__init__v s zETestUltraJSONTests.test_default_handler.<locals>._TestObject.__init__c s dS )NÚrecursive_attrr rþ ©Ú _TestObjectr r r y s zKTestUltraJSONTests.test_default_handler.<locals>._TestObject.recursive_attrc S s
|
t | jS r )r rµ rþ r r r Ú__str__} s zDTestUltraJSONTests.test_default_handler.<locals>._TestObject.__str__)r¼ r½ r¾ r Úpropertyr r r r r r r r u s r r rb r³ z"foo")Zdefault_handlerc S s dS )NZfoobarr ©r× r r r Ú
|
my_handler
s z;TestUltraJSONTests.test_default_handler.<locals>.my_handlerz"foobar"c S s t dd S )NúI raise for anything)rg r r r r Úmy_handler_raises s zBTestUltraJSONTests.test_default_handler.<locals>.my_handler_raisesr c S s dS )Né* r r r r r Úmy_int_handler s z?TestUltraJSONTests.test_default_handler.<locals>.my_int_handlerr c S s t ddd¡S )Nr« r( r) )r r r r r Úmy_obj_handler s z?TestUltraJSONTests.test_default_handler.<locals>.my_obj_handlerr« r( r) Úbar)Údefault) rh ri rà r, r- r r/ rg r. r r7 r8 rG )r1 rk r r r r Zobj_listr r r Útest_default_handlert s@ ÿ ÿýÿÿÿþ
|
ÿz'TestUltraJSONTests.test_default_handlerc C sB G dd d}|ddddd}t t |¡¡ddddks>td S ) Nc @ s" e Zd ZddddZdd ZdS )z:TestUltraJSONTests.test_encode_object.<locals>._TestObjectNr c S s || _ || _|| _|| _d S r ©r@ ÚbÚ_cr )r1 r@ r r r r r r r ¬ s zCTestUltraJSONTests.test_encode_object.<locals>._TestObject.__init__c S s dS rR r rþ r r r Úe² s z<TestUltraJSONTests.test_encode_object.<locals>._TestObject.e)r¼ r½ r¾ r r r r r r r « s r r% r( r) r r )r@ r r rO )r1 r Z test_objectr r r Útest_encode_objectª s z%TestUltraJSONTests.test_encode_object)Hr¼ r½ r¾ rh ÚmarkZskipifr r5 Ú parametrizer> rB rH rI rL rP ÚmathÚpirU rW rZ r\ r^ rl rm rs ru rv rx ry r| r~ r r r
r r r r r r r¤ r§ r rJ Z
|
datetime64ÚnanÚinfrª r¶ r¸ rº rÄ rÆ rÌ rÎ rÒ rØ rÚ rÜ rß rá rå ræ rì rî rð rô rù rü r r r r r r r r 7 s
|
7
|
ÿ
|
|
|
ÿ
|
üþ
|
þ
|
ýþ
|
|
|
"þ
|
ÿ
|
üþ
|
|
ûþ
|
|
úþ
|
ýþ
|
ÿ
|
|
|
|
ÿ
|
|
6r c @ s e Zd Zej dddg¡dd Zdd Zdd Zd
|
d Z d d Z
|
dd Z dd Z dd Z dd Zej ddddg¡dd Zd d Zd d Zd d! Zd"S )#ÚTestNumpyJSONTestsÚ
|
bool_inputTFc C s$ t |}t t |¡¡|ks td S r )Úboolr, r. r- r/ )r1 r! r r r r Ú test_bool» s z TestNumpyJSONTests.test_boolc C sH t jddddddddgtd}t jt t |¡¡td}t ||¡ d S )NTF©Údtype)rJ Úarrayr" r, r. r- rF Úassert_numpy_array_equal)r1 Z
|
bool_arrayr: r r r Útest_bool_arrayÀ s ÿz"TestNumpyJSONTests.test_bool_arrayc C s4 t |¡j}|d}|t t |¡¡|ks0td S )Nr% ©rJ r% Útyper, r. r- r/ ©r1 Úany_int_numpy_dtypeÚklassrK r r r Útest_intÇ s zTestNumpyJSONTests.test_intc C sB t jdtd}| |¡}t jt t |¡¡|d}t ||¡ d S )Néd r$ )
|
rJ Úarangere Úastyper& r, r. r- rF r' )r1 r, ÚarrZ arr_inputZ
|
arr_outputr r r Útest_int_arrayÍ s
|
ÿz!TestNumpyJSONTests.test_int_arrayc C sd |dkrt st d¡ t |¡j}|dkr8t d¡j}n t |¡j}|t t
|
|¡¡|ks`t d S )N)Úint64Úuint64z-Cannot test 64-bit integer on 32-bit platformr5 r4 ) r rh ÚskiprJ r% r* ZiinfoÚmaxr, r. r- r/ r+ r r r Ú test_int_maxÖ s
|
z TestNumpyJSONTests.test_int_maxc C s4 t |¡j}|d}|t t |¡¡|ks0td S )NgõJY8p@r) ©r1 Úfloat_numpy_dtyper- rK r r r Ú
|
test_floatå s z TestNumpyJSONTests.test_floatc C sJ t jdddtd}| |¡}t jt tj|dd¡|d}t ||¡ d S )Ng )@g×£p=
|
7g@geâX·û?r$ r" r# )
|
rJ r0 Úfloatr1 r& r, r. r- rF Úassert_almost_equal)r1 r: r2 Z float_inputZ float_outputr r r Útest_float_arrayë s
|
þz#TestNumpyJSONTests.test_float_arrayc C sD t |¡j}|t |¡jd }t |t tj |dd¡|¡ d S )Nr[ r" r# )
|
rJ r% r* Zfinfor7 rF r= r, r. r- r9 r r r Útest_float_maxõ s ÿz!TestNumpyJSONTests.test_float_maxc C s6 t d¡}| d¡}t t t t |¡¡¡|¡ d S )Né` )r( r( r( r( r) r( © rJ r0 ÚreshaperF r' r& r, r. r- ©r1 r2 r r r Útest_array_basicý s
|
|
z#TestNumpyJSONTests.test_array_basicÚshape)r[ r[ ©rS rS r )r/ r% c C s6 t d¡}| |¡}t t t t |¡¡¡|¡ d S )Nr/ rA )r1 rE r2 r r r Útest_array_reshaped s
|
|
z&TestNumpyJSONTests.test_array_reshapedc
|
C sV dg i i g ddddgddig }t j|td}t jt t |¡¡td}t ||¡ d S )Nr@ r g33333sX@r rý rµ r$ )rJ r& Úobjectr, r. r- rF r' )r1 Zarr_listr2 Úresultr r r Útest_array_list
|
s ÷ z"TestNumpyJSONTests.test_array_listc C sL t j}t jddd|d}| d¡}t jt t |¡¡|d}t ||¡ d S )Ng°rhí Y@gX9´Èvi@r% r$ rF )
|
rJ Úfloat32r0 rB r& r, r. r- rF r= )r1 r% r2 Zarr_outr r r Útest_array_float s
|
|
z#TestNumpyJSONTests.test_array_floatc C s8 t d¡}tjt|d t t d¡¡ W 5 Q R X d S )Nz:array(1) (0d array) is not JSON serializable at the momentrb r% ) rÔ rÕ rh ri rg r, r- rJ r& rï r r r Ú test_0d_array# s
|
z TestNumpyJSONTests.test_0d_arrayN)r¼ r½ r¾ rh r r r# r( r. r3 r8 r; r> r? rD rG rJ rL rM r r r r r º s
|
|
|
r c @ s¨ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d ddddg¡dd Z e j
|
dg dgg¡dd Z e j
|
dddg¡dd Ze j
|
d d d g¡d d! Zd"d# Zd$d% Ze j
|
d&d d g¡d'd( Ze j
|
d)d*d+d,d-d.d/d0d1d2g ¡e j
|
d3d4d5g¡d6d7 Zd8d9 Zd:d; Zd<d= Ze j
|
d>ed?d@ed4d@edAdBdBdCedAdDdEdCed4dBdFed5dGed5d5dHed5d5d5dIedJdJdJdIg ¡dKdL ZdMdN ZdOS )PÚTestPandasJSONTestsc C s t j}tdddgdddggddgd d
|
d g|d }|d kr<i nd |i}|j|k ¡ sVtt tj|f|¡}|j|k ¡ s|t|dkrt |}tf |}nt|}|dkrÀdddg|_
|
ddg|_ n$|dkrÔddg|_ n|dkrä| ¡ }|j|k ¡ sött ||¡ d S )Nr% r( r) r rS é r@ r ÚxÚyÚz)r Úcolumnsr% r r r r r r )rJ r4 r ZdtypesÚallr/ r, r. r- r rS r Z transposerF Zassert_frame_equal)r1 r r% Údfr9 r: rr r r r Útest_dataframe+ s0 ü z"TestPandasJSONTests.test_dataframec C s t dddgdddggddgd d
|
d gd }|| ¡ d }|d krBi nd|i}t tj|f|¡t tj|f|¡d }t tj|f|¡|kstd S )Nr% r( r) r rS rO r@ r rP rQ rR )r rS )Zdf1Zdf2r )r Úcopyr, r. r- r/ )r1 r rU ÚnestedÚkwargsÚexpr r r Útest_dataframe_nestedM s ÿþz)TestPandasJSONTests.test_dataframe_nestedc C s t j}tddddddgddd d
|
d dd g|d ¡ }|j|ks@t|d krLi nd|i}t tj|f|¡}|j|ksvt|dkrt |}tf |}nt|}|dkrÄd |_
|
| ¡ }ddddddg|_ n |dkräd |_
|
ddddd d g|_ |j|ksòtt ||¡ d S ) Nr[ rX é é( é2 é< ÚseriesrO é r¬ r] r" )Únamer r% r r )Nr Ú6Ú7Ú8ra Ú10Z15)r r r r% r( r) r rS )rJ r4 r Ú sort_valuesr% r/ r, r. r- r rb r rF Úassert_series_equal)r1 r r% r r9 r: rr r r r Ú test_series[ s0 ü
|
z TestPandasJSONTests.test_seriesc C s t ddddddgddd d
|
d dd gd ¡ }|| ¡ d}|d krDi nd|i}t tj|f|¡t tj|f|¡d}t tj|f|¡|kstd S )Nr[ rX r\ r] r^ r_ r` rO ra r¬ r] r" )rb r )Ús1Ús2r )r rg rW r, r. r- r/ )r1 r r rX rY rZ r r r Útest_series_nested{ s ÿ
|
þz&TestPandasJSONTests.test_series_nestedc C s t ddddddgdd}t t t |¡¡dd}t ||¡ tt tj|d d
|
¡}t f |}t ||¡ |j|jksztt ||¡ |j|jkstt t tj|d d
|
¡dd}t ||¡ t t tj|d d
|
¡dd}t ||¡ t t tj|dd
|
¡dd}t ||¡ d S ) Né é- é éb é+ é r )rb r )r r r ) r r, r. r- rF Úassert_index_equalr rb r/ )r1 Úir: rr r r r Ú
|
test_index s
|
z TestPandasJSONTests.test_indexc C s¤ d}t ttdddd d}tj||d}t t t |¡¡}t ||¡ t
|
tj t |¡|d}t
|
t tj||d¡}|jj tj¡}t ||_t ||¡ d S )Nr¯ z1/1/2000rX )Zperiods©Úfreqr ©r )r Úlistr r, r- rJ r& r. rF rs r ÚrandomZrandnÚlenr r r1 r4 rh )r1 r Úrngr3 r4 ÚtsZ
|
idx_valuesr r r Útest_datetime_index¡ s
|
z'TestPandasJSONTests.test_datetime_indexÚ invalid_arrz[31337,]z[,31337]z[]]z[,]c C s, d}t jt|d t |¡ W 5 Q R X d S )Nz[Expected object or value|Trailing data|Unexpected character found when decoding array valuerb rÅ )r1 r rk r r r Útest_decode_invalid_array² s ÿz-TestPandasJSONTests.test_decode_invalid_arrayr2 r c C s |t t|¡kstd S r r¹ rC r r r Útest_decode_arrayà s z%TestPandasJSONTests.test_decode_arrayÚ extreme_numrâ r c C s |t t|¡kstd S r r¹ )r1 r r r r Útest_decode_extreme_numbersÇ s z/TestPandasJSONTests.test_decode_extreme_numbersÚtoo_extreme_numrè ré c C s( t jtdd t |¡ W 5 Q R X d S rê rÅ )r1 r r r r Ú test_decode_too_extreme_numbersË s
|
þz3TestPandasJSONTests.test_decode_too_extreme_numbersc C s i t d¡kstd S )Nz{}
|
rò rþ r r r Ú%test_decode_with_trailing_whitespacesÓ s z9TestPandasJSONTests.test_decode_with_trailing_whitespacesc C s( t jtdd t d¡ W 5 Q R X d S )Nz Trailing datarb z{}
|
arÅ rþ r r r Ú)test_decode_with_trailing_non_whitespacesÖ s z=TestPandasJSONTests.test_decode_with_trailing_non_whitespacesÚvaluec C s( t jtdd t |¡ W 5 Q R X d S rê )rh ri rf r, r8 )r1 r r r r Ú test_decode_array_with_big_intÚ s
|
þz2TestPandasJSONTests.test_decode_array_with_big_intÚ float_numbergÞ3Óùñ?gªRCÊÀó?g¨B4¡æõ?g
|
LO÷?g jøù?g[_$´åÜú?g¹k ù ü?g}?5^ºIþ?gÍÌÌÌÌÌô?Úsignr` r% c C s& ||9 }t j|t t|¡dd d S )NgVç¯Ò<)Zrtol)rF r= r, r8 r )r1 r r r r r Útest_decode_floating_pointâ s z.TestPandasJSONTests.test_decode_floating_pointc C s. t }tddD ]}| |¡ qt |¡ d S )Nr i )ÚsetrÓ Úaddr, r- )r1 r rP r r r Útest_encode_big_setõ s z'TestPandasJSONTests.test_encode_big_setc C s dt t ¡kstd S )Nz[])r, r- r r/ rþ r r r Útest_encode_empty_setþ s z)TestPandasJSONTests.test_encode_empty_setc C sD ddddddddd h }t |¡}t |¡}|D ]}||ks.tq.d S )
|
Nr% r( r) r rS rO ra r¬ r] rA )r1 r rq rr r r r r Útest_encode_set s
|
|
|
z#TestPandasJSONTests.test_encode_setÚtdin )Údaysé rS )ÚhoursÚminutesÚsecondsrX r\ )r Ú nanoseconds)r )Ú microsecondsr )Z millisecondsr r iç c C s. t j|dd}d| ¡ d}||ks*td S )NT)Z iso_datesr r )r1 r rI r r r r Útest_encode_timedelta_iso s z-TestPandasJSONTests.test_encode_timedelta_isoc C s. t ddgdd}t|d}| ¡ dks*td S )Nz
|
2022-04-06z
|
2022-04-07ÚDrv rx z{})r r Úto_jsonr/ )r1 ÚprU r r r Útest_encode_periodindex s
|
z+TestPandasJSONTests.test_encode_periodindexN)r¼ r½ r¾ rV r[ ri rl ru r~ rh r r r r r r
r r r r r r r r r r r r r r rN * st " üþ
|
|
|
|
|
÷þ
|
|
÷þ
|
rN )%r r r* r7 rD r rÔ r r¥ ÚnumpyrJ rh r¢ Zpandas._libs.jsonZ_libsr, Z pandas.compatr Zpandasr r r r r r r r
|
r Zpandas._testingZ_testingrF r Zfixturer r r rN r r r r Ú<module> s8 , ÿ
|
p
|
