a
    ß™fb´  ã                   @   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   úClib/python3.9/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J ‚t| jd  ¡ t d¡ƒ d S )Nr   r   )Ú
isinstancer   r   ÚArrayr   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J ‚t|d dƒr0J ‚|d  ¡ dksDJ ‚d S )Né   r   Úinfo)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"J ‚t|d dƒs4J ‚t|d  ¡ t dd¡ƒ d S )Nr   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J ‚t|d dƒs0J ‚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  ¡ g d¢ksJ ‚d S )N)z a z---z  0z  1z  2z  3z  4z  5z  6z  7z  8z  9)r   Zpformat_allr   r   r   r   Útest_pformat*   s    zTestDaskHandler.test_pformatc                 C   s´   d| j d j_| j dd… }|d jjdks0J ‚|d jjdksDJ ‚| j d  ¡ }|jjdksbJ ‚|jjdksrJ ‚| j  | j d ¡ | j d jjdksšJ ‚| j d jjdks°J ‚d S )NzA dask columnr   é   é   r   )r   r   ZdescriptionÚnameÚcopyr   )r
   r   Úcolr   r   r   Útest_info_preserved/   s    z#TestDaskHandler.test_info_preservedN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r    r&   r   r   r   r   r   
   s   r   )
ZpytestZnumpyr   Znumpy.testingr   Zastropy.tabler   Zimportorskipr   r   r   r   r   r   Ú<module>   s
   
