+
    nDj                         ^ RI t  ^ RIt^ RIt^ RIHt R tR t	RR lt
RR ltR tR tR	 tRR
 ltRR ltRR ltR tRR ltR#   ] d    Rt LGi ; i)    N)utilc                     ^ RI Hp ^ RIHp VP	                  V 4      '       d    \
        P                  P                  V 4      p VP                  ! W3/ VB  V # )r   N)_pandas_api)	pyarrow.parquetparquetpyarrow.pandas_compatr   is_data_framepaTablefrom_pandaswrite_table)tablepathkwargspqr   s   &&,  V/data/cameron/venvs/s3viz/lib/python3.14/site-packages/pyarrow/tests/parquet/common.py_write_tabler      sC     1  ''$$U+NN5)&)L    c                  Z    ^ RI Hp VP                  ! V / VB pVP                  RR7       V# )r   NT)full)r   r   
read_tablevalidate)argsr   r   r   s   *,  r   _read_tabler   )   s+     MM4*6*E	NNNLr   c                     T;'       g    / pT;'       g    / p\         P                  ! 4       p\        W3/ VB  \         P                  ! VP	                  4       4      p\        V3/ VB # N)r
   BufferOutputStreamr   BufferReadergetvaluer   )r   read_table_kwargswrite_table_kwargswriterreaders   &&&  r   _roundtrip_tabler$   1   s`    )//R+11r""$F5"45__V__./Fv3!233r   c                 6   Vf   T pT;'       g    / p\        WVR7      pVP                  VP                  8X  g   Q hVP                  V4      '       g   Q h\        WBVR7      pVP                  VP                  8X  g   Q hVP                  V4      '       g   Q hR # )N)r    r!   )r$   schemaequals)r   expectedr    r!   results   &&&, r   _check_roundtripr*   <   s    )//R e1CEF==HOO+++==""""f1CEF==HOO+++==""""r   c                 x    \         P                  P                  V 4      p\        W!R 7      pVP	                  4       # ))r!   )r
   r   r   r$   	to_pandas)dfwrite_kwargsr   r)   s   &&  r   _roundtrip_pandas_dataframer/   N   s2    HH  $E0Fr   c                    \         P                  ! R 4      p\         P                  ! V4      p\         P                  P                  \	        VP
                  VP
                  4      \        VP                  VP                  4      WR7      # )int_sizedtype)npiinforandomrandintmaxmin)r3   r4   platform_int_infor6   s   &&  r   _random_integersr<   U   sf    (HHUOE99S,=,A,AB ,=,A,AB"&  5 5r   c                 X    \         P                  ! \        P                  ! WR 7      4      # )r4   )r
   arrayr5   aranger2   s   &&r   _range_integersrA   ^   s    88BIId011r   c                 |   \         P                  P                  V4       R \        V \         P                  4      R\        V \         P
                  4      R\        V \         P                  4      R\        V \         P                  4      R\        V \         P                  4      R\        V \         P                  4      R\        V \         P                  4      R\        V \         P                  4      R\         P                  P                  V 4      P                  \         P                  4      R	\         P                  ! V \         P                   R
7      R\         P                  P                  V 4      ^ 8  R\#        V 4       Uu. uF  p\$        P&                  ! ^
4      NK  	  upRR.V ,          RR.V ,          /# u upi )uint8uint16uint32uint64int8int16int32int64float32float64r>   boolstringsall_noneNall_none_category)r5   r7   seedr<   rC   rD   rE   rF   rG   rH   rI   rJ   randnastyperK   r@   rL   ranger   rands)r3   rQ   is   && r   
_test_dictrW   b   s+   IINN4!$1"43"43"43 rww/!$1!$1!$1299??4(//

;299T4		%)E$K8KqDJJrNK8TFTMdVd]  9s   F92c                 B    ^ RI pVP                  \        W4      4      pV# )r   N)pandas	DataFramerW   )r3   rQ   pdr-   s   &&  r   _test_dataframer\   v   s    	j,	-B Ir   c                 R    \         P                  P                  \        W4      4      # r   )r
   r   from_pydictrW   )r3   rQ   s   &&r   _test_tabler_      s    88
4 677r   c                    ^ RI Hp \        V \        P                  4      '       d   T pM\        P                  P                  V 4      p\        P                  ! 4       p\        W#RRR7       VP                  ^ 4       VP                  V4      # )r   NSNAPPYz2.6)compressionversion)r   r   
isinstancer
   r   r   ioBytesIOr   seekParquetFile)table_or_dfr   a_tablebufs   &   r   make_sample_filerl      s`     +rxx((((&&{3
**,C8UCHHQK>>#r   c                    ^ RI p\        P                  P                  V4       / R\        P                  ! V \        P
                  R7      bR\        P                  ! V \        P                  R7      bR\        P                  ! V \        P                  R7      bR\        P                  ! V \        P                  R7      bR\        P                  ! V \        P                  R7      bR\        P                  ! V \        P                  R7      bR	\        P                  ! V \        P                  R7      bR
\        P                  ! V \        P                  R7      bR\        P                  ! V \        P                  R7      bR\        P                  ! V \        P                  R7      bR\        P                  ! V \        P                  R7      bR\        P                  P                  V 4      ^ 8  bR\        P                  ! RV RR7      bR\        P                  ! RV RR7      bR\        P                  ! RV RR7      bR\        P                  ! ^ V RR7      bRTP!                  \#        V 4       Uu. uF  p\%        V4      NK  	  up4      bRR.V ,          RR.\#        V ^,
          4       Uu. uF  p\%        V4      NK  	  up,           R.,           RR.V ,          RR.^,          \#        V ^,
          4       Uu. uF  pR.V^,          ,          NK  	  up,           /CpV'       d   VR,          P'                  R 4      VR!&   VP)                  V4      # u upi u upi u upi )"r   NrC   r>   rD   rE   rF   rG   rH   rI   rJ   float16rK   rL   rM   datetime_msz2016-01-01T00:00:00.001zdatetime64[ms]datetime_usz2016-01-01T00:00:00.000001zdatetime64[us]datetime_nsz2016-01-01T00:00:00.000000001zdatetime64[ns]	timedeltaztimedelta64[s]str	empty_str str_with_nullsnull	null_listcategorystr_category)rY   r5   r7   rQ   r@   rC   rD   rE   rF   rH   rI   rJ   rn   rK   rL   rR   SeriesrT   rs   rS   rZ   )r3   rQ   categoricalr[   xarrayss   &&&   r   alltypes_sampler      s   IINN44rxx0"))D		2 	"))D		2 	"))D		2	
 			$bhh/ 	4rxx0 	4rxx0 	4rxx0 	299T4 	299T4 	299T4 			%) 	ryy!:D'79 	ryy!=t'79" 	ryy!@$'79#& 	RYYq$.>?'( 	ryy%+6+Q#a&+67)* 	bTD[4&E$(O#DOqCFO#DDvMdVaZU4!8_"M_D6QU#3#3_"MM1F4 !'!5!5j!A~<< 7#D"Ms   6M2M>M!)NN)'  r   )r   r   F)re   numpyr5   ImportErrorpyarrowr
   pyarrow.testsr   r   r   r$   r*   r/   r<   rA   rW   r\   r_   rl   r    r   r   <module>r      sm   $ 
  	4#$52(8  {  	Bs   A 	AA