Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/io/json/__pycache__/test_ujson.cpython-37.pyc
Назад
B �5�g�� � @ s6 d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl Z d dl Z d dl Z d dlm mZ d dlmZmZmZ d dlmZmZmZmZmZmZm Z m Z d dl m Z dd� Z!e j"ddddd gd �d d � �Z#e j"dd gd �dd� �Z dd� Z$G dd� d�Z%G dd� d�Z&G dd� d�Z'dS )� N)�IS64�PY310�is_platform_windows)� DataFrame� DatetimeIndex�Index�NaT�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 �U/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/io/json/test_ujson.py� <dictcomp>1 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 �orient4 s r Tc C s | j S )N)r )r r r r �numpy; s r c C s> t j}| d ks:|dks:| dkr&|d ks:t� r4t j}nt j}|S )Nr T)�np�int64r Zint32Zintp)r r �dtyper r r �get_int32_compat_dtype@ s r# c @ s� e Zd Zejje dd�dd� �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(ejd)ejjed*d�d+�ejd,ejjed*d�d+�g�d-d.� �Zd/d0� Zej�d1d2d3g�d4d5� �Zd6d7� Z d8d9� Z d: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-j0e-j0 g�dadb� �Z1dcdd� Z2dedf� Z3dgdh� Z4didj� Z5dkdl� Z6ej�dmdndodpdqg�drds� �Z7ej�dtdndog�dudv� �Z8ej�dwdxdydzd{d|g�d}d~� �Z9ej�dd�d�d�g�d�d�� �Z:ej�d�d�d�d�g�d�d�� �Z;ej�d�d�d�g�d�d�� �Z<d�d�� Z=d�d�� Z>d�d�� Z?d�d�� Z@ej�d�eAjBdR eAjBdS g�ejje d�d�d�d�� ��ZCej�d�d�d�d�d�d�g�d�d�� �ZDd�d�� ZEej�d�d�d�d�d�g�d�d�� �ZFd�d�� ZGd�d�� ZHd�d�� ZId�d�� ZJd,S )��TestUltraJSONTestsz$not compliant on 32-bit, xref #15865)�reasonc C s� t �d�}tj|dd�}t�|�}|dks.t�t �d�}tj|dd�}|dksRt�t�|�}|dksht�t �d �}tj|dd�}|d ks�t�t�|�}|d ks�t�t �d �}tj|dd�}|d ks�t�t�|�}|dks�t�t �d�}tj|dd�}|dk�st�t�|�}|dk�st�t �d�}tj|dd�}|dk�s@t�t�|�}|dk�sXt�t �d�}tj|dd�}|dk�s~t�t�|�}|dk�s�t�t �d�}tj|dd�}|dk�s�t�t�|�}|dk�s�t�d S )Nz 1337.1337� )�double_precisiong� ���@z0.95� z1.0g �?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_decimalM sN z&TestUltraJSONTests.test_encode_decimal� ensure_asciiTFc s> d�d}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 j�fd� i|��}|| ks"t��t�|�ks4t��t �|�ksFt�d S )Nr6 )r- r. r0 �json�loadsr/ )Zexpected_output� encode_kwargs�output)r6 � string_inputr r �helper� s z@TestUltraJSONTests.test_encode_string_conversion.<locals>.helperF)Zencode_html_charsTr )r1 r6 Znot_html_encodedZ html_encodedr< r )r6 r; r � test_encode_string_conversion� s z0TestUltraJSONTests.test_encode_string_conversion� long_numberl����Z:�`�7{ g#�_��t��gԟpA����c C s0 d|i}t j|dd�}t �|�}||ks,t�d S )N�ar& )r' )r- r. r/ r0 )r1 r> r2 r3 r4 r r r �test_double_long_numbers� s z+TestUltraJSONTests.test_double_long_numbersc C sl t j}x`dD ]X}t�||�r t�||��4 t�t�d��dksBt�tjddd�dksXt�W d Q R X P q W d S )N)z it_IT.UTF-8Z Italian_Italyg���ˇLz4.78T)� precise_floatg ��Q� @) �locale� LC_NUMERIC�tmZcan_set_localeZ set_localer- r8 �dumpsr0 )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,t�d S )Nr? g= ףp=@T)rA )r- r. r/ r0 )r1 r2 r3 r4 r r r � test_decimal_decode_test_precise� s z3TestUltraJSONTests.test_decimal_decode_test_precisec C sx d}|t �t �|��ks t�d}|t �t �|��ks8t�d}|t �t �|��ksTt�d}t�|t �t �|���stt�d S )Ng�Ww'&l�7g0��.�++g�j7�֖�g�?>��ӡ)r- r/ r. r0 r Zallclose)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 t�d S )NZvalue1)r- r/ r. r0 )r1 rJ 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|d�tt�|�d�ks(t�t|d�tt �|�d�ksFt�d S )N� )r- r. �roundr7 r8 r0 r/ )r1 rM r: r r r � test_encode_double_conversion� s z0TestUltraJSONTests.test_encode_double_conversionc C s d}t �|�}|dkst�d S )Ng �?z1.0)r- r. r0 )r1 Z decimal_inputr: r r r �test_encode_with_decimal� s z+TestUltraJSONTests.test_encode_with_decimalc C sd g gggd }t �|�}|t�|�ks*t�|t �|�ks<t�t�|�}t� |t j|d|j d�� d S )N� T)r r" ) r- r. r7 r8 r0 r/ r �arrayrD �assert_numpy_array_equalr" )r1 Z nested_inputr: r r r �"test_encode_array_of_nested_arrays� s z5TestUltraJSONTests.test_encode_array_of_nested_arraysc C s^ ddddgd }t �|�}|t�|�ks,t�|t �|�ks>t�t�t� |�t j|dd�� d S )Ngc AT��@� T)r ) r- r. r7 r8 r0 r/ rD rT r rS )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 �|�ks6t�xHdD ]@}t j||d�}t||�}|t�|�ksjt�|t �|�ks<t�q<W d S )Ng� )>@r& )r' )r* � )r- r. r7 r8 r0 r/ rO )r1 rM r: r' Z rounded_inputr r r �test_double_precision� s z(TestUltraJSONTests.test_double_precision� invalid_valrR ����9z Failing on Python 3.10 GH41940)ZmarksNc C sF d}t |t�rtnt}d}tj||d�� tj||d� W d Q R X d S )Ng�F7ݚ >@z`Invalid value '.*' for option 'double_precision', max is '15'|an integer is required \(got type )�match)r' )� isinstance�int� ValueError� TypeError�pytest�raisesr- r. )r1 rZ rM �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>t�d S )NzA string \ / z "A string \\ \/ \b \f \n \r \t")r- r. r7 r8 r0 r/ )r1 r; r: r r r � test_encode_string_conversion2 s z1TestUltraJSONTests.test_encode_string_conversion2� unicode_inputu@ Räksmörgås اسامة بن محمد بن عوض بن لادنu æ¥Ñc C s< t �|�}t �|�}|t�|�ks&t�|t�|�ks8t�d S )N)r- r. r/ r7 rE r0 r8 )r1 rh �enc�decr r r � test_encode_unicode_conversion s z1TestUltraJSONTests.test_encode_unicode_conversionc C s: d}t �|�}t �|�}||ks$t�|t�|�ks6t�d S )N�)r- r. r/ r0 r7 rE )r1 Z escaped_inputri rj r r r � test_encode_control_escaping( s z/TestUltraJSONTests.test_encode_control_escapingc C s@ d}t �|�}t �|�}|t�|�ks*t�|t�|�ks<t�d S )Nu ð)r- r. r/ r7 rE r0 r8 )r1 Zsurrogate_inputri rj r r r �"test_encode_unicode_surrogate_pair0 s z5TestUltraJSONTests.test_encode_unicode_surrogate_pairc C s@ d}t �|�}t �|�}|t�|�ks*t�|t�|�ks<t�d S )Nu ð°TRAILINGNORMAL)r- r. r/ r7 rE r0 r8 )r1 �four_bytes_inputri rj r r r � test_encode_unicode_4bytes_utf88 s z2TestUltraJSONTests.test_encode_unicode_4bytes_utf8c C s@ d}t �|�}t �|�}|t�|�ks*t�|t�|�ks<t�d S )Nu ó¿¿¿TRAILINGNORMAL)r- r. r/ r7 rE r0 r8 )r1 ro ri rj r r r �&test_encode_unicode_4bytes_utf8highest@ s z9TestUltraJSONTests.test_encode_unicode_4bytes_utf8highestc C sj g ggg}t �|�}|t�|�ks&t�|t�|�ks8t�|t �|�ksJt�t�t � |�t j|dd�� d S )NT)r ) r- r. r7 r8 r0 rE r/ rD rT r rS )r1 Zarr_in_arr_inputr: r r r �test_encode_array_in_arrayI s z-TestUltraJSONTests.test_encode_array_in_array� num_inputiiz i����l���� c C sD t �|�}|t�|�ks t�|t�|�ks.t�|t �|�ks@t�d S )N)r- r. r7 r8 r0 rE r/ )r1 rs r: r r r �test_encode_num_conversionU s z-TestUltraJSONTests.test_encode_num_conversionc C sZ ddddg}t �|�}|t�|�ks(t�|t �|�ks:t�t�t� |�t j|dd�� d S )Nr( r) r* � T)r ) r- r. r7 r8 r0 r/ rD rT r rS )r1 Z list_inputr: r r r �test_encode_list_conversionc s z.TestUltraJSONTests.test_encode_list_conversionc C s@ ddddd�}t �|�}|t�|�ks*t�|t �|�ks<t�d S )Nr( r) r* ru )Zk1Zk2Zk3Zk4)r- r. r7 r8 r0 r/ )r1 Z dict_inputr: r r r �test_encode_dict_conversionn s z.TestUltraJSONTests.test_encode_dict_conversion� builtin_valuec C sD t �|�}|t�|�ks t�|t�|�ks.t�|t �|�ks@t�d S )N)r- r. r7 r8 r0 rE r/ )r1 rx r: r r r �%test_encode_builtin_values_conversionu 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� utctimetupler_ r7 r8 r0 r/ )r1 Zdatetime_inputr: �expectedr r r � test_encode_datetime_conversion| s z2TestUltraJSONTests.test_encode_datetime_conversionc C sn t j�t�� �}tj|dd�}|j|j|jdddf}t � |�}t |�t � |�ksTt�t |�t�|�ksjt�d S )Nrz )r{ r )r| �dater} r~ r- r. �year�month�dayr r� r_ r7 r8 r0 r/ )r1 Z date_inputr: �tupr� r r r �test_encode_date_conversion� s z.TestUltraJSONTests.test_encode_date_conversion�testr( r) r* rV � r& i�< c C s* t �|�}d|�� � d�}||ks&t�d S )N�")r- r. � isoformatr0 )r1 r� r: r� r r r �!test_encode_time_conversion_basic� s z4TestUltraJSONTests.test_encode_time_conversion_basicc C s> t �ddddtj�}t�|�}d|�� � d�}||ks:t�d S )NrV r� r& i�< r� )r| r~ �pytz�utcr- r. r� r0 )r1 r� r: r� r r r � test_encode_time_conversion_pytz� s z3TestUltraJSONTests.test_encode_time_conversion_pytzc C sB t �ddddtj�� �}t�|�}d|�� � d�}||ks>t�d S )NrV r� r& i�< r� ) r| r~ �dateutil�tzZtzutcr- r. r� r0 )r1 r� r: r� r r r �$test_encode_time_conversion_dateutil� s z7TestUltraJSONTests.test_encode_time_conversion_dateutil� decoded_inputr c C s t �|�dkstd��d S )N�nullz Expected null)r- r. r0 )r1 r� r r r �test_encode_as_null� s z&TestUltraJSONTests.test_encode_as_nullc C s� t � ddddddd�}t|�}t�tj|dd��}||jd ksDt�t�tj|d d��}||jd ksjt�t�tj|d d��}||jd ks�t�t�tj|dd��}||jks�t�d}tjt |d�� tj|dd� W d Q R X d S )Ni� � � � r� i�I rz )r{ i ʚ;�msi@B �usi� �nsz*Invalid value 'foo' for option 'date_unit')r] �foo) r| r r- r/ r. �valuer0 rb rc r` )r1 �valZstampZ roundtripre r r r �test_datetime_units� s z&TestUltraJSONTests.test_datetime_unitsc C sH d}t j|dd�}t �|�}|tj|dd�ks2t�|t�|�ksDt�d S )Nu æ¥ÑF)r6 )r- r. r/ r7 rE r0 r8 )r1 � unencodedri rj r r r �test_encode_to_utf8� s z&TestUltraJSONTests.test_encode_to_utf8c C s, d}t �|�}t �t|��}||ks(t�d S )Nz{"obj": 31337})r- r/ r r0 )r1 rh 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 d 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 N)r� r� r� r� r r r r �O1� s r� z Maximum recursion level reached)r] )r� rb rc � 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 d Q R X d S )Nzfdsa sda v9sa fdsaz0Unexpected character found when decoding 'false')r] )rb rc r` r- r/ )r1 Z jibberishre r r r �test_decode_jibberish� s z(TestUltraJSONTests.test_decode_jibberish� broken_json�[�{�]�}c C s, d}t jt|d�� t�|� W d Q R X d S )NzExpected object or value)r] )rb rc r` r- r/ )r1 r� re r r r �test_decode_broken_json� s z*TestUltraJSONTests.test_decode_broken_json� too_big_charc C s, t jtdd�� t�|d � W d Q R X d S )Nz#Reached object decoding depth limit)r] i )rb rc r` r- 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 d Q R X d S )NzSUnexpected character found when decoding|Unmatched ''"' when when decoding 'string')r] )rb rc r` r- r/ )r1 r� re 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@ x:t d�D ].}tjtt�|�d�� t�|� W d Q R X q W d S )Ni� )r] )�rangerb rc r` �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 d 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 value)r] )rb rc r` r- r/ )r1 r� re r r r �test_decode_invalid_dict s z+TestUltraJSONTests.test_decode_invalid_dict�numeric_int_as_strZ31337z-31337c C s t |�t�|�kst�d S )N)r_ r- r/ r0 )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 �|�ksDt�d}t �|�}|t�|�ksdt�|t�|�ksvt�|t �|�ks�t�dt �d�ks�t�d S )Nz 31337 1337� z" \u0000\r\n "z )r- r. r7 r8 r0 rE r/ )r1 � wrapped_inputr: Z alone_inputr r r �test_encode_null_character4 s z-TestUltraJSONTests.test_encode_null_characterc C s d}t �|�t�|�ks t�d S )Nz"31337 \u0000 31337")r- r/ r7 r8 r0 )r1 r� r r r �test_decode_null_characterD s z-TestUltraJSONTests.test_decode_null_characterc C sb ddddddg}t �|�}|t�|�ks,t�|t �|�ks>t�t�t� |�t j|dtj d�� d S )Nl ���� T)r r" ) r- r. r7 r8 r0 r/ rD rT r rS r! )r1 � long_inputr: r r r � test_encode_list_long_conversionH s z3TestUltraJSONTests.test_encode_list_long_conversionc C sH d}t �|�}|t�|�ks t�|t�|�ks2t�|t �|�ksDt�d S )Nl ���� )r- r. r7 r8 r0 rE r/ )r1 r� r: r r r �test_encode_long_conversionZ s z.TestUltraJSONTests.test_encode_long_conversion�bigNumzGH-35288c C sT t jd }t�|�}t|�|ks$t�tjtdd�� t� |�|ksFt�W d Q R X d S )Nr( zValue is too big)r] ) �sys�maxsizer- r. r r0 rb rc r` r8 )r1 r� �encodingr r r �#test_dumps_ints_larger_than_maxsizeb s z6TestUltraJSONTests.test_dumps_ints_larger_than_maxsize�int_expZ1337E40z1.337E40z1337E+9z 1.337e+40z1.337E-4c C s t �|�t�|�kst�d S )N)r- r/ r7 r8 r0 )r1 r� r r r �test_decode_numeric_int_expn s z.TestUltraJSONTests.test_decode_numeric_int_expc C s, d}t jt|d�� t�d � W d Q R X d S )NzExpected 'str' or 'bytes')r] )rb rc ra r- r8 )r1 re r r r � test_loads_non_str_bytes_raisest s z2TestUltraJSONTests.test_loads_non_str_bytes_raisesr� l �M�+ l l l �� c C s& d|� d�}t �|�d |ks"t�d S )Nz{"id": r� �id)r- r/ r0 )r1 r� �docr r r �&test_decode_number_with_32bit_sign_bity s z9TestUltraJSONTests.test_decode_number_with_32bit_sign_bitc C s8 x2t d�D ]&}d�� }|d d d }t�|� q W d S )NrV � åi r) )r� r. r- )r1 r� �base� escape_inputr r r �test_encode_big_escape� s z)TestUltraJSONTests.test_encode_big_escapec C sD x>t d�D ]2}d�� }d}||d d d | }t�|� q W d S )NrV r� � "i 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 � fdd�d�}|� }t �|�}t �|�}|� ks@t�d S )N�keyiiz c s e Zd Z� fdd�ZdS )z1TestUltraJSONTests.test_to_dict.<locals>.DictTestc s � S )Nr )r1 )r r r �toDict� s z8TestUltraJSONTests.test_to_dict.<locals>.DictTest.toDictN)r� r� r� r� r )r r r �DictTest� s r� )r- r. r/ r0 )r1 r� �or: rj r )r r � test_to_dict� s z TestUltraJSONTests.test_to_dictc sX G � fdd�d�� d}t jt|d�� t�� d�� W d Q R X dtj� d�td�ksXt�dd � }d tj� d�|d�kszt�d d � }t jtd d�� tj� d�|d� W d Q R X dd� }t�tj� d�|d��dks�t�dd� }t�t�t � ddd���t�tj� d�|d��k�st�� d�� d�g}t � t j |td��t�tj|td��k�sTt�d S )Nc s2 e Zd Zdd� Ze� fdd��Zed�dd�ZdS ) z<TestUltraJSONTests.test_default_handler.<locals>._TestObjectc S s || _ d S )N)r� )r1 r� r r r �__init__� s zETestUltraJSONTests.test_default_handler.<locals>._TestObject.__init__c s � d�S )N�recursive_attrr )r1 )� _TestObjectr r r� � s zKTestUltraJSONTests.test_default_handler.<locals>._TestObject.recursive_attr)�returnc S s t | j�S )N)r r� )r1 r r r �__str__� s zDTestUltraJSONTests.test_default_handler.<locals>._TestObject.__str__N)r� r� r� r� �propertyr� r r� r )r� r r r� � s r� z Maximum recursion level reached)r] 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 d��d S )NzI raise for anything)ra )r� r r r �my_handler_raises� s zBTestUltraJSONTests.test_default_handler.<locals>.my_handler_raiseszI raise for anythingc 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 )Ni� r) r* )r| )r� r r r �my_obj_handler� s z?TestUltraJSONTests.test_default_handler.<locals>.my_obj_handleri� r) r* �bar)�default) rb rc r� r- r. r r0 ra r/ r| r7 r8 rE )r1 re r� r� r� r� Zobj_listr )r� r �test_default_handler� s. z'TestUltraJSONTests.test_default_handler)Kr� r� r� rb �markZskipifr r5 � parametrizer= r@ rF rG rI rL �math�pirP rQ rU rW rY r Zxfailr rf rg rk rm rn rp rq rr rt rv rw ry r� r� r| r~ r� r� r� r r 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 r$ L s� 8 .( r$ c @ sb 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!� Zej�d"i g ged#i fd$d%ged&i fd'gd$ged(i fd$i d'ged)i fd$d'gd$ged*i fd'd+g d,ged#i fd'd+iged(d-difd'd+d,d$iiied.d-difd$d/d0�d,d1iged2d-difg �d3d4� �Zd5d6� Zd%S )7�TestNumpyJSONTests� bool_inputTFc C s$ t |�}t�t�|��|ks t�d S )N)�boolr- r/ r. r0 )r1 r� �br r r � test_bool� s z TestNumpyJSONTests.test_boolc C sH t jddddddddgtd�}t jt�t�|��td�}t�||� d S )NTF)r" )r rS r� r- r/ r. rD rT )r1 Z bool_arrayr: r r r �test_bool_array� s z"TestNumpyJSONTests.test_bool_arrayc C s4 t �|�j}|d�}|t�t�|���|ks0t�d S )Nr( )r r" �typer- r/ r. r0 )r1 � any_int_dtype�klassrH r r r �test_int� s zTestNumpyJSONTests.test_intc C sB t jdtd�}|�|�}t jt�t�|��|d�}t� ||� d S )N�d )r" ) r �aranger_ �astyperS r- r/ r. rD rT )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)r! �uint64z-Cannot test 64-bit integer on 32-bit platformr r! ) r rb �skipr r" r� Ziinfo�maxr- r/ r. r0 )r1 r� r� rH r r r � test_int_max� s z TestNumpyJSONTests.test_int_maxc C s4 t �|�j}|d�}|t�t�|���|ks0t�d S )Ng�JY�8p@)r r" r� r- r/ r. r0 )r1 � float_dtyper� rH 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' ) r r �floatr rS r- r/ r. rD �assert_almost_equal)r1 r r 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 )NrV r& )r' ) r r" r� Zfinfor rD r r- r/ r. )r1 r r� rH r r r �test_float_max s z!TestNumpyJSONTests.test_float_maxc C sR t �d�}|�d�}t�t �t�t�|���|� t�tjt�|�dd�|� d S )N�` )r) r) r) r) r* r) T)r ) r r �reshaperD rT rS r- r/ r. )r1 r r r r �test_array_basic s z#TestNumpyJSONTests.test_array_basic�shape)rV rV )rN rN ru )r r( c C sR t �d�}|�|�}t�t �t�t�|���|� t�tjt�|�dd�|� d S )Nr T)r ) r r r rD rT rS r- r/ r. )r1 r r 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" )r rS �objectr- r/ r. rD rT )r1 Zarr_listr �resultr r r �test_array_list& s z"TestNumpyJSONTests.test_array_listc C sn t j}t jddd|d�}|�d�}t jt�t�|��|d�}t� ||� tjt�|�d|d�}t� ||� d S )Ng�rh�� Y@gX9��vi@r( )r" )rN rN ru T)r r" ) r Zfloat32r r rS r- r/ r. rD r )r1 r" r Zarr_outr r r �test_array_float6 s z#TestNumpyJSONTests.test_array_floatc C s8 t �d�}tjt|d�� t�t�d�� W d Q R X d S )Nz:array(1) (0d array) is not JSON serializable at the moment)r] r( ) r� r� rb rc ra r- r. r rS )r1 re r r r � test_0d_arrayB s z TestNumpyJSONTests.test_0d_arrayz!bad_input,exc_type,err_msg,kwargsz:nesting not supported for object or variable length dtypesr� NzZint\(\) argument must be a string, a bytes-like object or a( real)? number, not 'NoneType'r? zLCannot decode multidimensional arrays with variable length elements to numpyzVint\(\) argument must be a string, a bytes-like object or a( real)? number, not 'dict'z-invalid literal for int\(\) with base 10: 'a'r� �c�labelledz(labels only supported up to 2 dimensions� )r? r� r� z1cannot reshape array of size 3 into shape \(2,1\)c C s: t j||d��" tjt�|�fddi|�� W d Q R X d S )N)r] r T)rb rc r- r/ rE )r1 Z bad_input�exc_typer� �kwargsr r r �test_array_numpy_exceptH s @z*TestNumpyJSONTests.test_array_numpy_exceptc C s� dg i}t jt �|�ddd�}t�d�|d k�� s8t�t�dg�|d k�� sTt�|d d ksdt�ddig}t jt �|�ddd�}t�dg�|d k�� s�t�t�dg�|d k�� s�t�|d d ks�t�d }t j|ddd�}tjdd d d d dgtd�� d�}||d k�� �st�|d d k�s(t�t�ddg�|d k�� �sHt�d}t j|ddd�}tjdd d d d dgtd�� d�}||d k�� �s�t�t�dddg�|d k�� �s�t�t�ddg�|d k�� �s�t�d S )Nr? T)r r )r( r r r( r) r� z<[{"a": 42, "b":31}, {"a": 24, "c": 99}, {"a": 2.4, "b": 78}]� � �c g333333@�N )r" )r* r) r� zK{"1": {"a": 42, "b":31}, "2": {"a": 24, "c": 99}, "3": {"a": 2.4, "b": 78}}�1�2�3) r- r8 rE r �empty�allr0 rS r_ r )r1 Zlabelled_inputr: Z input_dumpsZ expected_valsr r r �test_array_numpy_labelled� s, "z,TestNumpyJSONTests.test_array_numpy_labelled)r� r� r� rb r� r� r� r� r� r r r r r r r r r r r` ra r r( r r r r r� � sh r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z e j � dddddg�dd� �Z e j � dg dgg�dd� �Ze j � ddd g�d d � �Ze j � d d d!g�d"d#� �Zd$d%� Zd&d'� Zd(d)� Ze j � d*d+d,d-d.d/d0d1d2d3g �e j � d4d5d6g�d7d8� ��Zd9d:� Zd;d<� Zd=d>� Ze j � d?ed@dA�ed5dA�edBdCdCdD�edBdEdFdD�ed5dCdG�ed6dH�ed6d6dI�ed6d6d6dJ�edKdKdKdJ�g �dLdM� �ZdNS )O�TestPandasJSONTestsc C s> |dkr|rt �d� t||�}tdddgdddggd d gd d d g|d�}|d krVi nd|i}|d krji nd|i}|j|k�� s�t�tjtj |f|�f|�}|j|k�� s�t�|dkr�t |�}tf |�}nt|�}|dkr�dddg|_ ddg|_ n(|dk�rddg|_ n|dk�r|� � }|j|k�� �s.t�t�||� d S )Nr zNot idiomatic pandasr( r) r* ru rN � r? r� �x�y�z)r �columnsr" r r r r r r )rb r r# r Zdtypesr' r0 r- r/ r. r r. r Z transposerD �assert_frame_equal) r1 r r r" �dfr9 � decode_kwargsr: rj r r r �test_dataframe� s4 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|��|ks�t�d S )Nr( r) r* ru rN r* r? r� r+ r, r- )r r. )Zdf1Zdf2r )r �copyr- r/ r. r0 )r1 r r0 �nestedr �expr r r �test_dataframe_nested� s &z)TestPandasJSONTests.test_dataframe_nestedc C s� |dkrt �d� tdddgdddggd d gd d d gtd�}|d krHi nd|i}ttjtj|f|�ddd�� }|d kr~|j}n|dkr�ddg|_t � ||� d S )N)r r z Incompatible with labelled=Truer( r) r* ru rN r* r? r� r+ r, r- )r r. r" r T)r r r r ) rb r r r_ r- r/ r. �Tr rD r/ )r1 r r0 r r: r r r � test_dataframe_numpy_labelled� s z1TestPandasJSONTests.test_dataframe_numpy_labelledc C s" t ||�}tddddddgddd d d dd g|d ��� }|j|ksDt�|d krPi nd|i}|d krdi nd|i}tjtj|f|�f|�}|j|ks�t�|dkr�t|�}tf |�}nt|�}|dkr�d |_ |�� }ddddddg|_ n"|dk�rd |_ dddd d d g|_ |j|k�st�t � ||� d S ) NrV rR � �( �2 �< �seriesr* � r� rX r& )�namer r" r r r )Nr �6�7�8r\ �10Z15)r r r r( r) r* ru rN ) r# r � sort_valuesr" r0 r- r/ r. r r? r rD �assert_series_equal) r1 r r r" rz r9 r1 r: rj 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|��|ks�t�d S )NrV rR r9 r: r; r< r= r* r> r� rX r& )r? r )�s1�s2r )r rD r3 r- r/ r. r0 )r1 r rz r4 r r5 r r r �test_series_nested s (z&TestPandasJSONTests.test_series_nestedc C s� t ddddddgdd�}t t�t�|��dd�}t�||� t tjt�|�d d �dd�}t�||� tt�tj|d d ���}t f |�}t�||� |j|jks�t�ttjtj|d d �d d ��}t f |�}t�||� |j|jks�t�t t�tj|d d ��dd�}t�||� t tjtj|d d �d d �dd�}t�||� t t�tj|dd ��dd�}t�||� t tjtj|dd �d d �dd�}t�||� t t�tj|dd ��dd�}t�||� t tjtj|dd �d d �dd�}t�||� d S )Nr �- � �b �+ � r )r? T)r r )r r r ) r r- r/ r. rD �assert_index_equalr r? r0 )r1 �ir: rj r r r � test_index& s8 z TestPandasJSONTests.test_indexc C s� d}t ttddd��d d�}tj||d�}t t�t�|���}t� ||� t tj � t |��|d�}t t�tj||d���}|jj�tj�}t |�|_t�||� d S )Nr� z1/1/2000rR )Zperiods)�freq)r{ )r )r �listr r- r. r rS r/ rD rO r �randomZrandn�lenr r r r! rE )r1 r{ �rngr3 r4 �tsZ idx_valuesr r r �test_datetime_indexT s z'TestPandasJSONTests.test_datetime_index� invalid_arrz[31337,]z[,31337]z[]]z[,]c C s, d}t jt|d�� t�|� W d Q R X d S )Nz[Expected object or value|Trailing data|Unexpected character found when decoding array value)r] )rb rc r` r- r/ )r1 rY re r r r �test_decode_invalid_arraye s z-TestPandasJSONTests.test_decode_invalid_arrayr iiz c C s |t �t|��kst�d S )N)r- r/ r r0 )r1 r r r r �test_decode_arrayv s z%TestPandasJSONTests.test_decode_array� extreme_numl ���� l���� c C s |t �t|��kst�d S )N)r- r/ r r0 )r1 r\ r r r �test_decode_extreme_numbersz s z/TestPandasJSONTests.test_decode_extreme_numbers�too_extreme_numZ9223372036854775808z-90223372036854775809c C s( t jtdd�� t�|� W d Q R X d S )Nz#Value is too big|Value is too small)r] )rb rc r` r- r/ )r1 r^ r r r � test_decode_too_extreme_numbers~ s z3TestPandasJSONTests.test_decode_too_extreme_numbersc C s i t �d�kst�d S )Nz{} )r- r/ r0 )r1 r r r �%test_decode_with_trailing_whitespaces� s z9TestPandasJSONTests.test_decode_with_trailing_whitespacesc C s( t jtdd�� t�d� W d Q R X d S )Nz Trailing data)r] z{} a)rb rc r` r- r/ )r1 r r r �)test_decode_with_trailing_non_whitespaces� s z=TestPandasJSONTests.test_decode_with_trailing_non_whitespacesc C s( t jtdd�� t�d� W d Q R X d S )NzValue is too big)r] z[18446098363113800555])rb rc r` r- r8 )r1 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)rD r r- r8 r )r1 rd rc r r r �test_decode_floating_point� s z.TestPandasJSONTests.test_decode_floating_pointc C s2 t � }x tdd�D ]}|�|� qW t�|� d S )Nr i�� )�setr� �addr- r. )r1 rz r+ r r r �test_encode_big_set� s z'TestPandasJSONTests.test_encode_big_setc C s dt �t� �kst�d S )Nz[])r- r. rf r0 )r1 r r r �test_encode_empty_set� s z)TestPandasJSONTests.test_encode_empty_setc C sH ddddddddd h }t �|�}t �|�}x|D ]}||ks0t�q0W d S ) Nr( r) r* ru rN r* r> r� rX )r- r. r/ r0 )r1 rz ri rj r r r r �test_encode_set� s z#TestPandasJSONTests.test_encode_set�tdin )�days� rN )�hours�minutes�secondsrR r9 )rl � nanoseconds)rq )� microsecondsrq )Z millisecondsrr rq i� c C s. t j|dd�}d|�� � d�}||ks*t�d S )NT)Z iso_datesr� )r- r. r� r0 )r1 rk r r� r r r �test_encode_timedelta_iso� s z-TestPandasJSONTests.test_encode_timedelta_isoN)r� r� r� r2 r6 r8 rF rI rQ rX rb r� r� rZ r[ r] r_ r` ra rb re rh ri rj r rs r r r r r) � s\ &! . r) )(r r| r+ r7 rB r� r� r� r~ r� r r rb r� Zpandas._libs.jsonZ_libsr- Z pandas.compatr r r Zpandasr r r r r r r r Zpandas._testingZ_testingrD r Zfixturer r# r$ r� r) r r r r �<module> s< ( Y
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка