a
    ߙfb:                     @   s`   d dl Z d dlZd dlmZ d dlmZ e jdd Ze jdd Z	dd	 Z
d
d Zdd ZdS )    N)SortedArray)Tablec                  C   s^   t dd tddD } t dd tddD }t| |g}||  }t||d  S )Nc                 S   s   g | ]}|d  qS )    .0xr   r   =lib/python3.9/site-packages/astropy/table/tests/test_array.py
<listcomp>       zarray.<locals>.<listcomp>      c                 S   s   g | ]}|qS r   r   r   r   r   r	   r
      r   col1)nparrayranger   Zargsortr   copy)col0r   tr   r   r	   r   
   s
    r   c                  C   s*   t dd tdD } t| | d  S )Nc                 S   s   g | ]}|gd  qS )
   r   r   r   r   r	   r
      r   zwide_array.<locals>.<listcomp>d   r   )r   r   aranger   r   )r   r   r   r	   
wide_array   s    r   c                 C   sP   t ddD ].}td|  | |d |f|gks
J q
| dg ksLJ d S )Nr   r   zSearching for r   )r      )r   printfind)r   ir   r   r	   test_array_find   s    r   c                 C   sd   t | dddg dks J t | dddg dks@J t | dddg dks`J d S )	N)r      )r      )TT)r   r   r   r   )FT)r   r   r   )TF)r   r   r   )r   allr   )r   r   r   r	   test_array_range"   s      r!   c                 C   s8   | d d j }t|tdd tdD ks4J d S )Nr   c                 S   s   g | ]
}|gqS r   r   r   r   r   r	   r
   -   r   z#test_wide_array.<locals>.<listcomp>r   )datar   r    r   r   )r   Z	first_rowr   r   r	   test_wide_array(   s    r#   )ZpytestZnumpyr   Zastropy.table.sorted_arrayr   Zastropy.table.tabler   Zfixturer   r   r   r!   r#   r   r   r   r	   <module>   s   
	
