o
    5c                     @   s6   d dl Zd dlmZmZ d dlmZ G dd dZdS )    N)PeriodIndex	factorizec                   @   s   e Zd Zdd Zdd ZdS )TestFactorizec                 C   s  t g ddd}tjg dtjd}t g ddd}| \}}t|| t|| |jdd\}}t|| t|| t g d	dd}tjg d
tjd}|jdd\}}t|| t|| tjg dtjd}t g ddd}| \}}t|| t|| d S )N)2014-01r   2014-02r   2014-03r   M)freq)r   r      r
      r   dtype)r   r   r   T)sort)r   r   r   r   r   r   )r   r   r
   r   r   r   )r   r   r
   r   r   r   )r   r   r   )r   nparrayintpr   tmassert_numpy_array_equalassert_index_equal)selfidx1exp_arrexp_idxarridxidx2 r   h/var/www/html/gps/gps/lib/python3.10/site-packages/pandas/tests/indexes/period/methods/test_factorize.pytest_factorize   s.   zTestFactorize.test_factorizec                 C   sV   g d}t |\}}tjg dtjd}t|| tjg dtd}t|| d S )N)r
   r          @      ?)r   r
   r   r   )y      ?        y       @        r   )r   r   r   r   r   r   object)r   r   labelsuniquesexpected_labelsexpected_uniquesr   r   r   test_factorize_complex*   s   z$TestFactorize.test_factorize_complexN)__name__
__module____qualname__r   r%   r   r   r   r   r   
   s    r   )	numpyr   pandasr   r   pandas._testing_testingr   r   r   r   r   r   <module>   s    