
    IR-e                     h    d dl Zd dlZd dlmZ d dlmZ  ej        d          Z G d d          Z	dS )    N)assert_equal)Tablez
dask.arrayc                   >    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	S )
TestDaskHandlerc                 p    t                      | _        t                              d          | j        d<   d S )N
   a)r   tdaarange)selfmethods     Dlib/python3.11/site-packages/astropy/table/mixins/tests/test_dask.pysetup_methodzTestDaskHandler.setup_method   s&    iimms    c                     | j                             | j         d                    t          | j         d                                         t	          j        t	          j        d          dg                     d S )Nr   r	   r   )r
   add_rowr   computenphstackr   r   s    r   test_add_rowzTestDaskHandler.test_add_row   s[    tvay!!!TVC[((**BIry}}a6H,I,IJJJJJr   c                     t          | j        d         t          j                  sJ t	          | j        d                                         t          j        d                     d S )Nr	   r   )
isinstancer
   r   Arrayr   r   r   r   r   s    r   test_get_columnzTestDaskHandler.test_get_column   sP    $&+rx00000TVC[((**BIbMM:::::r   c                     | j         d         }t          |d         t          j                  sJ t	          |d         d          rJ |d                                         dk    sJ d S )N   r	   info)r
   r   r   r   hasattrr   r   subs     r   test_slicing_row_singlez'TestDaskHandler.test_slicing_row_single   sj    fQi#c(BH-----3s8V,,,,,3x!!Q&&&&&&r   c                    | j         dd          }t          |d         t          j                  sJ t	          |d         d          sJ t          |d                                         t          j        dd                     d S )Nr   r	   r   r   	r
   r   r   r   r    r   r   r   r   r!   s     r   test_slicing_row_rangez&TestDaskHandler.test_slicing_row_range   sy    fQRRj#c(BH-----s3x(((((SX%%''1b)9)9:::::r   c                    | j         d         }t          |d         t          j                  sJ t	          |d         d          sJ t          |d                                         t          j        d                     d S )N)r	   r	   r   r   r%   r!   s     r   test_slicing_column_rangez)TestDaskHandler.test_slicing_column_range#   sq    fVn#c(BH-----s3x(((((SX%%''277777r   c                 H    | j                                         g dk    sJ d S )N)z a z---z  0z  1z  2z  3z  4z  5z  6z  7z  8z  9)r
   pformat_allr   s    r   test_pformatzTestDaskHandler.test_pformat)   s@    v!!## (
 (
 (
 
 
 
 
 
 
r   c                    d| j         d         j        _        | j         dd         }|d         j        j        dk    sJ |d         j        j        dk    sJ | j         d                                         }|j        j        dk    sJ |j        j        dk    sJ | 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   descriptionnamecopyr   )r   r"   cols      r   test_info_preservedz#TestDaskHandler.test_info_preserved9   s    '6s$fQqSk3x}!S((((3x}(O;;;;fSk  x}####x#6666tvay!!!vc{$++++vc{+>>>>>>r   N)__name__
__module____qualname__r   r   r   r#   r&   r(   r+   r3    r   r   r   r   
   s        $ $ $K K K; ; ;' ' '; ; ;8 8 8
 
 
 ? ? ? ? ?r   r   )
numpyr   pytestnumpy.testingr   astropy.tabler   importorskipr   r   r7   r   r   <module>r=      s         & & & & & &      V&&<? <? <? <? <? <? <? <? <? <?r   