U
    e»{jý  ã                   @   sD   d dl Zd dlZd dlmZ d dlmZ e d¡ZG dd„ dƒZ	dS )é    N)Úassert_equal)ÚTablez
dask.arrayc                   @   sL   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dS )ÚTestDaskHandlerc                 C   s   t ƒ | _t d¡| jd< d S )Né
   Úa)r   ÚtÚdaÚarange)ÚselfÚmethod© r   úc/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/table/mixins/tests/test_dask.pyÚsetup_method   s    zTestDaskHandler.setup_methodc                 C   s:   | j  | j d ¡ t| j d  ¡ t t d¡dg¡ƒ d S )Nr   r   r   )r   Úadd_rowr   ÚcomputeÚnpZhstackr	   ©r
   r   r   r   Útest_add_row   s    zTestDaskHandler.test_add_rowc                 C   s4   t | jd tjƒst‚t| jd  ¡ t d¡ƒ d S )Nr   r   )	Ú
isinstancer   r   ÚArrayÚAssertionErrorr   r   r   r	   r   r   r   r   Útest_get_column   s    zTestDaskHandler.test_get_columnc                 C   sH   | j d }t|d tjƒst‚t|d dƒr0t‚|d  ¡ dksDt‚d S )Né   r   Úinfo)r   r   r   r   r   Úhasattrr   ©r
   Úsubr   r   r   Útest_slicing_row_single   s    
z'TestDaskHandler.test_slicing_row_singlec                 C   sR   | j dd … }t|d tjƒs"t‚t|d dƒs4t‚t|d  ¡ t 	dd¡ƒ d S )Nr   r   r   r   ©
r   r   r   r   r   r   r   r   r   r	   r   r   r   r   Útest_slicing_row_range   s    z&TestDaskHandler.test_slicing_row_rangec                 C   sL   | j d }t|d tjƒst‚t|d dƒs0t‚t|d  ¡ t 	d¡ƒ d S )N)r   r   r   r   r   r   r   r   r   Útest_slicing_column_range#   s    
z)TestDaskHandler.test_slicing_column_rangec                 C   s.   | j  ¡ ddddddddd	d
ddgks*t‚d S )Nz a z---z  0z  1z  2z  3z  4z  5z  6z  7z  8z  9)r   Zpformat_allr   r   r   r   r   Útest_pformat)   s    ôzTestDaskHandler.test_pformatc                 C   s´   d| j d j_| j dd… }|d jjdks0t‚|d jjdksDt‚| j d  ¡ }|jjdksbt‚|jjdksrt‚| j  | j d ¡ | j d jjdksšt‚| j d jjdks°t‚d S )NzA dask columnr   é   é   r   )r   r   ÚdescriptionÚnamer   Úcopyr   )r
   r   Úcolr   r   r   Útest_info_preserved9   s    z#TestDaskHandler.test_info_preservedN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r    r!   r(   r   r   r   r   r   
   s   r   )
Únumpyr   ZpytestZnumpy.testingr   Zastropy.tabler   Zimportorskipr   r   r   r   r   r   Ú<module>   s
   
