U
    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| n}|d| d}tjt|d |jddgd W 5 Q R X d}tjt|d |jddgd W 5 Q R X W 5 Q R X 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    H/tmp/pip-unpacked-wheel-g7fro6k3/pandas/tests/io/pytables/test_errors.pytest_pass_spec_to_storer   s    
r"   c              
   C   s   t ddddgi}t ddddgitddd	d
}t| L}|jd|dd td}tjt|d |jd|ddd W 5 Q R X W 5 Q R X d S )Na                  z1/1/2000periods)indexframetableformatz/incompatible kind in col [integer - datetime64]r   T)r0   append)	r   r	   r   r   reescaper   r   r   )r   Zdf1df2r   r   r    r    r!   $test_table_index_incompatible_dtypes3   s    

r5   c                 C   s  t | v}dtdddfg}|D ]V\}}t }|||< td| d}tjt	|d |
d| | W 5 Q R X q"W 5 Q R X t }d	|d
< d|d< tddd|d< | jdd}t | 6}td}tjt	|d |
d| W 5 Q R X W 5 Q R X d S )Ndatei  r$   r%   [z&] is not implemented as a table columnr   Zdf1_fooZobj1barobj2Z	datetime1T)datetimezjCannot serialize the column [datetime1]
because its data contents are not [string] but [date] object dtypeZdf_unimplemented)r   r;   r6   r   r   r2   r3   r   r   r   r1   Z_consolidate_convert)r   r   Zdtypesnfr   r   r    r    r!   'test_unimplemented_dtypes_table_columns>   s&    
(
r?   c                 C   s  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 5 Q R X td}t	j
t|d |dd W 5 Q R X d}t	j
t|d |dd W 5 Q R X W 5 Q R X W 5 Q R X t| X}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 5 Q R X t| f}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 5 Q R X W 5 Q R X d S )NT)recordr8   stringr9   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>
   ABCDZ20130101r*   )r   r,   dfq)r0   Z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   ZmakeTimeDataFramelocr,   r   r2   r3   r   r   r   r   
ValueErrorr   SyntaxErrorr   r   nprandomrandnlistr	   to_hdfr
   )r   r   r   r   pathrD   r    r    r!   test_invalid_terms]   sN    
*



rN   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| d}dtd }||| ||||fD ]4}t	
d	}	tjt|	d
 ||| W 5 Q R X qW 5 Q R X d S )NrB   r$   r#   r$   r%   )1r%   rP   r%   Zdf_z?cannot match existing table structure for [0] on appending datar   )r   rH   rI   rJ   objectr   r   Zrandsr1   r2   r3   r   r   rF   )
r   r   r4   Zdf3Zdf4Zdf5r   namedr   r    r    r!   7test_append_with_diff_col_name_types_raises_value_error   s    
rT   c              
   C   sh   t tjddtdtdd}t| 4}d}tjt	|d |j
|dd	d
 W 5 Q R X W 5 Q R X d S )Nr'   r(   abcdABCDEr,   r   z)complib only supports \[.*\] compression.r   r   Zfoolib)Zcomplib)r   rH   rI   randrK   r   Zensure_cleanr   r   rF   rL   )r   r   rM   r   r    r    r!   test_invalid_complib   s
     rY   idxZ2019Dr&   UTC)freqr+   tzabcc              
   C   s\   t | | g}td|dgd}t|,}tjtdd ||d W 5 Q R X W 5 Q R X d S )Nr   r#   rW   zSaving a MultiIndexr   r   )r   Zfrom_arraysr   r   r   r   NotImplementedErrorrL   )rZ   r   mir   rM   r    r    r!   &test_to_hdf_multiindex_extension_dtype   s
    	
rb   c              	   C   s6   | ddd}d}t jt|d t| W 5 Q R X 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   rF   r
   )datapathZ	data_pathmessager    r    r!   test_unsuppored_hdf_file_error   s
    rg   c              
   C   s   t tjddtdtdd}t| v}d}tjt|d t	|d W 5 Q R X |
|d	 t|d
d}|  d}tjt|d t	|d	 W 5 Q R X W 5 Q R X d S )Nr'   r(   rU   rV   rW   zFile [\S]* does not existr   keyr   r)modez&The HDFStore must be open for reading.)r   rH   rI   rX   rK   r   r   r   OSErrorr
   rL   r   close)r   r   rM   r   r   r    r    r!   test_read_hdf_errors   s     
rm   c               	   C   s0   d} t jt| d ttdd W 5 Q R X d S )Nz5Support for generic buffers has not been implemented.r       r   )r   r   r`   r
   r   )r   r    r    r!   #test_read_hdf_generic_buffer_errors   s    ro   bad_versionrO   )r$   Z12Z123c              	   C   s.   d}t jt|d td| d W 5 Q R X d S )Nz5Version is incorrect, expected sequence of 3 integersr   Zvalues_block_0)version)r   r   rF   r   )rp   r   r    r    r!   )test_maybe_adjust_name_bad_version_raises   s    rr   )(r;   rc   r   r2   warningsr   ZnumpyrH   r   Zpandasr   r   r   r   r   r   r	   r
   Zpandas.tests.io.pytables.commonr   r   Zpandas.io.pytablesr   r   markZ
single_cpuZ
pytestmarkr"   r5   r?   rN   rT   rY   ZparametrizerK   rb   rg   rm   ro   rr   r    r    r    r!   <module>   s6   $	@

	