U
    e»{j  ã                   @   s¶   d dl Zd dlZd dlmZmZ d dlmZ G dd„ dƒZG dd„ deƒZ	ej
je dd	�ej
 d
dddg¡ej
 dedƒede d¡ dd¡ddd�eddd�g¡dd„ ƒƒƒZdS )é    N)ÚTimeÚ	TimeDelta)ÚARRAY_FUNCTION_ENABLEDc                   @   s4   e Zd Zdd„ Zd
dd„Zej dd¡dd	„ ƒZdS )ÚTestFunctionsTimec                 C   s$   t dt d¡ dd¡ddd�| _d S )NéPÃ  é   é   é   ÚmjdÚtai©ÚformatÚscale)r   ÚnpÚarangeÚreshapeÚt©Úcls© r   ú`/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/time/tests/test_functions.pyÚsetup_class   s    zTestFunctionsTime.setup_classNc                 O   s¦   |d kr| j j}|d kr | j j}|d kr0| j j}|| j f|ž|Ž}|| j jf|ž|Ž}|| j jf|ž|Ž}	|||	||d�}
t|tjƒr�t 	|
¡}
t 
||
k¡s¢t‚d S )Nr   )r   Ú	__class__r   r   Újd1Újd2Ú
isinstancer   ZndarrayÚarrayÚallÚAssertionError)ÚselfÚfuncr   r   r   ÚargsÚkwargsÚoutr   r   Úexpectedr   r   r   Úcheck   s    
zTestFunctionsTime.checkÚaxis)r   é   c                 C   s   | j tj|tdd� d S )NÚjd)r&   r   r   )r%   r   Údiffr   )r   r&   r   r   r   Ú	test_diff   s    zTestFunctionsTime.test_diff)NNN)	Ú__name__Ú
__module__Ú__qualname__r   r%   ÚpytestÚmarkÚparametrizer*   r   r   r   r   r   
   s   
r   c                   @   sD   e Zd Zdd„ Zej dd¡ej dejej	ej
f¡dd„ ƒƒZdS )	ÚTestFunctionsTimeDeltac                 C   s"   t t d¡ dd¡ddd�| _d S )Nr   r   r	   r(   r   r   )r   r   r   r   r   r   r   r   r   r   $   s    z"TestFunctionsTimeDelta.setup_classr&   )r   r'   Nr    c                 C   s   | j ||d� d S )N)r&   )r%   )r   r    r&   r   r   r   Útest_sum_like'   s    z$TestFunctionsTimeDelta.test_sum_likeN)r+   r,   r-   r   r.   r/   r0   r   ÚsumZmeanZmedianr2   r   r   r   r   r1   #   s   r1   z Needs __array_function__ support)ÚreasonÚ	attributeÚshapeÚndimÚsizer   z2001-02-03T04:05:06r   r   r   r	   r
   r   r   éd   r(   )r   c                 C   s(   t t|ƒ}|| ƒ}|t | |ƒks$t‚d S )N)Úgetattrr   r   )r   r5   ÚfunctionÚresultr   r   r   Útest_shape_attribute_functions-   s    
r=   )Únumpyr   r.   Zastropy.timer   r   Z.astropy.units.quantity_helper.function_helpersr   r   r1   r/   Zxfailr0   r   r   r=   r   r   r   r   Ú<module>   s$   
 ÿ
ýþ