
    o[we~%                        d dl Z d dl mZ d dlmZ d dlmZ d dlZd dlZd dlZd dl	Z	d dl
Zd dlZd dlmZ d dlmc mZ d dlZd dlmZmZmZmZmZmZ d dlmZ d dlmZm Z  d Z!d	 Z" G d
 d          Z#d Z$d Z%d Z&dS )    N)	timedelta)Decimal)StringIO)IS64)NA	DataFrameDatetimeIndexSeries	Timestamp	read_json)ArrowStringArrayStringArrayc                     t          ddgddggddg          } d}d}t          j        t          |          5  	 t	          |d	
           n# t
          $ r Y nw xY wd d d            n# 1 swxY w Y   t          j        t          |          5  t	          |                                 d	
           d d d            n# 1 swxY w Y   t          j        t          |          5  t	          dd
          }t          j        ||            d d d            n# 1 swxY w Y   t          j        t          |          5  	 t	          dd	
          }n# t
          $ r Y nw xY wd d d            n# 1 swxY w Y   t          j        t          |          5  	 t	          dd	
          }n# t
          $ r Y nw xY wt          j        ||            d d d            d S # 1 swxY w Y   d S )N      abcolumnsz[{"a": 1, "b": 2}
        {"a": 3, "b": 4}
        {"a": 5, "b": 6}
        {"a": 7, "b": 8}zPassing literal json to 'read_json' is deprecated and will be removed in a future version. To read from a literal string, wrap it in a 'StringIO' object.matchFlines!{"a": 1, "b": 2}
{"b":2, "a" :1}
T4{"a\\":"foo\\","b":"bar"}
{"a\\":"foo\"","b":"bar"}
)r   tmassert_produces_warningFutureWarningr   
ValueErrorto_jsonassert_frame_equal)expectedjsonlmsgresults       @lib/python3.11/site-packages/pandas/tests/io/json/test_pandas.pytest_literal_json_deprecationr'       sv   1a&1a&)C:>>>HE	:  
	#M	=	=	=  	e5))))) 	 	 	D	               
	#M	=	=	= 3 3(""$$E22223 3 3 3 3 3 3 3 3 3 3 3 3 3 3 
	#M	=	=	= 0 0@MMM
fh///0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 
	#M	=	=	=  	O  FF  	 	 	D	               
	#M	=	=	= 0 0	DERRRFF 	 	 	D	
fh///0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0s   A)AA)
AA)AA))A-0A-$B??CC%'DDD>E. EE.
EE.EE..E25E2GF('G(
F52G4F55GG!Gc                     |dv r|                     d          }|dk    r&t          t          |j                            |_        t	          j        | |           d S )N)recordsvaluesTdropr*   )reset_indexrangelenr   r   r!   )r%   r"   orients      r&   assert_json_roundtrip_equalr1   M   sc    &&&''T'22 X%5!6!677&(+++++    c                   |   e Zd Zej        d             Zej        d             Zej        d             Zd Zej	        
                    dg d          d             Zej	        
                    ddd	g          d
             Zej	        
                    dddg          ej	        
                    dddgddggddgddggddgddgg ed          dg ed          dggg          d                         Zej	        
                    dg d          d             Zd Zej	        
                    ddeg          ej	        
                    d d!dg          d"                         Zej	        
                    ddej        g          ej	        
                    d d!dg          d#                         Zej	        
                    dd$ej        ed%g          ej	        
                    d d!dg          d&                         Zej	        
                    d d!dg          d'             Zej	        
                    d d!dg          d(             Zej	        
                    d d!dg          d)             Zej	        
                    d d!dg          d*             Zej	                            d+e,          ej	        
                    d	d-d-gd.d/ggd-d0gd.d.ggd-d-gd.d.ggg          d1                         Zej	        
                    d2d3d4d5                     d6g          dfd7d8g          d9             Z!ej	        
                    dd!dg          ej	        
                    d d!dg          d:                         Z"ej	        
                    dd!dg          d;             Z#ej	        
                    d<ej$        ej$         g          ej	        
                    dd!dg          d=                         Z%ej	        &                    e' d>?          ej	        
                    d@g dA          dB                         Z(dC Z)dD Z*dE Z+dF Z,dG Z-dH Z.dI Z/dJ Z0dK Z1dL Z2dM Z3dN Z4ej	        
                    ddd$g          dO             Z5dP Z6dQ Z7ej	        
                    dej        eg          dR             Z8dS Z9dT Z:dU Z;ej	        
                    dVd! e<dWgdXY          fd e<dZg          fg          d[             Z=d\ Z>d] Z?d^ Z@d_ ZAd` ZBda ZCej	        
                    dbdcddg          ej	        
                    ded!dg          ej	        
                    dfeDjE        eDjD        eg          dg                                     ZFej	        
                    dhg di          dj             ZGej	        
                    dkg dl          dm             ZHdn ZIej	        
                    dkg dl          do             ZJdp ZKej	        
                    dqg dr          ds             ZLej	        
                    dqg dt          du             ZMdv ZNdw ZOdx ZPej	        jQ        ej	        jR        ej	        
                    dydz eSjT        d{|          gd}dXgd~ eSjT        d{|          gg          d                                     ZUd ZVd ZWd ZXej	        
                    ded!dg          ej	        
                    dbdddcg          ej	        
                    deSjY        eZg          d                                     Z[d Z\d Z]d Z^d Z_d Z`d Zad Zbej	        
                    d ed           edd|           ed          g          d             Zcd Zdej	        
                    d eSje        dd           eSje        ddd           eSje        dd          g          d             Zfd Zgd Zhej	        jR        eijj        d                         Zkd Zld Zmej	        
                    denjo        dz   enjo        dz    g          d             Zpej	        
                    dddg          d             Zqd Zrd Zsej	                            det,          ej	        
                    dg dg dg dg dg dg dg deju        dddgdeju        dddgg	          ej	        
                    ddevg          d                                     Zwd Zxej	        
                    dd$ddgddgddgddgddgg          ej	        
                    d	ddgddgddgg          d                         Zyd Zzej	        
                    dg d          d             Z{ej	        
                    dd!eedg          d             Z|ej	        
                    dg dâ          dĄ             Z}dń Z~ej	        
                    d eddgddggddgɦ          ddgddgddggdʜf eddgddggddgɦ                              d˦          ddgddgddggdʜf eddgddggddgddgddgg̦          ddgddgddggdʜf e<g d͢dάϦ          dg d͢dМf e<g d͢dάϦ                              d˦          dg d͢dМf e<ddgdddgddggѦ          dddgdМfg          d҄             Zej	        
                    d eddgddggddgɦ           eddgddggddgɦ                              d˦           eddgddggddgddgddgg̦           e<g d͢dάϦ           e<g d͢dάϦ                              d˦           e<ddgdddgddggѦ          g          dӄ             Zej	        
                    ddd	g          dԄ             Zej	        
                    dddg          dք             Zej	        
                    dddg          ej	        
                    dd!dg          d؄                         Zdل Zej	        
                    dg dۢ          d܄             Zej	        
                    dg dޢ          d߄             Zej	        
                    dddg          d             Zej	        
                    ddd ej        ddej	                            d?                     ej        ddej	                            d?                    g          d             Zej	        
                    dg d          d             Zej	        
                    dg d          d             Zd Zd Zd Zej	        
                    d eg d͢g dd          dfg          d             Zej	        jR        d             Zd Zd Zd Zd Zej	        
                    d e<ddd d          df e<dddd          df eddgd	d
gg          df eddgddgg          dfg          d             Zd Zej	        
                    dg d          d             Zej	        
                    dg d          d             Zd Zd$S (  TestPandasContainerc                 `   t          j                    }t          |          }dgdz  dgdz  z   dgdz  z   dgt          |          dz
  z  z   }t	          j        |d          |_        t          t          |                    |d<   t          j
        t          |          d	
          |d<   |S )Nbah   barbazfoo   Enameint64dtypesort)r   getSeriesDatar   r/   pdCategoricalIndexindexlistreversednparange)self_seriesd
_cat_framecats       r&   categorical_framez%TestPandasContainer.categorical_frameV   s    #%%x((
gkUGaK'5'A+53z??UWCW8XX.s===
x}}--
3Ys:gFFF
6r2   c                 x    t          j                    }d|_        |j                            d           |_        |S )Nts)r   makeTimeSeriesr>   rF   
_with_freq)rK   sers     r&   datetime_seriesz#TestPandasContainer.datetime_seriesb   s5     !!I((..	
r2   c                     t          t          j                              }|j                            d           |_        |S N)r   r   getTimeSeriesDatarF   rS   )rK   dfs     r&   datetime_framez"TestPandasContainer.datetime_framek   s6     r+--..8&&t,,	r2   c                     t          ddgddggddgddg	          }t          |                    |
                    }t          ||
          }|                                }t          |||           d S )Nr   r   cdz	index " 1z	index / 2za \ bzy / zrF   r   r0   r   r   r    r   copyr1   rK   r0   rY   datar%   r"   s         r&    test_frame_double_encoded_labelsz4TestPandasContainer.test_frame_double_encoded_labelss   s    3Z#s$,w'
 
 
 

&
11224///7799#FHf=====r2   r0   )splitr)   r*   c                     t          ddgddggddgddg          }t          |                    |	                    }t          ||	          }|                                }t          |||           d S )
Nr   r   r\   r]   r   xyr^   r_   r`   rb   s         r&   test_frame_non_unique_indexz/TestPandasContainer.test_frame_non_unique_index   s}    c
S#J/1vSzRRR

&
11224///7799#FHf=====r2   rF   r   c                     t          ddgddggddgddg          }d	| d
}t          j        t          |          5  |                    |           d d d            d S # 1 swxY w Y   d S )Nr   r   r\   r]   r   rg   rh   r^   z+DataFrame index must be unique for orient=''r   r_   r   pytestraisesr   r    rK   r0   rY   r$   s       r&   "test_frame_non_unique_index_raisesz6TestPandasContainer.test_frame_non_unique_index_raises   s    c
S#J/1vSzRRREFEEE]:S111 	& 	&JJfJ%%%	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	&   A""A&)A&re   r*   rc   r   r   r\   r]         ?      @g      @g      @r      2013010120130102c                    t          |ddgddg          }t          t          |                    |                    |dg          }|dk    rgt          |          }|j        d d df         j        d	k    r=|j        d d df                             t          j                  d
z  |j        d d df<   n|dk    r|}ddg|_	        t          j        ||           d S )Nr   r   rg   r^   r_   )r0   convert_datesr*   r   datetime64[ns]@B re   zx.1)r   r   r   r    ilocrA   viewrI   r?   r   r   r!   )rK   r0   rc   rY   r%   r"   s         r&   test_frame_non_unique_columnsz1TestPandasContainer.test_frame_non_unique_columns   s     tAq6C:>>>RZZvZ..//se
 
 
 X H}QQQT"(,<<<
 '/mAAAqD&9&>&>rx&H&HG&Saaad#wH #U|H
fh/////r2   )rF   r   r)   c                     t          ddgddggddgddg          }d	| d
}t          j        t          |          5  |                    |           d d d            d S # 1 swxY w Y   d S )Nr   r   r\   r]   r   r   rg   r^   z-DataFrame columns must be unique for orient='rk   r   r_   rl   ro   s       r&   $test_frame_non_unique_columns_raisesz8TestPandasContainer.test_frame_non_unique_columns_raises   s    c
S#J/1vSzRRRGfGGG]:S111 	& 	&JJfJ%%%	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	&rq   c                 b    |                                 |                     d          k    sJ d S )Nr   r_   r    )rK   float_frames     r&   test_frame_default_orientz-TestPandasContainer.test_frame_default_orient   s6    ""$$(;(;9(;(M(MMMMMMMr2   rA   Fconvert_axesTc                     t          |                    |                    }t          ||||          }|}t          |||           d S Nr_   r0   r   rA   r   r    r   r1   )rK   r0   r   rA   r   rc   r%   r"   s           r&   test_roundtrip_simplez)TestPandasContainer.test_roundtrip_simple   sV     ++6+::;;4\QVWWW#FHf=====r2   c                     t          |                    |                    }t          ||||          }|}t          |||           d S r   r   )rK   r0   r   rA   	int_framerc   r%   r"   s           r&   test_roundtrip_intframez+TestPandasContainer.test_roundtrip_intframe   sV     	)))88994\QVWWW#FHf=====r2   NU3c                    t          t          j        d          d t          d          D             d t          d          D             |          }t	          |                    |                    }t          ||||          }|                                }|s|                    t          j	                  }|rW|d	v rS|j
                            t          j	                  |_
        |j                            t          j	                  |_        nc|d
k    r,|r*|j
                            t          j	                  |_
        n1|r/|dk    r)|j
                            t          j	                  |_
        t          |||           d S )N)      c                 ,    g | ]}t          |          S  str.0is     r&   
<listcomp>z?TestPandasContainer.test_roundtrip_str_axes.<locals>.<listcomp>   s    ...SVV...r2   r   c                 ,    g | ]}t          |          S r   r   r   s     r&   r   z?TestPandasContainer.test_roundtrip_str_axes.<locals>.<listcomp>   s    ...a3q66...r2   r   )r   rF   rA   r_   r   r^   r)   re   )r   rI   zerosr.   r   r    r   ra   astyper?   r   rF   r1   )rK   r0   r   rA   rY   rc   r%   r"   s           r&   test_roundtrip_str_axesz+TestPandasContainer.test_roundtrip_str_axes   sr    HX..U1XX.....5::...	
 
 
 

&
11224\QVWWW7799 	1rx00H  	AV';;;'/66rx@@H%^2228<<HNNy  \ '/66rx@@H 	Af//'/66rx@@H#FHf=====r2   c                    |dv r<|j                             t          j                            d| d                     t          |                    |                    }t          |||          }|                                }|j	        
                    t                    |_	        d |j	        _        t          |||           d S )Nr^   z.Can't have duplicate index values for orient 'z')reasonr_   r0   r   )node
add_markerrm   markxfailr   r    r   ra   rF   r   r   r>   r1   )rK   requestr0   rO   r   rc   r%   r"   s           r&   test_roundtrip_categoricalz.TestPandasContainer.test_roundtrip_categorical   s    
 )))L##!!VFVVV "     )111@@AA4\JJJ$))++!..s33"#FHf=====r2   c                 |   t                      }t          |                    |                    }t          |||          }|dk    r6t	          j        g |rt          nt                    }t          ||          }n'|dv rt                      }n|                                }t          j
        ||           d S )Nr_   r   re   r@   r^   )r   r   r    r   rD   Indexfloatobjectra   r   r!   )rK   r0   r   empty_framerc   r%   idxr"   s           r&   test_roundtrip_emptyz(TestPandasContainer.test_roundtrip_empty   s    kk++6+::;;4\JJJW(2|&GeeIIIC sC888HH+++ {{HH"''))H
fh/////r2   c                 Z   t          |                    |                    }t          |||          }|                                }|sN|j                            t          j                  dz  }|dk    r|                    t                    }||_        t          |||           d S )Nr_   r   rz   re   )r   r    r   ra   rF   r|   rI   r?   r   r   r1   )rK   r0   r   rZ   rc   r%   r"   r   s           r&   test_roundtrip_timestampz,TestPandasContainer.test_roundtrip_timestamp  s     ..f.==>>4\JJJ!&&(( 	!.%%bh//7:C  jjoo HN#FHf=====r2   c                    t          j        g d          }g dg dg dg dd}t          ||          }t          |                    |                    }t          |||	          }|                                } |j        di |                    d
          	                    t          j                  }t          |||           d S )N)r   r   r\   r]   e)              ?       @      @      @)r   r   r   r   r   )foo1foo2foo3foo4foo5)TFTFTABCD)rc   rF   r_   r   numberr   )rD   r   r   r   r    r   ra   assignselect_dtypesr   rI   r?   r1   )	rK   r0   r   rF   r*   rY   rc   r%   r"   s	            r&   test_roundtrip_mixedz(TestPandasContainer.test_roundtrip_mixed  s    22233******999111	
 
 F%000

&
11224\JJJ7799"8?WWX%;%;H%E%E%L%LRX%V%VWW#FHf=====r2   z9#50456 Column multiindex is stored and loaded differently)r   rn   2022JANFEB2023c                     t          ddgddggt          j                            |                    }t	          |                    d                    }t          |d          }t          j        ||           d S )Nr   r   rt   r   r   re   r_   )	r   rD   
MultiIndexfrom_arraysr   r    r   r   r!   )rK   r   rY   rc   r%   s        r&   test_roundtrip_multiindexz-TestPandasContainer.test_roundtrip_multiindex2  s     VaVM--g66
 
 
 

'
22334000
fb)))))r2   zdata,msg,orient)z{"key":b:a:d}zExpected object or valuer   zO{"columns":["A","B"],"index":["2","3"],"data":[[1.0,"1"],[2.0,"2"],[null,"3"]]}|z;Length of values \(3\) does not match length of index \(2\))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 columnsre   )zN{"badkey":["A","B"],"index":["2","3"],"data":[[1.0,"1"],[2.0,"2"],[null,"3"]]}zunexpected key\(s\): badkeyre   c                     t          j        t          |          5  t          t	          |          |           d d d            d S # 1 swxY w Y   d S )Nr   r_   )rm   rn   r   r   r   )rK   rc   r$   r0   s       r&   $test_frame_from_json_bad_data_raisesz8TestPandasContainer.test_frame_from_json_bad_data_raisesG  s    F ]:S111 	5 	5htnnV4444	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5s   AAAc                    t          ddgg dg          }t          t          |                    |                    |||          }t	          j        |j        d                   sJ t          ddgg d	g          }t          t          |                    |                    |||          }t	          j        |j        d                   sJ d S )
Nr   r   r   r7      r_   r   r   r   12)456)r   r   r   r    rI   isnanr{   )rK   r0   r   rA   num_dfr%   obj_dfs          r&   !test_frame_from_json_missing_dataz5TestPandasContainer.test_frame_from_json_missing_datam  s     QFIII.//V^^6^2233%	
 
 
 xD)*****S#J899V^^6^2233%	
 
 
 xD)*******r2   c                     t          t          d          |          }t          t          j        g          }t          j        ||           d S )Nz[null]r@   )r   r   r   rI   nanr   r!   )rK   rA   r%   r"   s       r&   (test_frame_read_json_dtype_missing_valuez<TestPandasContainer.test_frame_read_json_dtype_missing_value  sH     8H--U;;;bfX&&
fh/////r2   infc                     t          ddgg dg          }||j        d<   t          |                                          }t	          ||          }t          j        |j        d                   sJ d S )Nr   r   r   r   r@   )r   locr   r    r   rI   r   r{   )rK   r   rA   rY   rc   r%   s         r&   test_frame_infinityz'TestPandasContainer.test_frame_infinity  ss    
 A			*++t

%%4u---xD)*******r2   z$not compliant on 32-bit, xref #15865r   zvalue,precision,expected_val))gffffff?r   r   )g333333?r   r   )g333333r   g       )gףp=
?r   r   )gCl?rt   r   )g?r;   r   c                 n    t          d|ig          }|                    |          }|d| dk    sJ d S )Na_float)double_precisionz{"a_float":{"0":}}r   r    )rK   value	precisionexpected_valrY   encodeds         r&   "test_frame_to_json_float_precisionz6TestPandasContainer.test_frame_to_json_float_precision  sO     E*+,,**i*88A|AAAAAAAAAr2   c                     t          g d          }d}t          j        t          |          5  |                    d           d d d            d S # 1 swxY w Y   d S Nr   r   rt   z+Invalid value 'garbage' for option 'orient'r   garbager_   rl   )rK   rY   r$   s      r&   test_frame_to_json_exceptz-TestPandasContainer.test_frame_to_json_except  s    yyy!!;]:S111 	) 	)JJiJ(((	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	)   AAAc                     t          ddg          }|j        rJ t          |                                          }t	          |t          |j                            }t          j        ||d           d S )Njimjoer   r@   Fcheck_index_type)	r   _is_mixed_typer   r    r   dictdtypesr   r!   )rK   rY   rc   r%   s       r&   test_frame_emptyz$TestPandasContainer.test_frame_empty  sr    u~...$$$$

%%4tBI777
fb5AAAAAAr2   c                 l    t          dg ig           }|                    d          }d}||k    sJ d S )NtestrF   r   r_   z{"test":{}}r   rK   rY   r%   r"   s       r&   test_frame_empty_to_jsonz,TestPandasContainer.test_frame_empty_to_json  sG    |2...9-- !!!!!!r2   c                 *   t          ddg          }|d                             d          |d<   |j        sJ |                                }t	          j        t          t          |          t          |j	                            |d           d S )Nr   r   r   i8r@   Fr   )
r   r   r   r    r   r!   r   r   r   r   )rK   rY   rc   s      r&   test_frame_empty_mixedtypez.TestPandasContainer.test_frame_empty_mixedtype  s    u~...uI$$T**5	    zz||
htnnDOO<<<"	
 	
 	
 	
 	
 	
r2   c                    g dg dg dg dg}t          |t          d          g d          }|j        sJ |                                }dD ]L}t	          |                    |	                    }t          ||d
          }t          j        ||           Mt          j
        t          |                    |_        t	          |                    d	                    }t          |dd
          }t          j        ||           t          j
        |j        d                   |_        t	          |                    d	                    }t          |dd
          }t          j        ||           d S )N)
   r   r:   皙?g{Gz?)   r   r8   g?g{Gz?)   rt   r9   g333333?gQ?)(   r   quxg?g{Gz?abcd)1st2nd3rd4th5thr^   )re   rF   r   r_   Fr   r)   r   r*   )r   rG   r   ra   r   r    r   r   r!   rD   
RangeIndexr/   rF   shaper   )rK   valsrY   rightr0   inplefts          r&   test_frame_mixedtype_orientz/TestPandasContainer.test_frame_mixedtype_orient  s   %%%%%%%%%%%%	
 V.Q.Q.Q
 
 
     		3 	/ 	/F2::V:4455CSeDDDD!$....mCGG,,rzzz3344YUCCC
dE***bhqk22rzzz2233XEBBB
dE*****r2   c                 4   t          j        dd          }t          t          j        |          d           }t          g dg dg dg dg dgg d	|
          }t          d                              d          |d<   t          d          |j        d|j	        
                    d          f<   |d         |d<   t           j        |j        d|j	        
                    d          f<    |ddd          }t          j                            |d          }t          |          }t!          j        ||           |                    dgd          }t          j                            |d          }t          |          }	t!          j        ||	           d S )Nz
2000-01-03z
2000-01-07freq)gÉ?gz~Kc?gX7?gH/ƿ)g|RпgXgsBfg(ƿ)g<1=?g9?gg??gLł
)gJQg[r?T?gj<DH>?g-ѿ)gIEx?gW!i`ygP04?g+]5?r   r   rF   z19920106 18:21:32.12nsdateru   rt   modifiedr   iojsonrc   ztsframe_v012.json)axisztsframe_iso_v012.json)rD   
date_ranger	   rI   asarrayr   r   as_unitr{   r   get_locNaTospathjoinr   r   r!   r,   )
rK   datapathdtirY   dirpathv12_jsondf_unserdf_isov12_iso_jsondf_unser_isos
             r&   test_v12_compatz#TestPandasContainer.test_v12_compat  s   mL,77BJsOO$777AAACCC???BBBAAA )((

 

 

 566>>tDD6
1::1F1F2:%%f---.F:57V2:%%j1112(4007<<)<==X&&
b(+++*A..w||G-DEE ..
fl33333r2   c           	         t          j        ddd          }t          t          |          d           }t	          g dg dg dg d	g d
g dd|          }|j                            t          j                  |_        t          |
                    d                    }t          |d          }t          j        ||dddd           d S )N20000101r  H)periodsr  r  )
g}S? wgbm?gz(~J	?g1?gk*MgvTj?g7j	" Y?gp+KS(g񒬁?gZtSy)
iL,i_}i4ziKeiL&i%giDiY)L i;eiD )
78c608f164a9974313d2ff52ca7f4af297236474bde7e2141a6bde47b1190be57a6691448d64d068)
gD!gIxJ g%
@gP\<g޿gPl?gFzn?g5mH/?gpH%gI?)
14f04af9d085da904bcfac8381504caf2ffef4a908e2f5c407e1af03addbd4a71f6a09ba4bfc4d87)
i/iiYi6iI i;>i	i( i0q{iѪ)float_1int_1str_1float_2str_2int_2r   re   r_   T)r   check_column_type	by_blockscheck_exact)rD   r#  r	   rG   r   r   r   rI   str_r   r    r   r   r!   )rK   rF   df_mixedrc   df_roundtrips        r&   test_blocks_compat_GH9037z-TestPandasContainer.test_blocks_compat_GH9037  s9   j"3???d5kk555            {I IT WL
 L
 L
^ $+2227;;(((8899 g666
!"	
 	
 	
 	
 	
 	
r2   c                     G d d          }d} ||          }t          d|j        gi          }|                                d| dk    sJ t          d|gi          }d}t          j        t
          |          5  |                                 d d d            n# 1 swxY w Y   t          |gd	gd
ddg          }t          j        t
          |          5  |                                 d d d            n# 1 swxY w Y   |                    t                    }d| d}	||	k    sJ |                    t                    d| dk    sJ d S )Nc                   "    e Zd ZddZdefdZdS )FTestPandasContainer.test_frame_nonprintable_bytes.<locals>.BinaryThingreturnNc                 R    || _         t                              |          | _        d S rW   )hexedbytesfromhexbinary)rK   r^  s     r&   __init__zOTestPandasContainer.test_frame_nonprintable_bytes.<locals>.BinaryThing.__init__j  s     "
#mmE22r2   c                     | j         S rW   )r^  rK   s    r&   __str__zNTestPandasContainer.test_frame_nonprintable_bytes.<locals>.BinaryThing.__str__n  s
    z!r2   r\  N)__name__
__module____qualname__rb  r   re  r   r2   r&   BinaryThingr[  i  sC        3 3 3 3" " " " " " "r2   rj  574b4454ba8c5eb4f98a8f45r   z{"A":{"0":""}}z6Unsupported UTF-8 sequence length when encoding stringr   r   r   r   r   r   default_handlerz"},"B":{"0":1}})r   r^  r    rm   rn   OverflowErrorr   )
rK   rj  r^  binthingdf_printabledf_nonprintabler$   rV  r%   r"   s
             r&   test_frame_nonprintable_bytesz1TestPandasContainer.test_frame_nonprintable_bytesf  s2   	" 	" 	" 	" 	" 	" 	" 	" +;u%% !#'7!899##%%)E)E)E)EEEEE $S8*$566F]=444 	& 	&##%%%	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& H:QC883*MMM]=444 	 		 	 	 	 	 	 	 	 	 	 	 	 	 	 	 !(((==/5///!!!!S119u999: : : : : :s$   ?B  B$'B$C??DDc                 |    t          ddz  dgddgi                                          }dddz   d}||k    sJ d S )Nr8   i r   r:   i9  {"z":{"0":1},"foo":{"0":1337}}r   rK   r%   r"   s      r&   test_label_overflowz'TestPandasContainer.test_label_overflow  sW    EFNQC?@@HHJJIIII!!!!!!r2   c           
         t          ddgddg          }d}t          j        t          |          5  |                    d           d d d            n# 1 swxY w Y   t          j        |t          t          |                    d	                    d	d
                     t          t          |                    d                    dd
          }t          j	        |j
        |j
                   d S )Nr   r   r   r   z.Series index must be unique for orient='index'r   rF   r_   re   series)r0   typr)   )r
   rm   rn   r   r    r   assert_series_equalr   r   assert_numpy_array_equalr*   )rK   sr$   unserializeds       r&   test_series_non_unique_indexz0TestPandasContainer.test_series_non_unique_index  s;   C:aV,,,>]:S111 	& 	&IIWI%%%	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	'2233G  	
 	
 	
 !QYYiY0011)
 
 
 	#AHl.ABBBBBs   AAAc                 b    |                                 |                     d          k    sJ d S )NrF   r_   r   )rK   string_seriess     r&   test_series_default_orientz.TestPandasContainer.test_series_default_orient  s6    $$&&-*?*?w*?*O*OOOOOOOr2   c                     t          |                    |                    }t          |d|          }|}|dv r|                    d          }|dk    rd |_        t          j        ||           d S Nr_   rz  r{  r0   r*   r)   Tr+   re   r   r    r   r-   r>   r   r|  )rK   r0   r  rc   r%   r"   s         r&   test_series_roundtrip_simplez0TestPandasContainer.test_series_roundtrip_simple  s    --V-<<==4Xf=== ***+++66HW HM
vx00000r2   c                     t          |                    |                    }t          |d||          }|}|dv r|                    d          }|dk    rd |_        t          j        ||           d S )Nr_   rz  )r{  r0   rA   r  Tr+   re   r  )rK   r0   rA   object_seriesrc   r%   r"   s          r&   test_series_roundtrip_objectz0TestPandasContainer.test_series_roundtrip_object  s    --V-<<==4XfEJJJ ***+++66HW HM
vx00000r2   c                 X   t          g g t          j                  }t          |                    |                    }t          |d|          }|                    d          }|dv r)|j                            t          j                  |_        t          j
        ||           d S )NrF   rA   r_   rz  r  Tr+   re   )r
   rI   float64r   r    r   r-   rF   r   r   r|  )rK   r0   empty_seriesrc   r%   r"   s         r&   test_series_roundtrip_emptyz/TestPandasContainer.test_series_roundtrip_empty  s    b"*===,,F,;;<<4Xf===+++66g%^222:>>HN
vx00000r2   c                     t          |                    |                    }t          |d|          }|}|dv r|                    d          }|dk    rd |_        t          j        ||           d S r  r  )rK   r0   rU   rc   r%   r"   s         r&    test_series_roundtrip_timeseriesz4TestPandasContainer.test_series_roundtrip_timeseries  s    //v/>>??4Xf==="***+++66HW HM
vx00000r2   c                 6   t          t          d          g d          }t          |                    |                    }t	          |d|          }|                                }|dv r|                    d	          }t          j        ||           d S )
Nr   r   r   r\   r]   r   fr   r_   rz  r  r  Tr+   )	r
   r.   r   r    r   ra   r-   r   r|  )rK   r0   rA   r~  rc   r%   r"   s          r&   test_series_roundtrip_numericz1TestPandasContainer.test_series_roundtrip_numeric  s    588#A#A#ABBB			00114Xf===6688***+++66H
vx00000r2   c                     t          g d          }d}t          j        t          |          5  |                    d           d d d            d S # 1 swxY w Y   d S r   )r
   rm   rn   r   r    )rK   r~  r$   s      r&   test_series_to_json_exceptz.TestPandasContainer.test_series_to_json_except  s    999;]:S111 	( 	(IIYI'''	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	(r   c                     t          g d          }t          t          |                                          dd          }t	          j        ||d           d S )N=
ףp=@r  r  rz  T)r{  precise_floatFr   )r
   r   r   r    r   r|  rK   r~  r%   s      r&   #test_series_from_json_precise_floatz7TestPandasContainer.test_series_from_json_precise_float  sX    %%%&&8AIIKK00hdSSS
vq5AAAAAAr2   c                     t          g d          }t          t          |                                          dt          j                  }t          dgdz            }t          j        ||           d S )Nr  rz  r{  rA   r   rt   )r
   r   r   r    rI   r?   r   r|  )rK   r~  r%   r"   s       r&   test_series_with_dtypez*TestPandasContainer.test_series_with_dtype  sf    %%%&&8AIIKK00hbhOOO1#'??
vx00000r2   zdtype,expected
2000-01-01ry   r@   l    ,Uqc                     t          dgd          }t          |                                          }t          |d|          }t	          j        ||           d S )Nr  ry   r@   rz  r  )r
   r   r    r   r   r|  )rK   rA   r"   r~  rc   r%   s         r&   test_series_with_dtype_datetimez3TestPandasContainer.test_series_with_dtype_datetime  s[     L>)9:::		$$4XU;;;
vx00000r2   c                     t          g dg dg          }t          t          |                                          d          }t	          j        ||           d S )Nr  T)r  r   r   r   r    r   r!   rK   rY   r%   s      r&   "test_frame_from_json_precise_floatz6TestPandasContainer.test_frame_from_json_precise_float  s]    ***,>,>,>?@@8BJJLL11FFF
fb)))))r2   c                     t          t          d          g dd          }t          t          |                                          d           }t          j        ||           d S )Nr   r  r?   r  r{  )r
   r.   r   r   r    r   r|  r  s      r&   test_typzTestPandasContainer.test_typ  s^    588#A#A#AQQQ8AIIKK00d;;;
vq)))))r2   c                 p   t          g dg dg          }t          t          |                                                    }t	          j        ||           t          g dg ddg d          }t          t          |                                                    }t	          j        ||           d S )Nr   r   r   r   )r   r   r   r   r  r  s      r&   test_reconstruction_indexz-TestPandasContainer.test_reconstruction_index  s    			999-..8BJJLL1122
fb)))YYYYYY77OOO8BJJLL1122
fb)))))r2   c                     t          j        d          5 }|||fD ]&}|                    |           t          |           '	 d d d            d S # 1 swxY w Y   d S )N	test.json)r   ensure_cleanr    r   )rK   r   r   rZ   r)  rY   s         r&   	test_pathzTestPandasContainer.test_path  s    _[)) 	 T"I~>    

4   $ 	  	  	  	  	  	  	  	  	  	  	  	  	  	  	  	  	  	 s   -AAAc                 4   t          |                                          }t          |          }t          j        ||           t          |                                          }t          |d          }t          j        ||d           |j        J d S )Nrz  r  F)check_names)r   r    r   r   r!   r|  r>   )rK   rU   rZ   r!  r%   s        r&   test_axis_datesz#TestPandasContainer.test_axis_dates  s    ..00114
fn555 //11224X...
vEJJJJ{"""""r2   c                    |}t          d                              d          |d<   t          |                                          }t	          |          }t          j        ||           d|d<   t          |                    d                    }t	          |d          }|                                }|d         j        	                    d	          |d<   |d         
                    d
          |d<   t          j        ||           t          t          d                              d          |j                  }t          |                                          }t	          |d          }t          j        ||           d S )Nru   r  r  r   r:   	date_unitFrx   r  r?   r   rz  r  )r   r%  r   r    r   r   r!   ra   r*   r|   r   r
   rF   r|  )rK   rU   rZ   rY   r!  r%   r"   rQ   s           r&   test_convert_datesz&TestPandasContainer.test_convert_dates'  sW   z**224886


%%4
fb)))5	

T
22334u5557799#F+277=="5/0099
fh/// Ij))11$77?TUUU

%%4X...
vr*****r2   date_formatepochiso	as_objectdate_typc                     |ddd          t           j        g}|r|                    d           t          ||          }|                    |          }|dk    rd}nd	}|r|                    d
d          }||k    sJ d S )Ni  r   )yearmonthdayr   r   r  r  z+{"1577836800000":1577836800000,"null":null}zA{"2020-01-01T00:00:00.000":"2020-01-01T00:00:00.000","null":null}}	,"a":"a"})rD   r'  appendr
   r    replace)rK   r  r  r  rc   rT   r%   r"   s           r&   test_date_index_and_valuesz.TestPandasContainer.test_date_index_and_values?  s     d!333RV< 	KKT&&&55'!!DHH T   	:''[99H!!!!!!r2   
infer_word)
trade_timer  datetimesold_atr  	timestamp
timestampsc                 
   ddl m} dd|diddig}t          dt          d          gdt          j        ggd|g          }t          t           ||                              d|g         }t          j	        ||           d S )	Nr   dumpsidr   l    $Ar   z
2002-11-08r   )
pandas.io.jsonr  r   r   rD   r'  r   r   r   r!   )rK   r  r  rc   r"   r%   s         r&   test_convert_dates_inferz,TestPandasContainer.test_convert_dates_inferV  s     	)(((((q*m4tQi@<(()Arv;7$
AS
 
 
 8EE$KK001142DE
fh/////r2   zdate,date_unit))20130101 20:43:42.123N)20130101 20:43:42r~  )r  ms)z20130101 20:43:42.123456us)z20130101 20:43:42.123456789r  c                    |}t          |                              d          |d<   t          j        |j        d|j                            d          f<   t          j        |j        d|j                            d          f<   |r|                    d|          }n|                    d          }t          t          |                    }|
                                }t          j        ||           d S )Nr  r  r   r7   r  r  r  r  )r   r%  rD   r'  r{   r   r&  r    r   r   ra   r   r!   )rK   r  r  rZ   rY   r!  r%   r"   s           r&   test_date_format_framez*TestPandasContainer.test_date_format_framen  s     t__,,T226
132:%%f---.132:%%f---. 	1::%9:EEDD::%:00D8D>>**7799
fh/////r2   c                     |}d}t          j        t          |          5  |                    dd           d d d            d S # 1 swxY w Y   d S )N*Invalid value 'foo' for option 'date_unit'r   r  r:   r  )rm   rn   r   r    )rK   rZ   rY   r$   s       r&   test_date_format_frame_raisesz1TestPandasContainer.test_date_format_frame_raises  s    :]:S111 	; 	;JJ5EJ:::	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	;s   AA	A	c                    t          t          |                              d          |j                  }t          j        |j        d<   t          j        |j        d<   |r|                    d|          }n|                    d          }t          t          |          d	          }|
                                }t          j        ||           d S )
Nr  r   r   r7   r  r  r  rz  r  )r
   r   r%  rF   rD   r'  r{   r    r   r   ra   r   r|  )rK   r  r  rU   rQ   r!  r%   r"   s           r&   test_date_format_seriesz+TestPandasContainer.test_date_format_series  s     IdOO++D119NOOOV
V
 	1::%9:EEDD::%:00D8D>>x8887799
vx00000r2   c                     t          t          d          |j                  }d}t          j        t
          |          5  |                    dd           d d d            d S # 1 swxY w Y   d S )Nr  r   r  r   r  r:   r  )r
   r   rF   rm   rn   r   r    )rK   rU   rQ   r$   s       r&   test_date_format_series_raisesz2TestPandasContainer.test_date_format_series_raises  s    I566o>STTT:]:S111 	; 	;JJ5EJ:::	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	;s   A&&A*-A*unit)r~  r  r  r  c                    |}t          d                              d          |d<   |j                            d          }t          d          |j        d|f<   t          d          |j        d|f<   t
          j        |j        d|f<   |                    d	|
          }t          t          |          |          }t          j        ||           t          t          |          d           }t          j        ||           d S )Nr  r  r  z19710101 20:43:42r   z21460101 20:43:42r   r   r  r  r  )r   r%  r   r&  r{   rD   r'  r    r   r   r   r!   )rK   r  rZ   rY   dlr!  r%   s          r&   test_date_unitz"TestPandasContainer.test_date_unit  s    233;;DAA6
Z''"#6772"#67722zzgz>> 8D>>T:::
fb))) 8D>>T:::
fb)))))r2   )r~  r  r  c           	         t          t          j        d|          gd| d          }t          t	          t          j        d|          gd| d|          t	          t          j        d|          gt
          |          d          }t                      }|                    |d|	           |                    d
           t          j
        t          |ddg          |dd           d S )Nz2023-01-01T11:22:33.123456zdatetime64[]r@   z2022-01-01T11:22:33.123456rA   rF   )r  date_objr  r  r   r  r  r  F)r   check_dtype)r	   rI   
datetime64r   r
   r   r   r    seekr   r!   r   )rK   r  rF   rY   bufs        r&   test_iso_non_nano_datetimesz/TestPandasContainer.test_iso_non_nano_datetimes  s-    ]7>>?''''
 
 
 ]#?FFG////  
 #]#?FFG    
 
 jj


3ET
:::
 	c&*)=>>>"		
 	
 	
 	
 	
 	
r2   c                 B    d}t          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   )rK   r~  s     r&   test_weird_nested_jsonz*TestPandasContainer.test_weird_nested_json  s%    " 	(1++r2   c                    t          t          j                            d                              d          t          d                    }t          d          |d<   t          d          |d<   d	|d
<   t          j	        dd          |_
        t          |                                          }t          |t          j        t          j        d          }t!          j        ||           d S )Nr   )r7   r   ABr   ru   r  r7   intsTboolsr7  )r  r  r@   )r   rI   randomdefault_rngstandard_normalrG   r   r.   rD   r#  rF   r   r    r   r?   bool_r   r!   )rK   dfj2r!  r%   s       r&   test_doc_examplez$TestPandasContainer.test_doc_example  s    I!!!$$44V<<d4jj
 
 
 !,,VQxxVW]:q999
''428'L'LMMM
ff-----r2   c                     |dddd          }t          j        |          }|                                }t          t	          |                    }|                    |j        |j                  }d}t          j	        t          |          5  |                    t          j        d	          }d d d            n# 1 swxY w Y   t          j        ||           d S )
Nr   r!  rc   z	teams.csvr^   z.The 'downcast' keyword in fillna is deprecatedr   F)downcast)rD   read_csvr    r   r   reindexrF   r   r   r   r   fillnarI   r   r!   )rK   r+  r)  rY   r~  r%   resr$   s           r&   test_round_trip_exceptionz-TestPandasContainer.test_round_trip_exception  s    xffk::[JJLL8A;;''nn28RZn@@>'SAAA 	5 	5**RVe*44C	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5
c2&&&&&s   "B??CCzfield,dtype
created_atUTCtz	closed_at
updated_atc                     d}|                     |           t          |j        d          }||         j        |k    sJ d S )Nzw{"created_at": ["2023-06-23T18:21:36Z"], "closed_at": ["2023-06-23T18:21:36"], "updated_at": ["2023-06-23T18:21:36Z"]}
)contentTr  )serve_contentr   urlrA   )rK   fieldrA   
httpserverrc   r%   s         r&   test_urlzTestPandasContainer.test_url  sS     J   ...:>>>>e}"e++++++r2   c           	         d }t          t          d          t          d          g          }|j        dk    sJ t          t	          |                                          d                              |          }t          j        ||           t          t          d          t          d          gt          j
        dd	g          
          }|j        dk    sJ t          t	          |                                          d                              |          }t          j        ||           t          t          d          t          d          g          }|d         j        dk    sJ t          j        |t          t	          |                                                                        |                     d S )Nc                 .    t          j        | d          S )Nr  r  )rD   to_timedelta)rg   s    r&   <lambda>z4TestPandasContainer.test_timedelta.<locals>.<lambda>$  s    boad;;; r2      r7   secondsztimedelta64[ns]rz  r  r   r   r   )r
   r   rA   r   r   r    applyr   r|  rD   r   r   r!   )rK   	converterrT   r%   frames        r&   test_timedeltaz"TestPandasContainer.test_timedelta#  s   ;;	immYq%9%9%9:;;y-----8CKKMM22AAAGG	RR
vs+++immYq%9%9%9:"(Aq6BRBRSSSy-----8CKKMM22AAAGG	RR
vs+++9R==)A*>*>*>?@@Qx~!22222
9Xemmoo6677==iHH	
 	
 	
 	
 	
r2   c                    t          t          d          t          d          gddgt          j        dd          d	          }t	          |                    d
                    }t          |          }t          j        |j        d
          |d<   t          j	        |j
                  |d<   t          j        ||           d S )Nr  daysr7   r  r   r   ru   )startr7  r   r   r\   r  r  r  r   r\   )r   r   rD   r#  r   r    r   r  r   to_datetimer\   r   r!   )rK   r  rc   r%   s       r&   test_timedelta2z#TestPandasContainer.test_timedelta27  s    R((()A*>*>*>?V]Q??? 
 
 55664ofhT:::snVX..s
eV,,,,,r2   c                    t          d          }t          d          }t          d||git                    }t          dt	          j        |                              d          j        |                    d          j        gi          }t          |	                    d                    }t          |ddi          }t          j        ||d	           d S )
Nr  ru   r   r@   r  r  r?   Fr   )r   r   r   r   rD   	Timedeltar%  _valuer   r    r   r   r!   )rK   tdrQ   r  r"   rc   r%   s          r&   test_mixed_timedelta_datetimez1TestPandasContainer.test_mixed_timedelta_datetimeE  s    r]]z""3R/8882<##++D118"**T:J:J:QRS
 
 55664W~666
fhGGGGGGr2   timedelta_typc                     |d           |d          t           j        g}|r|                    d           t          ||          }|dk    rd}nd}|r|                    d	d
          }|                    |          }||k    sJ d S )Nr   r   r   r   r   r  zA{"P1DT0H0M0S":"P1DT0H0M0S","P2DT0H0M0S":"P2DT0H0M0S","null":null}z7{"86400000":86400000,"172800000":172800000,"null":null}r  r  r  )rD   r'  r  r
   r  r    )rK   r  r  r+  rc   rT   r"   r%   s           r&   test_timedelta_to_jsonz*TestPandasContainer.test_timedelta_to_jsonQ  s    
 1%%%}}!'<'<'<bfE 	KKT&&&%S H QH 	:''[99H55!!!!!!r2   c                     t                      }t          dd|gi          }t          ddt          |          gi          }t          t	          |                    t                                        }t          j        ||d           d S )Nr      rn  Fr   )r   r   r   r   r   r    r   r!   )rK   r   r  r"   r%   s        r&   test_default_handlerz(TestPandasContainer.test_default_handlerh  s~    3E
+,,cAs5zz?3448EMM#M$F$FGGHH
hGGGGGGr2   c           	          ddl m} d }dt          ddt          dd          gt	          d	          d d
gdddg          g}d} |||d          |k    sJ d S )Nr   r  c                 p    t          | t                    rdd| j        fd| j        fgS t	          |           S )N)mathjsComplexreim)
isinstancecomplexrealimagr   )objs    r&   defaultzBTestPandasContainer.test_default_handler_indirect.<locals>.defaultr  s;    #w'' S-ch/?$AQRRs88Or2   	   r   STRr   r   zN/Ar  r   r   r   zR[9,[[1,null],["STR",null],[[["mathjs","Complex"],["re",4.0],["im",-5.0]],"N\/A"]]]r*   ro  r0   )r  r  r   r8  r   )rK   r  r<  df_listr"   s        r&   test_default_handler_indirectz1TestPandasContainer.test_default_handler_indirecto  s    ((((((	 	 	 %B0edE7RSSc
  
1 	 uWghGGG8SSSSSSr2   c           	          t          ddt          dd          gt          d          d t          dd          gdd	d
g          }d}|                    t          d          |k    sJ d S )Nr   gffffff@r   r?  r   g333333?r   r  r   r   r   zE[["(1+0j)","(nan+0j)"],["(2.3+0j)","(nan+0j)"],["(4-5j)","(1.2+0j)"]]r*   r@  )r   r8  r   r    r   )rK   rY   r"   s      r&   ,test_default_handler_numpy_unsupported_dtypez@TestPandasContainer.test_default_handler_numpy_unsupported_dtype  s    c71b>>*%,,gcSToo1VWW#J
 
 

% 	
 zz#hz??8KKKKKKr2   c           
         dfd}t          j        t                    5  t          dddt	                      gi                              |           d d d            n# 1 swxY w Y   t          j        t                    5  t          dddt          dd	          gi                              |           d d d            d S # 1 swxY w Y   d S )
Nraisinc                 "    t                    rW   )	TypeError)r;  r$   s    r&   my_handler_raiseszJTestPandasContainer.test_default_handler_raises.<locals>.my_handler_raises  s    C.. r2   r   r   r   r   rn  r   r?  )rm   rn   rH  r   r   r    r8  )rK   rI  r$   s     @r&   test_default_handler_raisesz/TestPandasContainer.test_default_handler_raises  s|   	! 	! 	! 	! 	! ]9C000 	 	sQ688,-..66 1 7   	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 ]9C000 	 	sQ71b>>2344<< 1 =   	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	s#   5A%%A),A)7CCCc                 R   t          dg di          }|d         |d<   |                                }|d                             d          |d<   ||                                k    sJ |d         }|d         }|                                |                                k    sJ d S )Nr   )r   r   r\   r   r   r   r   r   category)r   r    r   )rK   rY   r"   r~  scs        r&   test_categoricalz$TestPandasContainer.test_categorical  s    @@@ABBS'3::<<S'..,,32::<<''''sGWyy{{bjjll******r2   c                    t          j        ddd          }|                    d                              d           }t	          |t          j        dd          d          }|                                }||d<   |                                }||                                k    sJ t          |          }t          |          }|                                |                                k    sJ d S )	Nru   rt   
US/Easternr7  r	  utcr  rm  r   )rD   r#  
tz_converttz_localizer   ra   r    r
   )rK   tz_rangetz_naiverY   df_naiver"   stzs_naives           r&   test_datetime_tzz$TestPandasContainer.test_datetime_tz  s    =Q<HHH&&u--99$??XBM*a,P,P,PQQRR7799 ##%%2::<<''''X""{{}} 1 1111111r2   c                 f   t          t          j                            d                              d                    }t          j        |j        d d<   |                    d          }|                                }||                                k    sJ t          t          j                            d                              d                    }t          j        |j        d d<   |                    d          }|                                }||                                k    sJ d S )Nr   )r  r      Sparser  )
r   rI   r  r  r  r   r   r   r    r
   )rK   rY   sdfr"   r~  sss         r&   test_sparsezTestPandasContainer.test_sparse  s    ry,,Q//??HHIIVrr
ii!!::<<3;;==((((29((++;;B??@@Fbqb	XXh99;;2::<<''''''r2   rQ   z2013-01-10 05:00:00Zz2013-01-10 00:00:00rP  z2013-01-10 00:00:00-0500c                     ddl m} d} ||d          |k    sJ |                                } ||d          |k    sJ d S )Nr   r  z"2013-01-10T05:00:00.000Z"T	iso_dates)r  r  to_pydatetime)rK   rQ   r  expdts        r&   test_tz_is_utcz"TestPandasContainer.test_tz_is_utc  sr     	)(((((*uR4(((C////uR4(((C//////r2   c                     ddl m} t          d          }d} ||d          |k    sJ |                                } ||d          |k    sJ d S )Nr   r  z2013-01-10 05:00:00z"2013-01-10T05:00:00.000"Trb  )r  r  r   rd  )rK   r  rQ   re  rf  s        r&   test_tz_is_naivez$TestPandasContainer.test_tz_is_naive  s~    ((((((,--)uR4(((C////uR4(((C//////r2   rU  z2013-01-01 05:00:00Zr   r  z2013-01-01 00:00:00rQ  z2013-01-01 00:00:00-0500c                 x   ddl m} d}d} ||d          |k    sJ t          |          } ||d          |k    sJ  ||                    t                    d          |k    sJ t          d|i          } ||d          }||k    sJ  ||                    dt          i          d          sJ 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"}}Trb  DT)r  r  r	   r   r   r   )rK   rU  r  re  dfexpr,  rY   r%   s           r&   test_tz_range_is_utcz(TestPandasContainer.test_tz_range_is_utc  s    	)(((((G/ 	 uX...#5555H%% uSD)))S0000uSZZ''4888C????c{##rT***uRYYf~..$????????r2   c                 `   ddl m} t          j        dd          }d}d} ||d	          |k    sJ  ||                    t
                    d	          |k    sJ t          d
|i          } ||d	          }||k    sJ  ||                    d
t
          i          d	          sJ d S )Nr   r  z2013-01-01 05:00:00r   r  z5["2013-01-01T05:00:00.000","2013-01-02T05:00:00.000"]zD{"DT":{"0":"2013-01-01T05:00:00.000","1":"2013-01-02T05:00:00.000"}}Trb  rk  )r  r  rD   r#  r   r   r   )rK   r  r,  re  rl  rY   r%   s          r&   test_tz_range_is_naivez*TestPandasContainer.test_tz_range_is_naive  s    ((((((m11===EV uSD)))S0000uSZZ''4888C????c{##rT***uRYYf~..$????????r2   c                     t          t          d          d          }t          ddgddggddg          }t          j        ||           d S )	Nr   Tr   r   r   r   r   r   r   r   r   r   r!   rw  s      r&   test_read_inline_jsonlz*TestPandasContainer.test_read_inline_jsonl  s\     8$IJJRVWWWq!fq!f-SzBBB
fh/////r2   c                     t          d|j         dd|          }t          ddgddggddg	          }t          j        ||           d S )
Nzs3n://z/items.jsonlT)r   storage_optionsr   r   r   r   r   )r   r>   r   r   r!   )rK   s3_public_bucket_with_datas3sor%   r"   s        r&   test_read_s3_jsonlz&TestPandasContainer.test_read_s3_jsonl  sn    
 B/4BBB 
 
 

 q!fq!f-SzBBB
fh/////r2   c                 \   t          j        d          5 }t          |dd          5 }|                    d           d d d            n# 1 swxY w Y   t	          |d          }t          dd	gdd	ggd
dg          }t          j        ||           d d d            d S # 1 swxY w Y   d S )Nztmp_items.jsonwzutf-8encodingr   Tr   r   r   r   r   r   )r   r  openwriter   r   r!   )rK   r)  infiler%   r"   s        r&   test_read_local_jsonlz)TestPandasContainer.test_read_local_jsonl*  sF   _-.. 	4$dC'222 DfBCCCD D D D D D D D D D D D D D Dt4000F 1a&1a&!1C:FFFH!&(333	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4s4   B!A
B!
A	B!A	AB!!B%(B%c                 B   d}t          |          }t          |d          }t          ddgddggddg	          }t          j        ||           t          d          }t          |d          }t          ddgddggddg	          }t          j        ||           d S )
Nu5   {"a": "foo”", "b": "bar"}
{"a": "foo", "b": "bar"}
Tr   u   foo”r8   r:   r   r   r   )r   r   r   r   r!   rK   r!  r%   r"   s       r&   test_read_jsonl_unicode_charsz1TestPandasContainer.test_read_jsonl_unicode_chars3  s    
 I~~4t,,,{E2UENCcSVZXXX
fh/// QRR4t,,,{E2UENCcSVZXXX
fh/////r2   bigNumc                 .   t          |t          dg          }|                                }dt          |          z   dz   }||k    sJ t	          |t          dgdg          }|                                }dt          |          z   dz   }||k    sJ d S )	N	articleIdr  {"articleId":r  r   rA   rF   r   {"0":{"articleId":r   )r
   r   r    r   r   )rK   r  rz  r!  r"   rY   s         r&   test_to_json_large_numbersz.TestPandasContainer.test_to_json_large_numbersD  s     f[MBBB~~"S[[036xvVK=1#NNNzz||'#f++5<xr2   l        l            c                    t          dt          |          z   dz             }d}t          j        t          |          5  t          |           d d d            n# 1 swxY w Y   t          dt          |          z   dz             }t          j        t          |          5  t          |           d d d            d S # 1 swxY w Y   d S )Nr  r  z#Value is too small|Value is too bigr   r  r   )r   r   rm   rn   r   r   )rK   r  r!  r$   s       r&   test_read_json_large_numbersz0TestPandasContainer.test_read_json_large_numbersQ  s?    #f++5;<<4]:S111 	 	dOOO	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 ,s6{{:TABB]:S111 	 	dOOO	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	s$    AA #A $CCCc                 *   d}t          |          }t          |d          }t          ddg          }t          j        ||           d}t          |          }t          |          }t          ddgdg	          }t          j        ||           d S )
Nz'{"articleId": "1404366058080022500245"}rz  r  gg_SDr  r   z.{"0": {"articleId": "1404366058080022500245"}}r   r^   )r   r   r
   r   r|  r   r!   r  s       r&   test_read_json_large_numbers2z1TestPandasContainer.test_read_json_large_numbers2]  s    8~~4X...+k];;;
vx000?~~4[sKKK
fh/////r2   c                 &   t          ddgddggddg          }|                    dd          }d	}||k    sJ t          d
dgddggddg          }|                    dd          }d}||k    sJ t          j        t	          t          |          d          |           t          ddgddggddg          }|                    dd          }d}||k    sJ t          j        t	          t          |          d          |           d S )Nr   r   r   r   r   r)   T)r0   r   z{"a":1,"b":2}
{"a":1,"b":2}
zfoo}r8   zfoo"z/{"a":"foo}","b":"bar"}
{"a":"foo\"","b":"bar"}
r   zfoo\za\r   )r   r    r   r!   r   r   r   s       r&   test_to_jsonlz!TestPandasContainer.test_to_jsonlk  sE   AA'#s<<<9D993!!!!&%9C:NNN9D99G!!!!
i(8(8EEErJJJ %(65/:UCLQQQ9D99R!!!!
i(8(8EEErJJJJJr2   z%GH#13774 encoding kwarg not supportedval)   E, 17r2      a   b   c)r  r  r  r  )s   EE, 17r2   r  r  r  )r     r  r  r  )r2   r  r  r  )r  r  r  r  )   Ar2   r  r  r  r2   r  r  r  rL  c                 2   t          d |D             |          }d}t          j        d          5 }|                    ||           t	          t          |          |          }t          j        ||d           d d d            d S # 1 swxY w Y   d S )Nc                 f    g | ].}t          |t                    r|                    d           n|/S )latin-1)r7  r_  decode)r   rg   s     r&   r   z;TestPandasContainer.test_latin_encoding.<locals>.<listcomp>  s8    MMMAJq%$8$8?QXXi   aMMMr2   r@   r  r  rz  F)check_categorical)r
   r   r  r    r   r   r|  )rK   rA   r  rT   r{  r)  retrs          r&   test_latin_encodingz'TestPandasContainer.test_latin_encoding  s    $ MMMMM
 
 
 _[)) 	GTKKxK000Xd^^h???D"3FFFF	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	Gs   ABBBc                     t          dt          d          gi          }|                    dd                                          }|                                 |                    dd                                          }||k    sJ d S )Nr   r   T)rF   deep)r   r   memory_usagesumr    )rK   rY   size_before
size_afters       r&   "test_data_frame_size_after_to_jsonz6TestPandasContainer.test_data_frame_size_after_to_json  s    c!ffX''ooDto<<@@BB


__4d_;;??AA
j((((((r2   r   r   r   r   1.2.c                     t          ddgddgg||          }|                    d          }t          t          |          d          }t	          j        ||           d S )Nr   r   rt   r   r^   tabler_   r   r    r   r   r   r!   )rK   rF   r   r"   dfjsonr%   s         r&   .test_from_json_to_json_table_index_and_columnszBTestPandasContainer.test_from_json_to_json_table_index_and_columns  sn     q!fq!f-UGLLL!!!118F++G<<<
fh/////r2   c                     t          ddgddgddgd          }|                    d	          }t          t          |          d	          }t	          j        ||           d S )
Nr   r   r   r   r   r   r#  r  r_   r  )rK   r"   r  r%   s       r&   #test_from_json_to_json_table_dtypesz7TestPandasContainer.test_from_json_to_json_table_dtypes  sn    Aq6c
#sLLMM!!!118F++G<<<
fh/////r2   )re   r)   rF   r   c                    t          j        t          g dd          t          g dd          t          g dd          t          g dd	          t          g d
d          t          g dd          d          }|                    |          }t	          t          |          |dddd	ddd          }t          j        ||           d S )Nr   r?   r@   )Nr   r   r  )N r\   r   TFTboolr   r   NrL  )z
2020-01-01Nz
2020-01-03ry   )IntegerFloatObjectBoolCategoryDatetimer_   r0   rA   )r   	from_dictr
   r    r   r   r   r!   )rK   r0   r"   r  r%   s        r&   %test_to_json_from_json_columns_dtypesz9TestPandasContainer.test_to_json_from_json_columns_dtypes  s    &!)))7;;; 0 0 0	BBB AAA222&AAA"#3#3#3:FFF"666>N  	 	
 
 !!!00V"""&, 
 
 
 	fh/////r2   )r   r\   c                     t          ddgddgddgd          }|                    d	          }d
}t          j        t          |          5  t          |d|           d d d            d S # 1 swxY w Y   d S )Nr   r   r   r   r   r   r#  r  r_   z)cannot pass both dtype and orient='table'r   r  r   r    rm   rn   r   r   )rK   rA   rY   r  r$   s        r&   !test_read_json_table_dtype_raisesz5TestPandasContainer.test_read_json_table_dtype_raises  s     aV3*C:FFGG7++9]:S111 	; 	;fWE::::	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	;s   A--A14A1)rF   r   r)   r*   c                     t                      }t          t          d          |d          }t          j        |j        |j                   t          j        |j        |j                   d S )Nz{}Tr   )r   r   r   r   assert_index_equalrF   r   )rK   r0   r"   r%   s       r&   %test_read_json_table_empty_axes_dtypez9TestPandasContainer.test_read_json_table_empty_axes_dtype  s\     ;;8D>>&tLLL
flHN;;;
fnh.>?????r2   c                     t          ddgddggddgddg	          }|                    d
          }d}t          j        t          |          5  t          |d
d           d d d            d S # 1 swxY w Y   d S )Nr   r   rt   r   r   r   r  r  r^   r  r_   z0cannot pass both convert_axes and orient='table'r   Tr   r  )rK   rY   r  r$   s       r&   (test_read_json_table_convert_axes_raisesz<TestPandasContainer.test_read_json_table_convert_axes_raises  s    AA'SzD$<PPP7++@]:S111 	A 	AfW4@@@@	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	As   A00A47A4zdata, expectedr   r7   r   )r   rc   r:   r  r   r   r=   )r>   rc   )r>   rF   c                 l    |                     dd          }t          j        |          }||k    sJ d S )Nre   Fr0   rF   )r    r!  loadsrK   rc   r"   r%   s       r&   test_index_false_to_json_splitz2TestPandasContainer.test_index_false_to_json_split  s@    @ WE::F##!!!!!!r2   c                     |                     dd          }t          j        |          }t          j        j                            |d          t          |                              d          d}||k    sJ d S )Nr  Fr  r   r)   r_   )schemarc   )r    r!  r  rD   r   build_table_schemar   to_dictrK   rc   r%   r"   s       r&   test_index_false_to_json_tablez2TestPandasContainer.test_index_false_to_json_table  s    & WE::F## ej33D3FFdOO++9+==
 

 !!!!!!r2   c                     t          ddgddggddg          }d}t          j        t          |	          5  |                    |d
           d d d            d S # 1 swxY w Y   d S )Nr   r   r   r7   r   r   r   zU'index=False' is only valid when 'orient' is 'split', 'table', 'records', or 'values'r   Fr  rl   ro   s       r&   test_index_false_error_to_jsonz2TestPandasContainer.test_index_false_error_to_json5  s    
 AA'#s<<<. 	 ]:S111 	3 	3JJfEJ222	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3   AA #A r)   c                     t          ddgddggddg          }d}t          j        t          |	          5  |                    |d
           d d d            d S # 1 swxY w Y   d S )Nr   r   r   r7   r   r   r   zS'index=True' is only valid when 'orient' is 'split', 'table', 'index', or 'columns'r   Tr  rl   ro   s       r&   test_index_true_error_to_jsonz1TestPandasContainer.test_index_true_error_to_jsonC  s    
 AA'#s<<<- 	 ]:S111 	2 	2JJfDJ111	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2r  r  c                     t          ddgddgd          }|                    ||          }t          t          |          |          }t	          j        ||           d S )Nr   r   rt   r   r  r  r_   r  )rK   r0   rF   r"   r  r%   s         r&   "test_index_false_from_json_to_jsonz6TestPandasContainer.test_index_false_from_json_to_jsonQ  sl    
 Aq6A7788!!u!==8F++F;;;
fh/////r2   c                     t          t          d          dd          }t          dgt          dgd          	          }t	          j        ||           d S )
Nz{"2019-01-01T11:00:00.000Z":88}rz  rF   r  X   z2019-01-01 11:00:00r  r  r   )r   r   r
   r	   r   r|  rw  s      r&   test_read_timezone_informationz2TestPandasContainer.test_read_timezone_information[  sg    677Xg
 
 
 2$m5J4KPU&V&V&VWWW
vx00000r2   r  )zs3://example-fsspec/zgcs://another-fsspec/file.jsonzhttps://example-site.com/datazsome-protocol://data.txtc                     t          t          d| d                    }t          d|gi          }t          j        ||           d S )Nz{"url":{"0":"rl  r  rq  )rK   r  r%   r"   s       r&   test_read_json_with_url_valuez1TestPandasContainer.test_read_json_with_url_valuec  sQ     8$@c$@$@$@AABBecU^,,
fh/////r2   compression)r  z.gzz.bz2z.tarc                     d d| }t          j        t          d| d          5  t          |           d d d            d S # 1 swxY w Y   d S )N  aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaz.jsonzFile z does not existr   )rm   rn   FileNotFoundErrorr   )rK   r  long_json_paths      r&   'test_read_json_with_very_long_file_pathz;TestPandasContainer.test_read_json_with_very_long_file_pathr  s     '::[::]%L^%L%L%L
 
 
 	& 	&
 n%%%	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	&s   AAAzdate_format,key)r  i \&)r  
P1DT0H0M0Sc                     t          dggt          j        d          g          }d| d}|                    |          }||k    sJ d S )Nr   1Dr   rv  z
":{"0":1}}r  )r   rD   r'  r    )rK   r  keyrY   r"   r%   s         r&   test_timedelta_as_labelz+TestPandasContainer.test_timedelta_as_label  sa     ur|D'9'9&:;;;++++44!!!!!!r2   zorient,expected)rF   z{"('a', 'b')":{"('c', 'd')":1}})r   z{"('c', 'd')":{"('a', 'b')":1}}r  z,Produces JSON but not in a consistent manner)marksc                 n    t          dggdgdg          }|                    |          }||k    sJ d S )Nr   r  )r\   r]   r^   r_   r   rK   r0   r"   rY   r%   s        r&   test_tuple_labelsz%TestPandasContainer.test_tuple_labels  sJ    0 uZL:,GGG6**!!!!!!r2   indent)r   r   r   c                     t          ddgddggddg          }|                    |          }d	|z  }d
| d| | d| | d| d| d| | d| | d| d}||k    sJ d S )Nr:   r8   r9   r  r   r   r   r   z{
z"a":{
z"0":"foo",
z
"1":"baz"
z},
z"b":{
z"0":"bar",
z
"1":"qux"
z}
}r   )rK   r  rY   r%   spacesr"   s         r&   test_to_json_indentz'TestPandasContainer.test_to_json_indent  s    7#sLLL6**v		 				 	 	 
	 	 		 	
 	 	 	 
	 	 	 
	 	 	 	 	 !!!!!!r2   ))re   z{
    "columns":[
        "a",
        "b"
    ],
    "index":[
        0,
        1
    ],
    "data":[
        [
            "foo",
            "bar"
        ],
        [
            "baz",
            "qux"
        ]
    ]
})r)   zf[
    {
        "a":"foo",
        "b":"bar"
    },
    {
        "a":"baz",
        "b":"qux"
    }
])rF   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  aa  {
    "schema":{
        "fields":[
            {
                "name":"index",
                "type":"integer"
            },
            {
                "name":"a",
                "type":"string"
            },
            {
                "name":"b",
                "type":"string"
            }
        ],
        "primaryKey":[
            "index"
        ],
        "pandas_version":"1.4.0"
    },
    "data":[
        {
            "index":0,
            "a":"foo",
            "b":"bar"
        },
        {
            "index":1,
            "a":"baz",
            "b":"qux"
        }
    ]
}c                 v    t          ddgddggddg          }|                    |d	          }||k    sJ d S )
Nr:   r8   r9   r  r   r   r   r   )r0   r  r   r  s        r&   test_json_indent_all_orientsz0TestPandasContainer.test_json_indent_all_orients  sT    n 7#sLLL6!44!!!!!!r2   c                     t          j        t          d          5  t                                          d           d d d            d S # 1 swxY w Y   d S )Nzmust be a nonnegative integerr   r  )rm   rn   r   r   r    rd  s    r&    test_json_negative_indent_raisesz4TestPandasContainer.test_json_negative_indent_raises5  s    ]:-LMMM 	+ 	+KKr***	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+s   #AAAc           	          t          d          }t          |          }t          dd dt          j        dt          j         dg          }t          j        ||           d S )Nz?["a", NaN, "NaN", Infinity, "Infinity", -Infinity, "-Infinity"]r   NaNInfinityz	-Infinity)r   r   r   rI   r   r   r!   r  s       r&   test_emca_262_nan_inf_supportz1TestPandasContainer.test_emca_262_nan_inf_support9  sc    M
 
 4$rvzBF7KH
 
 	fh/////r2   c                     t          j        ddiddig          }t          dddgi          }t          t	          |                    }t          j        ||           d S )Ncol31900441201190696999Text)r!  r  r   r   r   r   r!   )rK   encoded_jsonr"   r%   s       r&   test_frame_int_overflowz+TestPandasContainer.test_frame_int_overflowD  sf    zE+A#BUFO"TUUe&<f%EFGG8L1122
fh/////r2   zdataframe,expectedr#  rg   rh   zU{"(0, 'x')":1,"(0, 'y')":"a","(1, 'x')":2,"(1, 'y')":"b","(2, 'x')":3,"(2, 'y')":"c"}c                 n    |                     d          }|                    d          }||k    sJ d S )NTfuture_stackrF   r_   )stackr    )rK   	dataframer"   rz  r%   s        r&   test_json_multiindexz(TestPandasContainer.test_json_multiindexK  sA     d33w//!!!!!!r2   c                 "   |j         d}}t          g dg dd          }|                    d| d| |           d}	 |d
 |j                                        D             v rd S t          j        d           |dz  }|dk    s
J d            Q)Nr  r   )r   r   r   r  zs3:///)rt  r7   Tc              3   $   K   | ]}|j         V  d S rW   )r  )r   r;  s     r&   	<genexpr>z1TestPandasContainer.test_to_s3.<locals>.<genexpr>b  s$      QQ3swQQQQQQr2   r  r   z,Timed out waiting for file to appear on moto)r>   r   r    objectsalltimesleep)rK   s3_public_bucketrv  mock_bucket_nametarget_filerY   timeouts          r&   
test_to_s3zTestPandasContainer.test_to_s3Z  s     )9(={+YYYYYY7788


;+;;k;;T
RRR	OQQ2B2J2N2N2P2PQQQQQJsOOOsNGQ;;; N;;;	Or2   c                     t          |t                    r:t          j                            d          }|j                            |           t          |gg                                          }|dk    sJ d S )Nznot implementedr   z{"0":{"0":null}})	r7  r   rm   r   r   r   r   r   r    )rK   nulls_fixturer   r   r%   s        r&   test_json_pandas_nullsz*TestPandasContainer.test_json_pandas_nullsh  sy    mW-- 	*;$$,=$>>DL##D)))]O,--5577+++++++r2   c                     t          t          d          d          }t          g d          }t          j        ||           d S )Nz[true, true, false]rz  r  )TTF)r   r   r
   r   r|  rw  s      r&   test_readjson_bool_seriesz-TestPandasContainer.test_readjson_bool_seriesq  sJ    8$9::III---..
vx00000r2   c                     t          dt          j        dd          ddg                              d          }|                                }d}||k    sJ d S )	NTz
2017-01-20z
2017-01-23r:   r8   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   rD   r#  r  r    r   s       r&   test_to_json_multiindex_escapez2TestPandasContainer.test_to_json_multiindex_escapew  sv    -l;;EN
 
 
 %T%
"
"	 	
 B 	 !!!!!!r2   c                      G d d          }t           |dddd          g          }t          j        |                                          ddddd	ik    sJ d S )
Nc                       e Zd ZddZd ZdS )GTestPandasContainer.test_to_json_series_of_objects.<locals>._TestObjectr\  Nc                 >    || _         || _        || _        || _        d S rW   r   r   _cr]   )rK   r   r   r  r]   s        r&   rb  zPTestPandasContainer.test_to_json_series_of_objects.<locals>._TestObject.__init__  s"    r2   c                     dS )Nr7   r   rd  s    r&   r   zITestPandasContainer.test_to_json_series_of_objects.<locals>._TestObject.e  s    qr2   rf  )rg  rh  ri  rb  r   r   r2   r&   _TestObjectr    s7               r2   r!  r   r   rt   r   r  0)r   r   r]   )r
   r!  r  r    )rK   r!  rz  s      r&   test_to_json_series_of_objectsz2TestPandasContainer.test_to_json_series_of_objects  s    	 	 	 	 	 	 	 	 qA!q999:;;z&..**++1115M5M/NNNNNNNr2   zdata,expectedy             @y              ?y      "@      )r   r   r   zW{"0":{"imag":8.0,"real":-6.0},"1":{"imag":1.0,"real":0.0},"2":{"imag":-5.0,"real":9.0}}yHz"Q?y@p=
ף"@yQ@(\ſz]{"0":{"imag":0.66,"real":-9.39},"1":{"imag":9.32,"real":3.95},"2":{"imag":-0.17,"real":4.03}}y             @y              y      @      y              $z{"0":{"0":{"imag":3.0,"real":-2.0},"1":{"imag":-3.0,"real":4.0}},"1":{"0":{"imag":0.0,"real":-1.0},"1":{"imag":-10.0,"real":0.0}}}yQѿ(\?yHzG(\ؿy=
ףp=?(\տy(\迚z{"0":{"0":{"imag":0.34,"real":-0.28},"1":{"imag":-0.34,"real":0.41}},"1":{"0":{"imag":-0.39,"real":-1.08},"1":{"imag":-1.35,"real":-0.78}}}c                 >    |                                 }||k    sJ d S rW   r   r  s       r&   test_complex_data_tojsonz,TestPandasContainer.test_complex_data_tojson  s)    F !!!!!!r2   c                 n    d}t          dddgi          }|                    d          }||k    sJ d S )NzY{"columns":["col1"],"index":[0,1],"data":[[13342205958987758245],[12388075603347835679]]}col1l   :!c&HI l   v8$Z_
 )rc   re   r_   r   )rK   r"   rY   r%   s       r&   test_json_uint64z$TestPandasContainer.test_json_uint64  sU    F 	 V&:<P%QRSSS7++!!!!!!r2   )re   r)   r*   rF   r   c                 \  	
 t          j        d          t          t          dt          j        dgd          t          g dd          t          dt          j        dgd	          t          g d
d	          g dg dg dg dd          }|dk    ret          t	          j        g dt          j                            }t          t	          j        ddt          gt          j                            }nHt                              g d                    }t                              g d                    }|                    |          }t          j        d|          5  t          t          |          ||          }d d d            n# 1 swxY w Y   t          t          dt          j        dgd          t          g dd          t          dt          j        dgd	          t          g d
d	          t          ddt          gd          t          g dd          ||d          
|dk    r(ddlm	 t          	
fd
j        D                       
|dk    r"t'          t)          dd                    
_        t+          j        |
           d S )Npyarrowr   rt   Int64r@   r   rr   rs   Float64)rr   r   rs   )TFNr  r#  r  )r   r   r\   r]   r   r  ghpythonr   r   r_   mode.string_storage)dtype_backendr0   TFbooleanr   ArrowExtensionArrayc           	      ^    i | ])}|                      |         d                     *S )Tfrom_pandas)array)r   r  r4  r"   pas     r&   
<dictcomp>zDTestPandasContainer.test_read_json_dtype_backend.<locals>.<dictcomp>  sL        ,,RXXhsmQUX-V-VWW  r2   r*   r\  )rm   importorskipr   r
   rI   r   r   r8  object_r   r   r    rD   option_contextr   r   pandas.arraysr4  r   rG   r.   r   r!   )rK   string_storager1  r0   rY   string_arraystring_array_naoutr%   r4  r"   r9  s            @@@r&   test_read_json_dtype_backendz0TestPandasContainer.test_read_json_dtype_backend  s#   
  ++QN':::IIIW555S"&#.i@@@OOO9===(((((($__%%%	 	
 
 X%%&rxrz'R'R'RSSL)"(Cb>*T*T*TUUOO ,BHH___,E,EFFL.rxx8H8H8H/I/IJJOjjj''4nEE 	 	]6  F	 	 	 	 	 	 	 	 	 	 	 	 	 	 	
 QN':::IIIW555S"&#.i@@@OOO9===T5"-Y???///yAAA!$	 	
 
 I%%999999      '/   H X#E!QKK00H
fh/////s   0 FF #F )re   r)   rF   c                    t          j        d          }t          dt          j        dgd          }|                    |          }t          j        d|          5  t          t          |          ||d	          }d d d            n# 1 swxY w Y   t          dt          j        dgd          }|dk    r3d
dl
m}	 t           |	|                    |d                              }t          j        ||           d S )Nr*  r   rt   r+  r@   r_   r0  rz  )r1  r0   r{  r   r3  Tr6  )rm   r;  r
   rI   r   r    rD   r=  r   r   r>  r4  r8  r   r|  )
rK   r?  r1  r0   r9  rT   rB  r%   r"   r4  s
             r&   test_read_json_nullable_seriesz2TestPandasContainer.test_read_json_nullable_series  sI     ++a^7333kkk((4nEE 	 	]6x  F	 	 	 	 	 	 	 	 	 	 	 	 	 	 	
 1bfa.888I%%99999911"((8QU(2V2VWWXXH
vx00000s   !BBBc                     d}t          j        t          |          5  t          dd           d d d            d S # 1 swxY w Y   d S )NzPdtype_backend numpy is invalid, only 'numpy_nullable' and 'pyarrow' are allowed.r   r   numpy)r1  )rm   rn   r   r   )rK   r$   s     r&   test_invalid_dtype_backendz.TestPandasContainer.test_invalid_dtype_backend  s    % 	 ]:S111 	5 	5fG4444	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5s   =AA)rg  rh  ri  rm   fixturerO   rU   rZ   rd   r   parametrizeri   rp   r   r}   r   r   r   r   rI   r?   r   r  intr   r   r   r   r   r   AssertionErrorr   r*  r   r   r   r   r   skipifr   r   r   r   r  r  r  r3  rX  rt  rx  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  network
single_cpurD   DatetimeTZDtyper  r  r%  r*  r'  r   r-  r0  rB  rD  rJ  rN  rZ  r`  rg  ri  r#  rm  ro  rr  r)  skip_if_not_us_localerw  r  r  sysmaxsizer  r  r  r  rH  r   r   r  r  r  r  r  r  r  r  r   rename_axisr  r  r  r  r  r  r  r  r  paramr  r  r  r  r  r  r  r  r  r  r  r#  r%  r(  rC  rE  rH  r   r2   r&   r4   r4   U   s       ^	 	 ^	 ^  ^ ^  ^
> 
> 
> [X'E'E'EFF> > GF> [X';<<& & =<& [X':;;[3Z#s$3Z#s$X3x i
##S)IIj,A,A3+GH		
 0 0  <;0( [X'F'F'FGG& & HG&N N N [Wuen55[^dE];;> > <; 65> [Wubh&788[^dE];;> > <; 98> [WtRZd&CDD[^dE];;> > <; ED>< [^dE];;> > <;>& [^dE];;0 0 <;0 [^dE];;> > <;>  [^dE];;> > <;>& [J    [fu~.fu~.fu~.	
 * * 	 * [D; V 
 
/	
! !D5 5E! !D5 [WtUm44[^dE];;+ + <; 54+( [WtUm440 0 540 [URVbfW$566[WtUm44+ + 54 76+ [D)OPP[&	
 	
 	

 
B B
 
 QPB
) ) )B B B" " "

 

 

+ + +<4 4 4<^
 ^
 ^
@$
 $
 $
L" " "C C C$P P P
1 
1 
1 [Wudm44
1 
1 54
1	1 	1 	1
1 
1 
1 [Wrz3&788	1 	1 98	1( ( (B B B
1 1 1 [66<.0@AAABFFL>**+	
 1 1 1* * *
* * *
* * *     
# 
# 
#+ + +0 []We,<==[[4-88[Z(-9JI)VWW" " XW 98 >="( [	
 	
 	
 
0 
0 
0 [	
 	
 	
	 	0 0	 	0; ; ; [	
 	
 	
	 	1 1	 	1; ; ; [V%<%<%<==* * >=*$ [V%6%6%677$
 $
 87$
L  *. . .' ' ' [[[-2-7778*+-2-7778	
 , ,   ,
 
 
(- - -
H 
H 
H [[4-88[]UG,<==[_r|Y.GHH" " IH >= 98"(H H HT T T*L L L  + + +2 2 2 ( ( (  [I,--I+===I011	
 0 0 00 0 0 [BM0!<<<BM/|LLLBM4a@@@	
 @ @ @*@ @ @"0 0 0 [	0 	0  	04 4 40 0 0" [Xa3;?9K'LMM
  
  NM
  [Xe'<==	 	 >=	0 0 0K K K, [EiXX[111,,,...999###+++111VS$%263d3
	
  [Wz6&:;;
G 
G <;  YX 
G) ) ) [$Ac
S#Jc
T4LQ  [Y#sc3Z$(NOO0 0 PO 00 0 0 [X'O'O'OPP0 0 QP0< [Wt3S-A-A&BCC; ; DC; [X'P'P'PQQ@ @ RQ@A A A [ 	Aq6Aq6*S#J??? #J!Q!Q0@AA
 	Aq6Aq6*S#J???KKERR #J!Q!Q0@AA
 	VaV$sCj#scSVZ@X   !#J!Q!Q0@AA	 VIIIC(((3			*J*JKyyys+++77>>iii00
 1vCc
S#J/GHHHq!f--)	
 8" "9 8" [YAA'#s<<<YAA'#s<<<HHOO	VaV$sCj#scSVZ@X   VIIIC(((VIIIC(((44U;;VQFc3Z#s,DEEE	
 " " " [X';<<3 3 =<3 [X	8'<==2 2 >=2 [X'9::[WtUm440 0 54 ;:01 1 1 [	
 	
 	
 0 0 0 [### & &	 & [/1FG " " " [<>FLk''I (     FLk''I (    	
 ," "- ," [Xyyy11" " 21"& [q	
 q	
 q	
t tj" "kt tj"+ + +	0 	0 	00 0 0 [ 				@@AAD	
	 	" "	 	"
 [O O O, , ,1 1 1" " "(O O O [ 7v&99::0 =\lKKLL2 	GW-/ABCC2 	#]3lM5RS 4	)	
   B" "C   B"
" " " [DDD 80 80 80t [X'D'D'DEE1 1 FE1(5 5 5 5 5r2   r4   c                      t          t          d                    } |                                 }t          j        t
          d          5  t          |d           d d d            d S # 1 swxY w Y   d S )Nr   zThe engine type foor   r:   )enginer
   r.   r    rm   rn   r   r   rT   rB  s     r&   test_invalid_enginerZ  %  s    
q

C
++--C	z)>	?	?	? % %#e$$$$% % % % % % % % % % % % % % % % % %s   A++A/2A/c                      t          t          d                    } |                                 }t          j        t
          d          5  t          |dd           d d d            d S # 1 swxY w Y   d S )Nr   z&currently pyarrow engine only supportsr   r*  F)rW  r   rX  rY  s     r&   test_pyarrow_engine_lines_falser\  -  s    
q

C
++--C	z)Q	R	R	R 6 6#iu55556 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s   A,,A03A0c           	         t          j        d           t          ddgddggddgdd	g
          }|                                }t	          j        dd          5  t          t          |                    }d d d            n# 1 swxY w Y   t          ddgddggdt	          j        ddgd          t	          j        dd	gd                    }t          j
        ||           d S )Nr*  r   r   r\   r]   zrow 1zrow 2zcol 1zcol 2r^   zfuture.infer_stringTzstring[pyarrow_numpy]r@   r  )rm   r;  r   r    rD   r=  r   r   r   r   r!   )r0   rY   rB  r%   r"   s        r&   $test_json_roundtrip_string_inferencer^  5  sB   
	"""	
sc3Z '(:WgDV
 
 
B **,,C		0$	7	7 * *8C==))* * * * * * * * * * * * * * *
sc3Z %h)1HIII'7+3JKKK	  H &(+++++s   BB
B)'r  r   decimalr   r   r   r!  r(  rR  r  rG  rI   rm   pandas.compatr   pandas.util._test_decoratorsutil_test_decoratorsr)  pandasrD   r   r   r	   r
   r   r   pandas._testing_testingr   pandas.core.arraysr   r   r'   r1   r4   rZ  r\  r^  r   r2   r&   <module>rh     s                       				 



             ) ) ) ) ) ) ) ) )                                *0 *0 *0Z, , ,M5 M5 M5 M5 M5 M5 M5 M5`>% % %6 6 6, , , , ,r2   