Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/io/json/__pycache__/test_pandas.cpython-37.pyc
Назад
B �5�g8� � @ sb d dl Z d dl mZ d dlmZ d dlmZ d dlZd dlZd dlZd dl Z d dl Z d dl m Z mZmZmZ d dlm mZ d dlZd dlmZmZmZmZmZm Z d dl m Z e � � Z!ee!�Z"e"�#� Z$dgd d gd d gd d ge%e$�d Z&ej'e&d d�e$_(e)e*e&��e$d < e j+e%e$�dd�e$d< dd� Z,e j-�.d�e j-�.d�G dd� d���Z/dS )� N)� timedelta)�Decimal)�StringIO)�IS64�PY38�PY310�is_platform_windows)� DataFrame� DatetimeIndex�Series� Timestamp�compat� read_jsonZbah� �bar�baz�foo� �E)�name�int64)�dtype�sortc C sD |dks|dkr |j dd�}|dkr4tt|j��|_t�| |� d S )N�records�valuesT)�drop)� reset_index�range�len�columns�tm�assert_frame_equal)�result�expected�orient� r% �V/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/io/json/test_pandas.py�assert_json_roundtrip_equal* s r' zBignore:an integer is required (got type float)*:DeprecationWarningz6ignore:the 'numpy' keyword is deprecated:FutureWarningc @ s� e Zd Zejdd�dd� �Zejdd� �Zejdd� �Zd d � Zej � d d d dg�dd� �Z ej � d ddg�dd� �Z ej � d d dg�ej � dddgddggddgd d ggd dgd d gge d �d ge d!�d ggg�d"d#� ��Zej � d ddd g�d$d%� �Zd&d'� Zej � d(d)eg�ej � d*dd)g�ej � d+dd)g�d,d-� ���Zej � d(d)ejg�ej � d*dd)g�ej � d+dd)g�d.d/� ���Zej � d(d0ejed1g�ej � d*dd)g�ej � d+dd)g�d2d3� ���Zej � d*dd)g�ej � d+dd)g�d4d5� ��Zej � d*dd)g�ej � d+dd)g�d6d7� ��Zej � d*dd)g�ej � d+dd)g�d8d9� ��Zej � d*dd)g�ej � d+dd)g�d:d;� ��Z ej � d<d=d>d?� d@dAg�d fdBdCg�dDdE� �Z ej � d(dd)g�ej � d*dd)g�ej � d+dd)g�dFdG� ���Z ej � d(dd)g�dHdI� �Z ej � dJej!ej"g�ej � d(dd)g�dKdL� ��Z#ej j$e% dMdN�ej � dOdPdQdRdSdTdUg�dVdW� ��Z&dXdY� Z'dZd[� Z(d\d]� Z)d^d_� Z*d`da� Z+dbdc� Z,ddde� Z-dfdg� Z.dhdi� Z/djdk� Z0ej � d+dd)g�dldm� �Z1ej � d(d)d0g�ej � d+dd)g�dndo� ��Z2ej � d+dd)g�dpdq� �Z3ej � d+dd)g�drds� �Z4ej � d(ejeg�ej � d+dd)g�dtdu� ��Z5dvdw� Z6dxdy� Z7dzd{� Z8ej � d|de9d}gd~d�fd)e9d�g�fg�d�d�� �Z:d�d�� Z;d�d�� Z<d�d�� Z=d�d�� Z>d�d�� Z?d�d�� Z@ej � d�d�d�g�ej � d�dd)g�ej � d�eAjBeAjAe g�d�d�� ���ZCej � d�d�d�d�d�d�d�d�g�d�d�� �ZDej � d�d�d�d�d�d�g�d�d�� �ZEd�d�� ZFej � d�d�d�d�d�d�g�d�d�� �ZGd�d�� ZHej � d�d�d�d�d�g�d�d�� �ZId�d�� ZJd�d�� ZKd�d�� ZLeMjNej jOd�d�� ��ZPeMjNej jOej � d�d�eQjRd�d��gd�d~gd�eQjRd�d��gg�d�dĄ ���ZSd�dƄ ZTd�dȄ ZUej � d�dd)g�ej � d�d�d�g�ej � d�eQjVeWg�d�d˄ ���ZXd�d̈́ ZYd�dτ ZZd�dф Z[d�dӄ Z\d�dՄ Z]d�dׄ Z^d�dل Z_ej j$e`d�dN�ej � d�e d܃e d�d�d��e d߃g�d�d� ��Zaej j$e`d�dN�ej � d�eQjbd�d�d�eQjbd�d�d�d�eQjbd�d�d�g�d�d� ��Zcd�d� Zdeejfd�d� �Zgd�d� Zhd�d� Ziej � d�ejjkd ejjkd� g�d�d�� �Zlej � d�ejjkd ejjkd� g�ej j$emj% d�dN�d�d�� ��Znd�d�� Zod�d�� Zpd�d�� Zqd��d � Zrej � dd0d d�g�d�dgddg�d�dg�d�dgg�ej � dddg�d�dg�d�dgg��d�d� ��Zs�d �d � Ztej � d(dee�d �g��d �d � �Zu�d�d� Zvej � �dewd d�g�d�dggddg�d�ddgd d�g�d�dgg�d�fewd d�g�d�dggddg�d��x�d�ddgd d�g�d�dgg�d�fewd d�g�d�dggddgddgddgg�d�ddgd d�g�d�dgg�d�fe9d d�d g�d�d��dd d�d g�d�fe9d d�d g�d�d��x�d��dd d�d g�d�fe9d d�g�dddgddgg�d��dd d�g�d�fg��d�d � �Zyej � dewd d�g�d�dggddg�d�ewd d�g�d�dggddg�d��x�d�ewd d�g�d�dggddgddgddgg�d�e9d d�d g�d�d�e9d d�d g�d�d��x�d�e9d d�g�dddgddgg�d�g��d �d � �Zzej � d d dddg��d �d � �Z{ej � d d �d!g�ej � ddd)g��d"�d#� ��Z|�d$�d%� Z}ej � �d&�d'�d(g��d)�d*� �Z~ej � �d+�d,�d-ejd �d.ej j��d/�ej�d!�d.ej j��d/�g��d0�d1� �Z�ej � �d2d d�dg��d3�d4� �Z�ej � �d+�d5�d6�d7�d8�d9�d:g��d;�d<� �Z��d=�d>� Z��d?�d@� Z��dA�dB� Z��dC�dD� Z�ej � �dEewd d�d gdddg�dF���dGfg��dH�dI� �Z�ej j�e�� � o�e��dJd)�dK��dL�dM� �Z��dN�dO� Z��dP�dQ� Z��dR�dS� Z�d0S (T �TestPandasContainerT)Zautousec c s t �� | _d V d S )N)� _cat_frame�copy� categorical)�selfr% r% r& �setup7 s zTestPandasContainer.setupc C s t �� }d|_|j�d �|_|S )N�ts)r ZmakeTimeSeriesr �index� _with_freq)r, �serr% r% r& �datetime_series= s z#TestPandasContainer.datetime_seriesc C s t t�� �}|j�d �|_|S )N)r r ZgetTimeSeriesDatar/ r0 )r, �dfr% r% r& �datetime_frameF s z"TestPandasContainer.datetime_framec C sN t ddgddggddgddgd �}t|j|d �|d �}|�� }t|||� d S ) N�a�b�c�dz index " 1z index / 2za \ bzy / z)r/ r )r$ )r r �to_jsonr* r' )r, r$ r3 r" r# r% r% r& � test_frame_double_encoded_labelsN s z4TestPandasContainer.test_frame_double_encoded_labelsr$ �splitr r c C sN t ddgddggddgddgd�}t|j|d �|d �}|�� }t|||� d S ) Nr5 r6 r7 r8 � �x�y)r/ r )r$ )r r r9 r* r' )r, r$ r3 r" r# r% r% r& �test_frame_non_unique_indexZ s "z/TestPandasContainer.test_frame_non_unique_indexr/ r c C sX t ddgddggddgddgd�}d |� d �}tjt|d �� |j|d � W d Q R X d S ) Nr5 r6 r7 r8 r< r= r> )r/ r z+DataFrame index must be unique for orient='�')�match)r$ )r �pytest�raises� ValueErrorr9 )r, r$ r3 �msgr% r% r& �"test_frame_non_unique_index_raisesb s " z6TestPandasContainer.test_frame_non_unique_index_raises�datar5 r6 r7 r8 g �?g @g @g @r< � �20130101Z20130102c C s� t |ddgddgd�}t|j|d�|dgd�}|dkr�t |�}|jd d �df jd kr�|jd d �df �tj�d |jd d �df< n |d kr�|}t� ||� d S ) Nr< � r= )r/ r )r$ )r$ � convert_datesr r zdatetime64[ns]i@B r; ) r r r9 �ilocr �view�npr r r! )r, r$ rG r3 r" r# r% r% r& � test_frame_non_unique_columnsi s .z1TestPandasContainer.test_frame_non_unique_columnsc C sX t ddgddggddgddgd�}d |� d �}tjt|d �� |j|d � W d Q R X d S ) Nr5 r6 r7 r8 r< rJ r= )r/ r z-DataFrame columns must be unique for orient='r@ )rA )r$ )r rB rC rD r9 )r, r$ r3 rE r% r% r& �$test_frame_non_unique_columns_raises� s " z8TestPandasContainer.test_frame_non_unique_columns_raisesc C s |� � |j dd�kst�d S )Nr )r$ )r9 �AssertionError)r, � float_framer% r% r& �test_frame_default_orient� s z-TestPandasContainer.test_frame_default_orientr F� convert_axes�numpyc C s2 |j |d�}t|||||d�}|}t|||� d S )N)r$ )r$ rT rU r )r9 r r' ) r, r$ rT rU r rR rG r" r# r% r% r& �test_roundtrip_simple� s z)TestPandasContainer.test_roundtrip_simplec C sX |j |d�}t|||||d�}|}|rHtr0t� rH|sH|dkrH|�tj�}t|||� d S )N)r$ )r$ rT rU r r; )r9 r r r �astyperN Zint32r' ) r, r$ rT rU r � int_framerG r" r# r% r% r& �test_roundtrip_intframe� s z+TestPandasContainer.test_roundtrip_intframeN�U3c C s t t�d�dd� td�D �dd� td�D �|d�}|r\|dkr\|d kr\|j�tjjd d �� |j |d �}t |||||d �}|� � } |s�| � tj �} |r�|dkr�| j� tj �| _| j� tj �| _n:|dkr�|r�| j� tj �| _n |r�|d kr�| j� tj �| _t|| |� d S )N)�� � c S s g | ] }t |��qS r% )�str)�.0�ir% r% r& � <listcomp>� s z?TestPandasContainer.test_roundtrip_str_axes.<locals>.<listcomp>r\ c S s g | ] }t |��qS r% )r] )r^ r_ r% r% r& r` � s r[ )r r/ r rZ r; z Can't decode directly to array)�reason)r$ )r$ rT rU r )r/ r r )r rN Zzerosr �node� add_markerrB �mark�xfailr9 r r* rW r r r/ r' ) r, �requestr$ rT rU r r3 rG r" r# r% r% r& �test_roundtrip_str_axes� s, z+TestPandasContainer.test_roundtrip_str_axesc C s� |dkr&|j �tjjd|� d�d�� | jj|d�}|r^|dkr^|j �tjjd|� d�d�� t||||d �}| j�� }|j � t �|_ d |j _ |s�|d kr�|� � }t|||� d S ) N)r/ r z.Can't have duplicate index values for orient 'z'))ra )r$ )r r zOrient z is broken with numpy=True)r$ rT rU r/ )rb rc rB rd re r+ r9 r r* r/ rW r] r � sort_indexr' )r, rf r$ rT rU rG r" r# r% r% r& �test_roundtrip_categorical� s z.TestPandasContainer.test_roundtrip_categoricalc C sx |j |d�}t||||d�}|�� }|rD|j�t�|_|j�t�|_|rh|dkrh|jdgdd�jdd�}t � ||� d S ) N)r$ )r$ rT rU r r r< )�axisT)r ) r9 r r* r/ rW �floatr �reindexr r r! )r, r$ rT rU Z empty_framerG r" r# r% r% r& �test_roundtrip_empty� s z(TestPandasContainer.test_roundtrip_emptyc C sb |j |d�}t||||d�}|�� }|sR|j�tj�d }|dkrL|�t�}||_t |||� d S )N)r$ )r$ rT rU i@B r; ) r9 r r* r/ rM rN r rW r] r' ) r, r$ rT rU r4 rG r" r# �idxr% r% r& �test_roundtrip_timestamp s z,TestPandasContainer.test_roundtrip_timestampc C s� |r"|dkr"|j �tjjdd�� t�dddddg�}d d d d d gd d d d d gdddddgdddddgd�}t||d�}|j|d�}t ||||d�} |� � } | j f | � d�� tj��} |s�|dkr�| �� } t| | |� d S )Nr; z Can't decode directly to array)ra r5 r6 r7 r8 �eg g �?g @g @g @Zfoo1Zfoo2Zfoo3Zfoo4Zfoo5TF)�A�B�C�D)rG r/ )r$ )r$ rT rU �numberr/ )rb rc rB rd re �pd�Indexr r9 r r* ZassignZ select_dtypesrW rN r rh r' ) r, rf r$ rT rU r/ r r3 rG r" r# r% r% r& �test_roundtrip_mixed s z(TestPandasContainer.test_roundtrip_mixedzdata,msg,orient)z {"key":b:a:d}zExpected object or valuer zO{"columns":["A","B"],"index":["2","3"],"data":[[1.0,"1"],[2.0,"2"],[null,"3"]]}�|z:Shape of passed values is \(3, 2\), indices imply \(2, 2\)zHPassed arrays should have the same length as the rows Index: 3 vs 2 rows)zW{"columns":["A","B","C"],"index":["1","2","3"],"data":[[1.0,"1"],[2.0,"2"],[null,"3"]]}z+3 columns passed, passed data had 2 columnsr; )zN{"badkey":["A","B"],"index":["2","3"],"data":[[1.0,"1"],[2.0,"2"],[null,"3"]]}zunexpected key\(s\): badkeyr; c C s. t jt|d�� tt|�|d� W d Q R X d S )N)rA )r$ )rB rC rD r r )r, rG rE r$ r% r% r& �$test_frame_from_json_bad_data_raises6 s %z8TestPandasContainer.test_frame_from_json_bad_data_raisesc C s� t ddgdddgg�}t|j|d�|||d�}t�|jd �sBt�t d d gd d d gg�}t|j|d�|||d�}t�|jd �s�t�d S )Nr< rJ r\ r � )r$ )r$ rT r )r rJ �1�2�4�5�6)r r r9 rN �isnanrL rQ )r, r$ rT rU r Znum_dfr" Zobj_dfr% r% r& �!test_frame_from_json_missing_data^ s z5TestPandasContainer.test_frame_from_json_missing_datac C s( t d|d�}ttjg�}t�||� d S )Nz[null])r )r r rN �nanr r! )r, r$ r r" r# r% r% r& �(test_frame_read_json_dtype_missing_valuet s z<TestPandasContainer.test_frame_read_json_dtype_missing_value�infc C sH t ddgdddgg�}||jd< t|�� |d�}t�|jd �sDt�d S )Nr< rJ r\ r r{ )r rJ )r )r �locr r9 rN r� rL rQ )r, r$ r� r r3 r" r% r% r& �test_frame_infinity} s z'TestPandasContainer.test_frame_infinityz$not compliant on 32-bit, xref #15865)ra z value,precision,expected_val)gffffff�?r< g �?)g333333�?r< g @)g333333��r< g �)gףp= ��?rJ g �?)g�C�l��?rH g �?)g�������?r g �?c C s2 t d|ig�}|j|d�}|d|� d�ks.t�d S )NZa_float)Zdouble_precisionz{"a_float":{"0":z}})r r9 rQ )r, �value� precisionZ expected_valr3 �encodedr% r% r& �"test_frame_to_json_float_precision� s z6TestPandasContainer.test_frame_to_json_float_precisionc C s< t dddg�}d}tjt|d�� |jdd� W d Q R X d S )Nr< rJ rH z+Invalid value 'garbage' for option 'orient')rA �garbage)r$ )r rB rC rD r9 )r, r3 rE r% r% r& �test_frame_to_json_except� s z-TestPandasContainer.test_frame_to_json_exceptc C sf t ddgd�}|jrt�tjt|�� t|j�d�|dd� t dg ig d�jd d �}d }||ksbt�d S ) N�jim�joe)r )r F)�check_index_type�test)r/ r )r$ z {"test":{}}) r �_is_mixed_typerQ r r! r r9 �dict�dtypes)r, r3 r" r# r% r% r& �test_frame_empty� s z$TestPandasContainer.test_frame_emptyc C sP t ddgd�}|d �d�|d< |js*t�tjt|�� t|j �d�|dd� d S )Nr� r� )r �i8)r F)r� ) r rW r� rQ r r! r r9 r� r� )r, r3 r% r% r& �test_frame_empty_mixedtype� s z.TestPandasContainer.test_frame_empty_mixedtypec C s dddddgdddd d gd d d ddgdddddgg}t |td�dddddgd�}|jsZt�|�� }x2d D ]*}|j|d �}t||d d �}t�||� qhW t � t |��|_ |jd d �}t|d d d �}t�||� t � |j d �|_|jd!d �}t|d!d d �}t�||� d S )"N� r< r g�������?g{�G�z�?� rJ r g�������?g{�G�z�?� rH r g333333�?g� ��Q��?�( r\ �quxg�������?g{�G�z�?�abcdZ1stZ2ndZ3rdZ4thZ5th)r/ r )r; r/ r )r$ F)r$ rT r r )r �listr� rQ r* r9 r r r! rN �aranger r/ �shaper )r, �valsr3 �rightr$ Zinp�leftr% r% r& �test_frame_mixedtype_orient� s( z/TestPandasContainer.test_frame_mixedtype_orientc C s t �dd�}tt�|�d d�}tddddgdd d d gd d ddgddddgddddggddddg|d �}td �|d < td �|jd |j� d �f< |d |d!< t j |jd"|j� d!�f< |d#d$d%�}t j � |d&�}t|�}t�||� |jd!gd"d'�}t j � |d(�}t|�} t�|| � d S ))Nz 2000-01-03z 2000-01-07)�freqg� É��?gz�~Kc�?g�X�7���?g�H/�ƿg�|�Rпg����X���gs�B�f��g����(�ƿg��<�1=�?g����9�?g���g?�?gL��ł ��g���J�Q��g�[r?T��?gj<�DH>�?g-���ѿgI��E�x�?gW!i`y��gP0�4���?g�+�]�5�?rq rr rs rt )r r/ z19920106 18:21:32.12�daterI rH �modifiedr< �io�jsonrG ztsframe_v012.json)rj ztsframe_iso_v012.json)rv � date_ranger rN Zasarrayr r rL r �get_loc�NaT�os�path�joinr r r! r ) r, Zdatapath�dtir3 �dirpathZv12_jsonZdf_unserZdf_isoZ v12_iso_jsonZ df_unser_isor% r% r& �test_v12_compat� s, z#TestPandasContainer.test_v12_compatc C s� t jdddd�}tt|�d d�}tdddd d d d d ddg ddddddddddg ddd d d d d d!d"d#g d$d%d&d'd(d)d*d+d,d-g d.d/d0d1d2d3d4d5d6d7g d8d9d:d;d<d=d>d?d@dAg dB�|dC�}|j�dD�|_t|jdEdF�dEdF�}t j ||dGdGdGdGdH� d S )INZ20000101r� �H)�periodsr� )r� g}S? w�g��b�m��?gz(~J� �?g��1���?gk*��M�gv�T�j��?g7j " Y�?gp�+KS(�g���?gZt�S�y�i�L,i_�}i4z�iKe�iL&�i�%gi��DiY)L i;ei�D� Z78c608f1Z64a99743Z13d2ff52Zca7f4af2Z97236474Zbde7e214Z1a6bde47Zb1190be5Z7a669144Z8d64d068gD!����gIxJ ��g��%� @g��P�\<�g�����gPl����?gF�z�n�?g5mH/ �?gp�H�%��g� I�?Z14f04af9Zd085da90Z4bcfac83Z81504cafZ2ffef4a9Z08e2f5c4Z07e1af03Zaddbd4a7Z1f6a09baZ4bfc4d87i�/i���i�Yi �6i�I� i;�>i�� i�(� i0q{iѪ�)Zfloat_1Zint_1Zstr_1Zfloat_2Zstr_2Zint_2)r/ �unicoder; )r$ T)r� Zcheck_column_typeZ by_blocksZ check_exact) rv r� r r� r r rW r r9 r r! )r, r/ �df_mixedZ df_roundtripr% r% r& �test_blocks_compat_GH9037� s� z-TestPandasContainer.test_blocks_compat_GH9037c C s� G dd� d�}d}||�}t d|jgi�}|�� d|� d�ksBt�t d|gi�}d}tjt|d�� |�� W d Q R X t |gd gd �dd gd �}tjt|d�� |�� W d Q R X |jtd �}d|� d�} || ks�t�|jtd �d|� d�ks�t�d S )Nc @ s" e Zd Zdd� Zed�dd�ZdS )zFTestPandasContainer.test_frame_nonprintable_bytes.<locals>.BinaryThingc S s || _ t�|�| _d S )N)�hexed�bytes�fromhex�binary)r, r� r% r% r& �__init__R s zOTestPandasContainer.test_frame_nonprintable_bytes.<locals>.BinaryThing.__init__)�returnc S s | j S )N)r� )r, r% r% r& �__str__V s zNTestPandasContainer.test_frame_nonprintable_bytes.<locals>.BinaryThing.__str__N)�__name__� __module__� __qualname__r� r] r� r% r% r% r& � BinaryThingQ s r� Z574b4454ba8c5eb4f98a8f45rq z {"A":{"0":"z"}}z6Unsupported UTF-8 sequence length when encoding string)rA r< )rq rr rr )r )�default_handlerz"},"B":{"0":1}})r r� r9 rQ rB rC � OverflowErrorr] ) r, r� r� ZbinthingZ df_printableZdf_nonprintablerE r� r" r# r% r% r& � test_frame_nonprintable_bytesN s" z1TestPandasContainer.test_frame_nonprintable_bytesc C s< t dd dgddgi��� }ddd � d�}||ks8t�d S )Nr i�� r< r i9 z{"z":{"0":1},"foo":{"0":1337}})r r9 rQ )r, r" r# r% r% r& �test_label_overflowt s z'TestPandasContainer.test_label_overflowc C s� t ddgddgd�}d}tjt|d�� |jdd� W d Q R X t�|t|jd d�d d d �� t|jd d�d d d �}t�|j |j � d S ) Nr5 r6 r< )r/ z.Series index must be unique for orient='index')rA r/ )r$ r; �series)r$ �typr ) r rB rC rD r9 r �assert_series_equalr Zassert_numpy_array_equalr )r, �srE Z unserializedr% r% r& � test_series_non_unique_indexz s z0TestPandasContainer.test_series_non_unique_indexc C s |� � |j dd�kst�d S )Nr/ )r$ )r9 rQ )r, � string_seriesr% r% r& �test_series_default_orient� s z.TestPandasContainer.test_series_default_orientc C sR |j |d�}t|d||d�}|}|dkr4|jdd�}|dkrBd |_t�||� d S )N)r$ r� )r� r$ rU )r r T)r r; )r9 r r r r r� )r, r$ rU r� rG r" r# r% r% r& � test_series_roundtrip_simple� s z0TestPandasContainer.test_series_roundtrip_simplec C sT |j |d�}t|d|||d�}|}|dkr6|jdd�}|dkrDd |_t�||� d S )N)r$ r� )r� r$ rU r )r r T)r r; )r9 r r r r r� )r, r$ rU r Z object_seriesrG r" r# r% r% r& � test_series_roundtrip_object� s z0TestPandasContainer.test_series_roundtrip_objectc C sT |j |d�}t|d||d�}|}|dkr6|jdd�}n|j�t�|_t�||� d S )N)r$ r� )r� r$ rU )r r T)r )r9 r r r/ rW rk r r� )r, r$ rU Z empty_seriesrG r" r# r% r% r& �test_series_roundtrip_empty� s z/TestPandasContainer.test_series_roundtrip_emptyc C sR |j |d�}t|d||d�}|}|dkr4|jdd�}|dkrBd |_t�||� d S )N)r$ r� )r� r$ rU )r r T)r r; )r9 r r r r r� )r, r$ rU r2 rG r" r# r% r% r& � test_series_roundtrip_timeseries� s z4TestPandasContainer.test_series_roundtrip_timeseriesc C sd t td�ddddddgd�}|j|d �}t|d ||d �}|�� }|d krT|jd d�}t�||� d S )Nr{ r5 r6 r7 r8 rp �f)r/ )r$ r� )r� r$ rU )r r T)r )r r r9 r r* r r r� )r, r$ rU r r� rG r" r# r% r% r& � test_series_roundtrip_numeric� s z1TestPandasContainer.test_series_roundtrip_numericc C s< t dddg�}d}tjt|d�� |jdd� W d Q R X d S )Nr< rJ rH z+Invalid value 'garbage' for option 'orient')rA r� )r$ )r rB rC rD r9 )r, r� rE r% r% r& �test_series_to_json_except� s z.TestPandasContainer.test_series_to_json_exceptc C s4 t dddg�}t|�� ddd�}tj||dd� d S )Ng= ףp=@r� T)r� � precise_floatF)r� )r r r9 r r� )r, r� r" r% r% r& �#test_series_from_json_precise_float� s z7TestPandasContainer.test_series_from_json_precise_floatc C s@ t dddg�}t|�� dtjd�}t dgd �}t�||� d S )Ng= ףp=@r� )r� r r\ rH )r r r9 rN r r r� )r, r� r" r# r% r% r& �test_series_with_dtype� s z*TestPandasContainer.test_series_with_dtypezdtype,expectedz 2000-01-01zdatetime64[ns])r l ,�Uqc C s4 t dgdd�}|�� }t|d|d�}t�||� d S )Nz 2000-01-01zdatetime64[ns])r r� )r� r )r r9 r r r� )r, r r# r� rG r" r% r% r& � test_series_with_dtype_datetime� s z3TestPandasContainer.test_series_with_dtype_datetimec C s8 t dddgdddgg�}t|�� dd�}t�||� d S )Ng= ףp=@T)r� )r r r9 r r! )r, r3 r" r% r% r& �"test_frame_from_json_precise_float� s z6TestPandasContainer.test_frame_from_json_precise_floatc C s> t td�ddddddgdd �}t|�� d d �}t�||� d S ) Nr{ r5 r6 r7 r8 rp r� r )r/ r )r� )r r r r9 r r� )r, r� r" r% r% r& �test_typ� s z TestPandasContainer.test_typc C sp t dddgdddgg�}t|�� �}t�||� t dddgdddgd�dd d gd �}t|�� �}t�||� d S ) Nr< rJ rH r\ r r{ )r5 r6 rq rr rs )r/ )r r r9 r r! )r, r3 r" r% r% r& �test_reconstruction_index� s $ z-TestPandasContainer.test_reconstruction_indexc C s@ t �d��,}x$|||gD ]}|�|� t|� qW W d Q R X d S )Nz test.json)r � ensure_cleanr9 r )r, rR rX r4 r� r3 r% r% r& � test_path s z TestPandasContainer.test_pathc C sR |� � }t|�}t�||� |� � }t|dd�}tj||dd� |jd ksNt�d S )Nr� )r� F)Z check_names)r9 r r r! r� r rQ )r, r2 r4 r� r" r% r% r& �test_axis_dates s z#TestPandasContainer.test_axis_datesc C s� |}t d�|d< |�� }t|�}t�||� d|d< |jdd�}t|dd�}|�� }|d j�d �|d< |d �d �|d< t�||� t t d�|j d �}|�� }t|d d �}t� ||� d S )NrI r� g �?r �ns)� date_unitF)rK r� r )r/ r� )r� ) r r9 r r r! r* r rM rW r r/ r� )r, r2 r4 r3 r� r" r# r. r% r% r& �test_convert_dates s z&TestPandasContainer.test_convert_dates� date_format�epoch�iso� as_object�date_typc C sl |dddd�t jg}|r"|�d� t||d�}|j|d�}|dkrHd}nd }|r\|�d d �}||ksht�d S ) Ni� r< )�year�month�dayr5 )r/ )r� r� z+{"1577836800000":1577836800000,"null":null}zC{"2020-01-01T00:00:00.000Z":"2020-01-01T00:00:00.000Z","null":null}�}z ,"a":"a"})rv r� �appendr r9 �replacerQ )r, r� r� r� rG r1 r" r# r% r% r& �test_date_index_and_values/ s z.TestPandasContainer.test_date_index_and_values� infer_wordZ trade_timer� �datetimeZsold_atr� � timestampZ timestampsc C sf ddl m} dd|diddig}tdtd�gdtjggd|gd�}t||��d|g }t�||� d S ) Nr )�dumps�idr< l $�A�rJ z 2002-11-08)r ) �pandas.io.jsonr� r r rv r� r r r! )r, r� r� rG r# r" r% r% r& �test_convert_dates_inferF s z,TestPandasContainer.test_convert_dates_inferzdate,date_unit)z20130101 20:43:42.123N)z20130101 20:43:42r� )z20130101 20:43:42.123�ms)z20130101 20:43:42.123456�us)z20130101 20:43:42.123456789r� c C s� |}t |�|d< tj|jd|j�d�f< tj|jd|j�d�f< |rT|jd|d�}n |jdd�}t|�}|�� }|j � d�|_ |d j � d�|d< t � ||� d S )Nr� r< r r� )r� r� )r� �UTC)r rv r� rL r r� r9 r r* r/ � tz_localize�dtr r! )r, r� r� r4 r3 r� r" r# r% r% r& �test_date_format_frame] s z*TestPandasContainer.test_date_format_framec C s4 |}d}t jt|d�� |jddd� W d Q R X d S )Nz*Invalid value 'foo' for option 'date_unit')rA r� r )r� r� )rB rC rD r9 )r, r4 r3 rE r% r% r& � test_date_format_frame_raisesw s z1TestPandasContainer.test_date_format_frame_raisesc C s� t t|�|jd�}tj|jd< tj|jd< |r>|jd|d�}n |jdd�}t|dd�}|�� }|j� d �|_|j � d �}t � ||� d S ) N)r/ r< r r� )r� r� )r� r� )r� r� ) r r r/ rv r� rL r9 r r* r� r� r r� )r, r� r� r2 r. r� r" r# r% r% r& �test_date_format_series} s z+TestPandasContainer.test_date_format_seriesc C sB t td�|jd�}d}tjt|d�� |jddd� W d Q R X d S )Nz20130101 20:43:42.123)r/ z*Invalid value 'foo' for option 'date_unit')rA r� r )r� r� )r r r/ rB rC rD r9 )r, r2 r. rE r% r% r& � test_date_format_series_raises� s z2TestPandasContainer.test_date_format_series_raises�unitr� r� r� r� c C s� |}t d�|d< |j�d�}t d�|jd|f< t d�|jd|f< tj|jd|f< |jd|d �}t||d �}t� ||� t|d d �}t� ||� d S ) Nz20130101 20:43:42r� z19710101 20:43:42r< z21460101 20:43:42rJ r\ r� )r� r� )r� ) r r r� rL rv r� r9 r r r! )r, r� r4 r3 �dlr� r" r% r% r& �test_date_unit� s z"TestPandasContainer.test_date_unitc C s d}t |� d S )Nad { "status": "success", "data": { "posts": [ { "id": 1, "title": "A blog post", "body": "Some useful content" }, { "id": 2, "title": "Another blog post", "body": "More content" } ] } })r )r, r� r% r% r& �test_weird_nested_json� s z*TestPandasContainer.test_weird_nested_jsonc C sx t tj�dd�td�d�}td�|d< td�|d< d|d < tjddd �|_ |� � }t |tj tj d �d �}t�||� d S ) Nr rJ ZAB)r rI r� �intsT�bools)r� )r r )r )r rN �random�randnr� r r rv r� r/ r9 r r Zbool_r r! )r, Zdfj2r� r" r% r% r& �test_doc_example� s z$TestPandasContainer.test_doc_examplec C s� t ddd�}tddgddggddgd�}d }tjt|d �� tj||d d � W d Q R X t d�}tddgddggddgd�}t�||� d S ) Nz#[{"a": 1, "b": 2}, {"b":2, "a" :1}]T)rU r< rJ r5 r6 )r z�DataFrame\.index are different DataFrame\.index values are different \(100\.0 %\) \[left\]: Index\(\['a', 'b'\], dtype='object'\) \[right\]: RangeIndex\(start=0, stop=2, step=1\))rA F)r� )r r rB rC rQ r r! )r, r" r# � error_msgr% r% r& �test_misc_example� s z%TestPandasContainer.test_misc_examplec C s< d}t �|�}|�� }t|�}t�|j|j|jd�|� d S )Nz<https://raw.github.com/hayd/lahman2012/master/csvs/Teams.csv)r/ r ) rv Zread_csvr9 r r r! rl r/ r )r, �csvr3 r� r" r% r% r& �test_round_trip_exception_� s z.TestPandasContainer.test_round_trip_exception_z field,dtypeZ created_atr� )�tzZ closed_atZ updated_atc C s& d}t |dd�}|| j|ks"t�d S )Nz@https://api.github.com/repos/pandas-dev/pandas/issues?per_page=5T)rK )r r rQ )r, �fieldr �urlr" r% r% r& �test_url� s z TestPandasContainer.test_urlc C sV dd� }t td�tdd�g�}|jdks,t�t|�� dd��|�}t�||� t td�tdd�gt � d d g�d �}|jdks�t�t|�� dd��|�}t�||� t td�tdd�g�}|d jdks�t�t� |t|�� ��|�� t tdd �tdd�gd d gt j dd d�d��}t|jdd��}t j|jdd�|d< t �|j�|d< t� ||� d S )Nc S s t j| dd�S )Nr� )r� )rv � to_timedelta)r= r% r% r& �<lambda>� � z4TestPandasContainer.test_timedelta.<locals>.<lambda>� r )�secondsztimedelta64[ns]r� )r� r r< )r/ )�daysrJ rI )�startr� )r5 r6 r7 r� )r� )r� r5 r7 )r r r rQ r r9 �applyr r� rv rw r r! r� r r5 � to_datetimer7 )r, � converterr� r" �framer% r% r& �test_timedelta� s( $ z"TestPandasContainer.test_timedeltac C sr t dtd�td�gitd�}t dt�|jd �jt|jd �jgi�}t|j dd�dd id�}t j ||d d � d S ) Nr5 r rI )r r r< r� )r� r F)r� ) r r r �objectrv � Timedeltar5 r� r r9 r r! )r, r r# r" r% r% r& � test_mixed_timedelta_datetime s (z1TestPandasContainer.test_mixed_timedelta_datetime� timedelta_typc C sp |dd�|dd�t jg}|r&|�d� t||d�}|dkr@d}nd}|rT|�d d �}|j|d �}||kslt�d S ) Nr< )r rJ r5 )r/ r� zA{"P1DT0H0M0S":"P1DT0H0M0S","P2DT0H0M0S":"P2DT0H0M0S","null":null}z7{"86400000":86400000,"172800000":172800000,"null":null}r� z ,"a":"a"})r� )rv r� r� r r� r9 rQ )r, r� r� r rG r1 r# r" r% r% r& �test_timedelta_to_json" s z*TestPandasContainer.test_timedelta_to_jsonc C sN t � }tdd|gi�}tddt|�gi�}t|jtd��}tj||dd� d S )Nr5 � )r� F)r� )r r r] r r9 r r! )r, r� r r# r" r% r% r& �test_default_handler9 s z(TestPandasContainer.test_default_handlerc C sb ddl m} dd� }dtddtdd �gtd �d d gd �d dgd�g}d}|||dd�|ks^t�d S )Nr )r� c S s( t | t�r dd| jfd| jfgS t| �S )N)Zmathjs�Complex�reZim)� isinstance�complex�real�imagr] )�objr% r% r& �defaultC s zBTestPandasContainer.test_default_handler_indirect.<locals>.default� r< ZSTRr\ ���r� zN/A)r5 r6 r5 r6 )r zR[9,[[1,null],["STR",null],[[["mathjs","Complex"],["re",4.0],["im",-5.0]],"N\/A"]]]r )r� r$ )r� r� r r% rk rQ )r, r� r) Zdf_listr# r% r% r& � test_default_handler_indirect@ s z1TestPandasContainer.test_default_handler_indirectc C sP t ddtdd�gtd�d tdd�gd�d d gd �}d }|jtd d�|ksLt�d S )Nr< gffffff@r\ r+ r� g333333�?r )r5 r6 r5 r6 )r zE[["(1+0j)","(nan+0j)"],["(2.3+0j)","(nan+0j)"],["(4-5j)","(1.2+0j)"]]r )r� r$ )r r% rk r9 r] rQ )r, r3 r# r% r% r& �,test_default_handler_numpy_unsupported_dtypeU s $ z@TestPandasContainer.test_default_handler_numpy_unsupported_dtypec s� d� � fdd�}t jt� d��" tdddt� gi�j|d� W d Q R X t jt� d��& tdddtd d �gi�j|d� W d Q R X d S ) NZraisinc s t � ��d S )N)� TypeError)r( )rE r% r& �my_handler_raisese s zJTestPandasContainer.test_default_handler_raises.<locals>.my_handler_raises)rA r5 r< rJ )r� r\ r+ )rB rC r. r r r9 r% )r, r/ r% )rE r& �test_default_handler_raisesb s z/TestPandasContainer.test_default_handler_raisesc C sx t ddddddddgi�}|d |d< |�� }|d �d�|d< ||�� ksPt�|d }|d }|�� |�� kstt�d S )Nrq r5 r6 r7 rr �category)r r9 rW rQ )r, r3 r# r� �scr% r% r& �test_categoricalq s z$TestPandasContainer.test_categoricalc C s� t jdddd�}|�d��d �}t|t jddd�d��}|�� }||d< |�� }||�� ks`t�t|�}t|�}|�� |�� ks�t�d S ) NrI rH z US/Eastern)r� r �utc)r� )rq rr rq ) rv r� Z tz_convertr� r r* r9 rQ r )r, �tz_rangeZtz_naiver3 Zdf_naiver# ZstzZs_naiver% r% r& �test_datetime_tz~ s z$TestPandasContainer.test_datetime_tzc C s� t tj�dd��}tj|jd d�< |�d�}|�� }||�� ksDt�t tj�d��}tj|jd d�< |�d�}|�� }||�� ks�t�d S )Nr� r\ � ZSparse) r rN r r r� r� rW r9 rQ r )r, r3 Zsdfr# r� �ssr% r% r& � test_sparse� s z TestPandasContainer.test_sparsezsegfault GH 42130r. z2013-01-10 05:00:00Zz2013-01-10 00:00:00z US/Easternz2013-01-10 00:00:00-0500c C sD ddl m} d}||dd�|ks$t�|�� }||dd�|ks@t�d S )Nr )r� z"2013-01-10T05:00:00.000Z"T)� iso_dates)r� r� rQ Z to_pydatetime)r, r. r� �expr� r% r% r& �test_tz_is_utc� s z"TestPandasContainer.test_tz_is_utcr5 z2013-01-01 05:00:00ZrJ )r� z2013-01-01 00:00:00)r� r z2013-01-01 00:00:00-0500c C sl ddl m} d}d}||dd�|ks(t�t|�}||dd�|ksDt�td|i�}||dd�}||ksht�d S )Nr )r� z7["2013-01-01T05:00:00.000Z","2013-01-02T05:00:00.000Z"]zF{"DT":{"0":"2013-01-01T05:00:00.000Z","1":"2013-01-02T05:00:00.000Z"}}T)r: ZDT)r� r� rQ r r )r, r5 r� r; Zdfexpr� r3 r" r% r% r& �test_tz_range_is_utc� s z(TestPandasContainer.test_tz_range_is_utcc C s8 t ddd�}tddgddggddgd�}t�||� d S ) Nz!{"a": 1, "b": 2} {"b":2, "a" :1} T)�linesr< rJ r5 r6 )r )r r r r! )r, r" r# r% r% r& �test_read_inline_jsonl� s z*TestPandasContainer.test_read_inline_jsonlc C s: t dd|d�}tddgddggddgd�}t�||� d S ) Nz s3n://pandas-test/items.jsonlT)r> �storage_optionsr< rJ r5 r6 )r )r r r r! )r, � s3_resource�s3sor" r# r% r% r& �test_read_s3_jsonl� s z&TestPandasContainer.test_read_s3_jsonlc C sn t �d��Z}t|d��}|�d� W d Q R X t|dd�}tddgddggdd gd �}t �||� W d Q R X d S ) Nztmp_items.json�wz!{"a": 1, "b": 2} {"b":2, "a" :1} T)r> r< rJ r5 r6 )r )r r� �open�writer r r! )r, r� �infiler" r# r% r% r& �test_read_local_jsonl� s z)TestPandasContainer.test_read_local_jsonlc C s| d}t |�}t|dd�}tddgddggddgd �}t�||� d}t|dd�}tddgddggddgd �}t�||� d S ) Nu5 {"a": "foo”", "b": "bar"} {"a": "foo", "b": "bar"} T)r> u foo”r r r5 r6 )r )r r r r r! )r, r� r" r# r% r% r& � test_read_jsonl_unicode_chars� s z1TestPandasContainer.test_read_jsonl_unicode_chars�bigNumc C sp t |tdgd�}|�� }dt|� d }||ks4t�t|tdgdgd�}|�� }dt|� d }||kslt�d S ) N� articleId)r r/ z {"articleId":r� r )r r/ r z{"0":{"articleId":z}})r r r9 r] rQ r )r, rJ r� r� r# r3 r% r% r& �test_to_json_large_numbers� s z.TestPandasContainer.test_to_json_large_numberszGH-35279c C st t dt|� d �}d}tjt|d�� t|� W d Q R X t dt|� d �}tjt|d�� t|� W d Q R X d S )Nz {"articleId":r� z#Value is too small|Value is too big)rA z{"0":{"articleId":z}})r r] rB rC rD r )r, rJ r� rE r% r% r& � test_read_json_large_numbers s z0TestPandasContainer.test_read_json_large_numbersc C sh d}t |�}t|dd�}tddgd�}t�||� d}t |�}t|�}tddgdgd �}t�||� d S ) Nz'{"articleId": "1404366058080022500245"}r� )r� gg�_SDrK )r/ z.{"0": {"articleId": "1404366058080022500245"}}r )r/ r )r r r r r� r r! )r, r� r" r# r% r% r& � test_read_json_large_numbers2 s z1TestPandasContainer.test_read_json_large_numbers2c C s� t ddgddggddgd�}|jddd�}d }||ks:t�t d d gd d ggddgd�}|jddd�}d }||kstt�t�t|dd�|� t dd gd d ggddgd�}|jddd�}d}||ks�t�t�t|dd�|� d S )Nr< rJ r5 r6 )r r T)r$ r> z {"a":1,"b":2} {"a":1,"b":2} zfoo}r zfoo"z/{"a":"foo}","b":"bar"} {"a":"foo\"","b":"bar"} )r> zfoo\za\z4{"a\\":"foo\\","b":"bar"} {"a\\":"foo\"","b":"bar"} )r r9 rQ r r! r )r, r3 r" r# r% r% r& � test_to_jsonl s z!TestPandasContainer.test_to_jsonlc C s� t �d� dddddgddddgdddddgdddddgddddgddddgd ddddgtjdddgd tjdddgg }d d � |D �}g }x0d tgD ]$}x |D ]}|�t||d �� q�W q�W ddd�}x|D ] }||� q�W d S )Nz3encoding not implemented in .to_json(), xref #13774s E�, 17r � a� b� cs EE, 17s ��s A��c S s g | ]}d d� |D ��qS )c S s$ g | ] }t |t�r |�d �n|�qS )zlatin-1)r$ r� �decode)r^ r= r% r% r& r` C s zFTestPandasContainer.test_latin_encoding.<locals>.<listcomp>.<listcomp>r% )r^ r> r% r% r& r` C s z;TestPandasContainer.test_latin_encoding.<locals>.<listcomp>r1 )r �latin-1c S sD t �d��0}| j||d� t||d�}t j| |dd� W d Q R X d S )Nz test.json)�encodingF)Zcheck_categorical)r r� r9 r r� )r� rU r� Zretrr% r% r& � roundtripL s z:TestPandasContainer.test_latin_encoding.<locals>.roundtrip)rT )rB �skiprN r� r r� r )r, r Zexamplesr �valrV r� r% r% r& �test_latin_encoding2 s& z'TestPandasContainer.test_latin_encodingc C sN t dtd�gi�}|jddd��� }|�� |jddd��� }||ksJt�d S )Nr5 r< T)r/ �deep)r r] Z memory_usage�sumr9 rQ )r, r3 Z size_beforeZ size_afterr% r% r& �"test_data_frame_size_after_to_jsonU s z6TestPandasContainer.test_data_frame_size_after_to_jsong �?g @r| r} z1.z2.c C sB t ddgddgg||d�}|jdd�}t|dd�}t�||� d S )Nr< rJ rH r\ )r/ r �table)r$ )r r9 r r r! )r, r/ r r# �dfjsonr" r% r% r& �.test_from_json_to_json_table_index_and_columns_ s zBTestPandasContainer.test_from_json_to_json_table_index_and_columnsc C sD t ddgddgddgd��}|jdd �}t|dd �}t�||� d S ) Nr< rJ g @g @r r� )r5 r6 r7 r] )r$ )r r9 r r r! )r, r# r^ r" r% r% r& �#test_from_json_to_json_table_dtypesj s z7TestPandasContainer.test_from_json_to_json_table_dtypes)r6 r7 c C sX t ddgddgddgd��}|jdd �}d }tjt|d �� t|d|d � W d Q R X d S ) Nr< rJ g @g @r r� )r5 r6 r7 r] )r$ z)cannot pass both dtype and orient='table')rA )r$ r )r r9 rB rC rD r )r, r r3 r^ rE r% r% r& �!test_read_json_table_dtype_raisesq s z5TestPandasContainer.test_read_json_table_dtype_raisesc C s^ t ddgddggddgddgd �}|jd d �}d }tjt|d �� t|d dd� W d Q R X d S )Nr< rJ rH r\ g �?g @z1.z2.)r/ r r] )r$ z0cannot pass both convert_axes and orient='table')rA T)r$ rT )r r9 rB rC rD r )r, r3 r^ rE r% r% r& �(test_read_json_table_convert_axes_raisesz s " z<TestPandasContainer.test_read_json_table_convert_axes_raiseszdata, expectedr\ r )r )r rG r )r r/ rq )r )r rG )r r/ c C s( |j ddd�}t�|�}||ks$t�d S )Nr; F)r$ r/ )r9 r� �loadsrQ )r, rG r# r" r% r% r& � test_index_false_to_json_split� s z2TestPandasContainer.test_index_false_to_json_splitc C sL |j ddd�}t�|�}tjjj|dd�t|�jdd�d�}||ksHt�d S )Nr] F)r$ r/ )r/ r )r$ )ZschemarG ) r9 r� rc rv r� Zbuild_table_schemar �to_dictrQ )r, rG r" r# r% r% r& � test_index_false_to_json_table� s z2TestPandasContainer.test_index_false_to_json_tablec C sL t ddgddggddgd�}d}tjt|d �� |j|d d � W d Q R X d S ) Nr< rJ r\ r r5 r6 )r z?'index=False' is only valid when 'orient' is 'split' or 'table')rA F)r$ r/ )r rB rC rD r9 )r, r$ r3 rE r% r% r& � test_index_false_error_to_json� s z2TestPandasContainer.test_index_false_error_to_jsonr] c C s@ t ddgddgd��}|j||d�}t||d�}t�||� d S )Nr< rJ rH r\ )r5 r6 )r$ r/ )r$ )r r9 r r r! )r, r$ r/ r# r^ r" r% r% r& �"test_index_false_from_json_to_json� s z6TestPandasContainer.test_index_false_from_json_to_jsonc C s6 t dddd�}tdgtdgdd�d �}t�||� d S ) Nz {"2019-01-01T11:00:00.000Z":88}r� r/ )r� r$ �X z2019-01-01 11:00:00r� )r )r/ )r r r r r� )r, r" r# r% r% r& � test_read_timezone_information� s z2TestPandasContainer.test_read_timezone_informationzdate_format,key)r� i \&)r� Z P1DT0H0M0Sc C s@ t dggt�d�gd�}d|� d�}|j|d�}||ks<t�d S )Nr< Z1D)r z{"z ":{"0":1}})r� )r rv r r9 rQ )r, r� �keyr3 r# r" r% r% r& �test_timedelta_as_label� s z+TestPandasContainer.test_timedelta_as_labelzorient,expected)r/ z {"('a', 'b')":{"('c', 'd')":1}})r z {"('c', 'd')":{"('a', 'b')":1}}� )Zmarksc C s2 t dggdgdgd�}|j|d�}||ks.t�d S )Nr< )r5 r6 )r7 r8 )r/ r )r$ )r r9 rQ )r, r$ r# r3 r" r% r% r& �test_tuple_labels� s z%TestPandasContainer.test_tuple_labels�indentc C s� t ddgddggddgd�}|j|d�}d | }d |� d |� |� d |� |� d |� d|� d|� |� d|� |� d|� d�}||ks�t�d S )Nr r r r� r5 r6 )r )ro � z{ z"a":{ z "0":"foo", z "1":"baz" z}, z"b":{ z "0":"bar", z "1":"qux" z} })r r9 rQ )r, ro r3 r" �spacesr# r% r% r& �test_to_json_indent� s Fz'TestPandasContainer.test_to_json_indent)r; z�{ "columns":[ "a", "b" ], "index":[ 0, 1 ], "data":[ [ "foo", "bar" ], [ "baz", "qux" ] ] })r zf[ { "a":"foo", "b":"bar" }, { "a":"baz", "b":"qux" } ])r/ zn{ "0":{ "a":"foo", "b":"bar" }, "1":{ "a":"baz", "b":"qux" } })r zn{ "a":{ "0":"foo", "1":"baz" }, "b":{ "0":"bar", "1":"qux" } })r zV[ [ "foo", "bar" ], [ "baz", "qux" ] ])r] ab { "schema":{ "fields":[ { "name":"index", "type":"integer" }, { "name":"a", "type":"string" }, { "name":"b", "type":"string" } ], "primaryKey":[ "index" ], "pandas_version":"0.20.0" }, "data":[ { "index":0, "a":"foo", "b":"bar" }, { "index":1, "a":"baz", "b":"qux" } ] }c C s: t ddgddggddgd�}|j|dd �}||ks6t�d S ) Nr r r r� r5 r6 )r r\ )r$ ro )r r9 rQ )r, r$ r# r3 r" r% r% r& � test_json_indent_all_orients s w z0TestPandasContainer.test_json_indent_all_orientsc C s, t jtdd�� t� jdd� W d Q R X d S )Nz must be a nonnegative integer)rA ���)ro )rB rC rD r r9 )r, r% r% r& � test_json_negative_indent_raises� s z4TestPandasContainer.test_json_negative_indent_raisesc C s: d}t |�}tdtjdtjdtj dg�}t�||� d S )Nz?["a", NaN, "NaN", Infinity, "Infinity", -Infinity, "-Infinity"]r5 �NaN�Infinityz -Infinity)r r rN r� r� r r! )r, rG r" r# r% r% r& � test_emca_262_nan_inf_support� s z1TestPandasContainer.test_emca_262_nan_inf_supportc C sD t dddg�}t�t��" t|�� dd�}t�||� W d Q R X d S )Nr< rJ rH T)rU )r r Zassert_produces_warning� FutureWarningr r9 r! )r, r# r" r% r% r& �'test_deprecate_numpy_argument_read_json� s z;TestPandasContainer.test_deprecate_numpy_argument_read_jsonc C s> t �ddiddig�}tdddgi�}t|�}t�||� d S )N�colZ31900441201190696999�Text)r� r� r r r r! )r, Z encoded_jsonr# r" r% r% r& �test_frame_int_overflow� s z+TestPandasContainer.test_frame_int_overflowzdataframe,expected)r= r> zU{"(0, 'x')":1,"(0, 'y')":"a","(1, 'x')":2,"(1, 'y')":"b","(2, 'x')":3,"(2, 'y')":"c"}c C s$ |� � }|jdd�}||ks t�d S )Nr/ )r$ )�stackr9 rQ )r, Z dataframer# r� r" r% r% r& �test_json_multiindex� s z(TestPandasContainer.test_json_multiindexz localhost connection rejected)ra �strictc C s� dd l }d\}}tdddgdddgd��}|jd |� d |� �|d � d }xF|d d� |�d�j�� D �krjP |�d� |d8 }|dksJtd��qJW d S )Nr )z pandas-testz test.jsonr< rJ rH r\ r{ )r= r> zs3://�/)r@ r c s s | ] }|j V qd S )N)rk )r^ r( r% r% r& � <genexpr>� s z1TestPandasContainer.test_to_s3.<locals>.<genexpr>z pandas-testg�������?z,Timed out waiting for file to appear on moto)�timer r9 ZBucket�objects�all�sleeprQ )r, rA rB r� Zmock_bucket_nameZ target_filer3 �timeoutr% r% r& � test_to_s3� s z TestPandasContainer.test_to_s3c C sD t |t�r$tjjdd�}|j�|� t|gg��� }|dks@t �d S )Nznot implemented)ra z{"0":{"0":null}}) r$ r rB rd re rb rc r r9 rQ )r, Z nulls_fixturerf rd r" r% r% r& �test_json_pandas_nulls� s z*TestPandasContainer.test_json_pandas_nullsc C s* t ddd�}tdddg�}t�||� d S )Nz[true, true, false]r� )r� TF)r r r r� )r, r" r# r% r% r& �test_readjson_bool_series� s z-TestPandasContainer.test_readjson_bool_seriesc C s: t dt�dd�ddgd��� }|�� }d}||ks6t�d S )NTz 2017-01-20z 2017-01-23r r )r/ r a� {"(Timestamp('2017-01-20 00:00:00'), 'foo')":true,"(Timestamp('2017-01-20 00:00:00'), 'bar')":true,"(Timestamp('2017-01-21 00:00:00'), 'foo')":true,"(Timestamp('2017-01-21 00:00:00'), 'bar')":true,"(Timestamp('2017-01-22 00:00:00'), 'foo')":true,"(Timestamp('2017-01-22 00:00:00'), 'bar')":true,"(Timestamp('2017-01-23 00:00:00'), 'foo')":true,"(Timestamp('2017-01-23 00:00:00'), 'bar')":true})r rv r� r~ r9 rQ )r, r3 r" r# r% r% r& � test_to_json_multiindex_escape� s z2TestPandasContainer.test_to_json_multiindex_escape)�r� r� r� rB Zfixturer- r2 r4 r: rd Z parametrizer? rF r rO rP rS rk rV rN r rY Zfloat64�intrg ri rm ro rx r� rz r� r� r� ZNINFr� Zskipifr 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� r� r� r� r� r� r r r r r �networkZsingler rv ZDatetimeTZDtyper r r r r r r! r, r- r0 r3 r6 r9 r r< r� r= r? �tdZskip_if_not_us_localerC rH rI �sys�maxsizerL r rM rN rO rY r\ r_ r` ra rb r Z rename_axisrd rf rg rh rj rl �paramrW rn rr rs ru rx rz r} r re r r r� r� r� r� r% r% r% r& r( 2 s� & ' `& $ ( # 26 & ( , ( $ (, 2" * $ , r( )0r� r �decimalr r� r r� r� r� rU rN rB Z pandas.compatr r r r Z pandas.util._test_decorators�utilZ_test_decoratorsr� Zpandasrv r r r r r r Zpandas._testingZ_testingr Z getSeriesDataZ_seriesdZ_framer* r) r �catZCategoricalIndexr/ r� �reversedr� r' rd �filterwarningsr( r% r% r% r& �<module> s0 0
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка