
    Eg                     H    d dl Zd dlZd dlmZ d dlmZ  G d d          ZdS )    N)TimedeltaArrayc                   >    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	S )
TestTimedeltaArrayConstructorc           	         t          j        g dd                              d          }d}t          j        t
          |          5  t          j        t          d          5  t          |
                    ddd	                     d d d            n# 1 swxY w Y   d d d            n# 1 swxY w Y   t          j        t
          |          5  t          j        t          d          5  t          |d
g                                                    d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )N)r            zm8[h]dtypem8[ns]%TimedeltaArray.__init__ is deprecatedmatchzOnly 1-dimensionalr   r   r   )nparrayastypetmassert_produces_warningFutureWarningpytestraises
ValueErrorr   reshapesqueeze)selfarrdepr_msgs      e/var/www/sysmax/venv/lib/python3.11/site-packages/pandas/tests/arrays/timedeltas/test_constructors.pytest_only_1dim_acceptedz5TestTimedeltaArrayConstructor.test_only_1dim_accepted	   s(   h|||7333::8DD:'XFFF 	5 	5z1EFFF 5 5s{{1a334445 5 5 5 5 5 5 5 5 5 5 5 5 5 5	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5
 'XFFF 	3 	3z1EFFF 3 3sA3x//112223 3 3 3 3 3 3 3 3 3 3 3 3 3 3	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3sl   	B-%%B
B-B	B-B	B--B14B1D</)D$D<$D(	(D<+D(	,D<<E E c                 t   t          j        g dt           j                  dz  dz  }d}d}t          j        t
          |          5  t          j        t          |          5  t          |
                    d          d	
           d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )N)r   r   r   r
   i  i ʚ;zQInferred frequency None from passed values does not conform to passed frequency Dr   r   ztimedelta64[ns]D)freq)r   r   int64r   r   r   r   r   r   r   view)r   r   msgr   s       r   test_freq_validationz2TestTimedeltaArrayConstructor.test_freq_validation   s`   hyyy111D85@, 	 ;'XFFF 	F 	Fz555 F Fsxx(9::EEEEF F F F F F F F F F F F F F F	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	Fs6   B-$%B	B-B	B-B	B--B14B1c                     d}t          j        t          |          5  t          j        t
          d          5  t          g d           d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )Nr   r   listr   r   r	   )r   r   r   r   r   r   r   )r   r   s     r   test_non_array_raisesz3TestTimedeltaArrayConstructor.test_non_array_raises%   s    :'XFFF 	* 	*z888 * *yyy)))* * * * * * * * * * * * * * *	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	*s4   A0AA0A	A0A	 A00A47A4c                     d}t          j        t          |          5  t          j        t          j        g dd                     d d d            d S # 1 swxY w Y   d S )Nz3dtype bool cannot be converted to timedelta64\[ns\]r   r)   boolr
   )r   r   	TypeErrorr   _from_sequencer   r   r   r%   s     r   test_other_type_raisesz4TestTimedeltaArrayConstructor.test_other_type_raises+   s    D]9C000 	M 	M)"(999F*K*K*KLLL	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	Ms   +AAAc                 >   d}t          j        t          |          5  t          j        t          j        g dd          d           d d d            n# 1 swxY w Y   d}t          j        t          |          5  t          j        t          j        g dd          t          j        d                     d d d            n# 1 swxY w Y   d	}t          j        t          |          5  t          j        t          j        g dd          t          j        d
                     d d d            n# 1 swxY w Y   d}t          j        t          |          5  t          j        t          j        g dd          d           d d d            n# 1 swxY w Y   d}t          j        t          |          5  t          j        t          j        g dd          t          j        d                     d d d            d S # 1 swxY w Y   d S )Nz;dtype 'category' is invalid, should be np.timedelta64 dtyper   r)   i8r
   categoryz8dtype 'int64' is invalid, should be np.timedelta64 dtyper#   zCdtype 'datetime64\[ns\]' is invalid, should be np.timedelta64 dtypezM8[ns]zHdtype 'datetime64\[us, UTC\]' is invalid, should be np.timedelta64 dtypezM8[us, UTC]z;Supported timedelta64 resolutions are 's', 'ms', 'us', 'ns'zm8[Y])r   r   r   r   r.   r   r   r   r/   s     r   test_incorrect_dtype_raisesz9TestTimedeltaArrayConstructor.test_incorrect_dtype_raises0   se   K]:S111 	 	)$///z   	 	 	 	 	 	 	 	 	 	 	 	 	 	 	
 I]:S111 	 	)$///rx7H7H   	 	 	 	 	 	 	 	 	 	 	 	 	 	 	
 U]:S111 	 	)$///rx7I7I   	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 X 	 ]:S111 	 	)$///}   	 	 	 	 	 	 	 	 	 	 	 	 	 	 	
 L]:S111 	 	)$///rx7H7H   	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	sY   -AAA??C

CC2?D==EE%-FF"%F"?HHHc                 V   t          j        g dd          }t          j        d          }d}d}t          j        t
          |          5  t          j        t          |          5  t          ||           d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )Nr)   zm8[s]r
   r   z&Values resolution does not match dtyper   r   )
r   r   r   r   r   r   r   r   r   r   )r   r   r   r%   r   s        r   "test_mismatched_values_dtype_unitsz@TestTimedeltaArrayConstructor.test_mismatched_values_dtype_unitsQ   s*   hyyy000""7:'XFFF 	1 	1z555 1 1s%00001 1 1 1 1 1 1 1 1 1 1 1 1 1 1	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1s6   B(B:BB
	
BB
	BB"%B"c                     t          j        g dd          }t          j        |d          }|j        |u sJ t          j        |d          }|j        |usJ |j        j        |usJ d S )Nr)   r   r
   F)copyT)r   r   r   r.   _ndarraybase)r   datar   s      r   	test_copyz'TestTimedeltaArrayConstructor.test_copy[   s    x			222+Du===|t####+Dt<<<|4''''| ,,,,,,    c                     d}t          j        t          |          5  t          j        g t
                     d d d            d S # 1 swxY w Y   d S )Nz9dtype 'object' is invalid, should be np.timedelta64 dtyper   r
   )r   r   r   r   r.   objectr/   s     r   test_from_sequence_dtypez6TestTimedeltaArrayConstructor.test_from_sequence_dtyped   s    I]:S111 	< 	<)"F;;;;	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	<s   AAAN)__name__
__module____qualname__r   r&   r*   r0   r4   r6   r<   r@    r=   r   r   r      s        3 3 3F F F* * *M M M
  B1 1 1- - -< < < < <r=   r   )	numpyr   r   pandas._testing_testingr   pandas.core.arraysr   r   rD   r=   r   <module>rI      sz               - - - - - -_< _< _< _< _< _< _< _< _< _<r=   