
    .i
                     H    d dl Zd dlZd dlmZmZmZ d dlmZ	  G d d      Z
y)    N)SeriesTimedeltaIndex
date_rangec                   <    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
y	)
TestSeriesDiffc                     t        t        j                  j                  d      j	                  d            }t        j                  t        d      5  |j                  d       d d d        y # 1 sw Y   y xY w)N   zperiods must be an integer)matchg      ?)	r   nprandomdefault_rngstandard_normalpytestraises
ValueErrordiff)selfseriess     c/var/www/app/trading-bot/venv/lib/python3.12/site-packages/pandas/tests/series/methods/test_diff.py!test_diff_series_requires_integerz0TestSeriesDiff.test_diff_series_requires_integer   sV    		--a0@@CD]]:-IJ 	KK	 	 	s   A..A7c                     t        t        j                  d            }t        j                  |      }t        j                  g d      }t        j                  ||       y )N   )   r   r   r   )r   r   aranger   arraytmassert_numpy_array_equal)r   serresexpecteds       r   test_diff_npzTestSeriesDiff.test_diff_np   sA     RYYq\"ggcl88L)
##C2    c                 `    d}|dz   }t        ||g      }|j                         }|d   dk(  sJ y )Nl     _r   )r   r   )r   abr   results        r   test_diff_intzTestSeriesDiff.test_diff_int   s8    EaVnayA~~r"   c                 j   t        t        j                  dt        j                        t	        dd      d      }|j                          |j                  d      }||j                  d      z
  }t        j                  ||       |j                  d      }||z
  }t        j                  ||       y )	N
   )dtypez
2020-01-01periodsts)indexnamer   )	r   r   r   float64r   r   shiftr   assert_series_equal)r   r-   r&   r    s       r   test_diff_tzzTestSeriesDiff.test_diff_tz%   s    IIb

+\26

 		 $
vx0 7
vx0r"   c                    t        t        dd            }|j                         }||j                  d      z
  }t	        j
                  ||       ||j                  d      z
  }|j                         }t	        j
                  ||       y )N20130102r   r+   r   )r   r   r   r2   r   r3   r   r   r&   r    s       r   test_diff_dt64zTestSeriesDiff.test_diff_dt648   so    Z
A671%
vx0 &,,q/)==?
vx0r"   c                     t        t        ddd      d      }|j                         }t        t        dgdgd	z  z         d      }t	        j
                  ||       y )
Nz2000-01-01 09:00:00r   z
US/Eastern)r,   tzfoo)r/   NaTz1 days   )r   r   r   r   r   r3   r7   s       r   test_diff_dt64tzzTestSeriesDiff.test_diff_dt64tzD   sW    ,aLI
 .%H:>)ABO
vx0r"   c                     g d}t         j                  ddddg}t        |      }|j                         }t        |      }t	        j
                  ||       y )N)FTTFFTF)r   nanr   r   r   r3   )r   dataoutputr   r&   r    s         r   test_diff_boolzTestSeriesDiff.test_diff_boolN   sH    0&&$tU3Tl&>
vx0r"   c                     t        dddt        j                  ddg      }|j                         }||j	                  d      z
  }t        j                  ||       y )NFTg      @r   )r   r   r@   r   r2   r   r3   r7   s       r   test_diff_object_dtypez%TestSeriesDiff.test_diff_object_dtypeW   sJ    eT3e<=1%
vx0r"   N)__name__
__module____qualname__r   r!   r'   r4   r8   r>   rC   rE    r"   r   r   r      s*    
31&
1111r"   r   )numpyr   r   pandasr   r   r   pandas._testing_testingr   r   rI   r"   r   <module>rN      s&      
 P1 P1r"   