o
    5cZ                  	   @   s  d dl Z d dlmZ d dlZd dlmZ d dlZd dlZd dl	m
Z
mZmZmZmZmZmZ d dlmZmZ d dlmZmZ ejjZdd Zd	d
 Zdd Zdd Zdd Zdd Z ej!deddddde
e"dgdd Z#dd Z$dd Z%d d! Z&ej!d"d#d$g d%d&gd'd( Z'dS ))    N)BytesIO)catch_warnings)CategoricalIndex	DataFrameHDFStore
MultiIndex_testing
date_rangeread_hdf)ensure_clean_pathensure_clean_store)Term_maybe_adjust_namec              	   C   s   t  }t| Y}|d| d}tjt|d |jddgd W d    n1 s+w   Y  d}tjt|d |jddgd W d    n1 sLw   Y  W d    d S W d    d S 1 sdw   Y  d S )	Ndfzqcannot pass a column specification when reading a Fixed format store. this store must be selected in its entiretymatchA)columnszucannot pass a where specification when reading from a Fixed format store. this store must be selected in its entiretyz	columns=Awhere)tmmakeDataFramer   putpytestraises	TypeErrorselect)
setup_pathr   storemsg r    Z/var/www/html/gps/gps/lib/python3.10/site-packages/pandas/tests/io/pytables/test_errors.pytest_pass_spec_to_storer   s    
"r"   c              	   C   s   t dg di}t dg ditdddd}t| >}|jd|d	d
 td}tjt|d |jd|d	dd W d    n1 sCw   Y  W d    d S W d    d S 1 s[w   Y  d S )Na)         )         z1/1/2000r&   periods)indexframetableformatz/incompatible kind in col [integer - datetime64]r   T)r0   append)	r   r	   r   r   reescaper   r   r   )r   df1df2r   r   r    r    r!   $test_table_index_incompatible_dtypes3   s   

"r6   c              
   C   sh  t | J}dtdddfg}|D ]5\}}t }|||< td| d}tjt	|d |
d| | W d    n1 sAw   Y  qW d    n1 sQw   Y  t }d	|d
< d|d< tddd|d< | jdd}t | 3}td}tjt	|d |
d| W d    n1 sw   Y  W d    d S W d    d S 1 sw   Y  d S )Ndatei  r$   r%   [z&] is not implemented as a table columnr   df1_fooobj1barobj2	datetime1T)datetimezjCannot serialize the column [datetime1]
because its data contents are not [string] but [date] object dtypedf_unimplemented)r   r?   r7   r   r   r2   r3   r   r   r   r1   _consolidate_convert)r   r   dtypesnfr   r   r    r    r!   'test_unimplemented_dtypes_table_columns>   s6   

"rF   c              
   C   sz  t | }tdd t }d|d< d|j|jdd df< |jd|d	d
 td}t	j
t|d t  W d    n1 sAw   Y  td}t	j
t|d |dd W d    n1 scw   Y  d}t	j
t|d |dd W d    n1 sw   Y  W d    n1 sw   Y  W d    n1 sw   Y  t| 1}ttjddtdtdddd}|j|dd	dd t|ddd t|ddd W d    n1 sw   Y  t| L}ttjddtdtdddd}|j|dd	d
 d}t	j
t|d t|ddd W d    n1 sw   Y  W d    d S W d    d S 1 s6w   Y  d S )NT)recordr:   stringr<   r   r'   r   r.   r/   z:__init__() missing 1 required positional argument: 'where'r   zWcannot process expression [df.index[3]], [2000-01-06 00:00:00] is not a valid conditionzdf.index[3]zinvalid syntaxzindex>
   ABCD20130101r*   )r   r,   dfq)r0   data_columnsz0index>Timestamp('20130104') & columns=['A', 'B']r   z
A>0 or C>0a  The passed where expression: A>0 or C>0\n\s*contains an invalid variable reference\n\s*all of the variable references must be a reference to\n\s*an axis \(e.g. 'index' or 'columns'\), or a data_column\n\s*The currently defined references are: index,columns\n)r   r   r   makeTimeDataFramelocr,   r   r2   r3   r   r   r   r   
ValueErrorr   SyntaxErrorr   r   nprandomrandnlistr	   to_hdfr
   )r   r   r   r   pathrL   r    r    r!   test_invalid_terms]   sf   




 $rX   c           
   
   C   s  t tjdd}t dtjdi}t dtjdi}t dtjdi}t ddtftjdi}t| B}dtd }||| ||||fD ]$}t	
d	}	tjt|	d
 ||| W d    n1 slw   Y  qMW d    d S 1 s}w   Y  d S )NrI   r$   r#   r$   r%   )1r%   rZ   r%   df_z?cannot match existing table structure for [0] on appending datar   )r   rR   rS   rT   objectr   r   randsr1   r2   r3   r   r   rP   )
r   r   r5   df3df4df5r   namedr   r    r    r!   7test_append_with_diff_col_name_types_raises_value_error   s$   
"rc   c              	   C   s   t tjddtdtdd}t| 2}d}tjt	|d |j
|dd	d
 W d    n1 s2w   Y  W d    d S W d    d S 1 sJw   Y  d S )Nr'   r(   abcdABCDEr,   r   z)complib only supports \[.*\] compression.r   r   foolib)complib)r   rR   rS   randrU   r   ensure_cleanr   r   rP   rV   )r   r   rW   r   r    r    r!   test_invalid_complib   s    "rk   idx2019Dr&   UTC)freqr+   tzabcc              	   C   s   t | | g}td|dgd}t|.}tjtdd ||d W d    n1 s,w   Y  W d    d S W d    d S 1 sDw   Y  d S )Nr   r#   rf   zSaving a MultiIndexr   r   )r   from_arraysr   r   r   r   NotImplementedErrorrV   )rl   r   mir   rW   r    r    r!   &test_to_hdf_multiindex_extension_dtype   s   	
"rv   c                 C   sL   | ddd}d}t jt|d t| W d    d S 1 sw   Y  d S )Niodataz"legacy_hdf/incompatible_dataset.h5z_Dataset\(s\) incompatible with Pandas data types, not table, or no datasets found in HDF5 file.r   )r   r   rP   r
   )datapath	data_pathmessager    r    r!   test_unsuppored_hdf_file_error   s   
"r|   c              	   C   s   t tjddtdtdd}t| ]}d}tjt|d t	|d W d    n1 s.w   Y  |
|d	 t|d
d}|  d}tjt|d t	|d	 W d    n1 s\w   Y  W d    d S W d    d S 1 stw   Y  d S )Nr'   r(   rd   re   rf   zFile [\S]* does not existr   keyr   r)modez&The HDFStore must be open for reading.)r   rR   rS   ri   rU   r   r   r   OSErrorr
   rV   r   close)r   r   rW   r   r   r    r    r!   test_read_hdf_errors   s     

"r   c                  C   sF   d} t jt| d ttdd W d    d S 1 sw   Y  d S )Nz5Support for generic buffers has not been implemented.r       r   )r   r   rt   r
   r   )r   r    r    r!   #test_read_hdf_generic_buffer_errors   s   "r   bad_versionrY   )r$   12123c                 C   sD   d}t jt|d td| d W d    d S 1 sw   Y  d S )Nz5Version is incorrect, expected sequence of 3 integersr   values_block_0)version)r   r   rP   r   )r   r   r    r    r!   )test_maybe_adjust_name_bad_version_raises   s   "r   )(r?   rw   r   r2   warningsr   numpyrR   r   pandasr   r   r   r   r   r   r	   r
   pandas.tests.io.pytables.commonr   r   pandas.io.pytablesr   r   mark
single_cpu
pytestmarkr"   r6   rF   rX   rc   rk   parametrizerU   rv   r|   r   r   r   r    r    r    r!   <module>   s8    $	@

	