a
    ߙfbD                     @   sB   d dl Z e dZd dlmZ dd Zdd Zdd	 Zd
d ZdS )    NasdfTablec                  C   s0   g d} g d}g d}t | ||gdddidS )N)         )g       @g      @gffffff @)xyzabcnamezfirst table)namesmetar   r    r   Alib/python3.9/site-packages/astropy/io/misc/asdf/tests/test_io.py
make_table   s    r   c                 C   s   t | d}t }|| t|F}d| v s:J t|d tsLJ t	|d |ks`J W d    n1 st0    Y  t
|}t	||ksJ d S )N
table.asdfdata)strjoinr   writer   openkeys
isinstancer   allreadtmpdirtmpfiletableafnew_tr   r   r   test_table_io   s    
2
r%   c                 C   s   t | d}t }|j|dd t|V}d| v s>J d| vsNJ t|d ts`J t	|d |kstJ W d    n1 s0    Y  t
t t|}W d    n1 s0    Y  tj|dd}t	||ksJ d S )Nr   Z	something)Zdata_keyr   )r   r   r   r   r   r   r   r   r   r   pytestraisesKeyErrorr   r   r   r   r   test_table_io_custom_key&   s    2(r)   c                 C   s   t | d}t }dd }|j||d t|X}d| v sFJ d|d v sVJ d| vsfJ t|d d |ks~J W d    n1 s0    Y  t	t
 t|}W d    n1 s0    Y  dd	 }tj||d
}t||ksJ d S )Nr   c                 S   s   t t | ddS )N)bar)foo)dict)Ztabr   r   r   make_custom_treeC   s    z3test_table_io_custom_tree.<locals>.make_custom_tree)Z	make_treer+   r*   r   c                 S   s   | d d S )Nr+   r*   r   )Zasdffiler   r   r   
find_tableS   s    z-test_table_io_custom_tree.<locals>.find_table)r.   )r   r   r   r   r   r   r   r   r&   r'   r(   r   r   )r    r!   r"   r-   r#   r$   r.   r   r   r   test_table_io_custom_tree=   s    6(r/   )	r&   Zimportorskipr   Zastropy.tabler   r   r%   r)   r/   r   r   r   r   <module>   s   
