o
    5c                  
   @   s,  d dl Z d dlZd dlZd dlmZmZmZ d dlm	Z
 ejdddgddggdd Zejjdd	 Zejd
deg dfdeg dfdeg dfgdd Zejjddejd
deg dfdeg dfdeg dfgdd Zejddddgdgdd Zdd Zdd ZdS )    N)	DataFrameSeries
date_rangesubsetaBc                 C   sd   t g dg dg dd}td}tjt|d ||  W d    d S 1 s+w   Y  d S )N)r   r      Ar   CzIndex(['a'], dtype='object'))match)r   reescapepytestraisesKeyError
duplicated)r   dfmsg r   `/var/www/html/gps/gps/lib/python3.10/site-packages/pandas/tests/frame/methods/test_duplicated.py+test_duplicated_with_misspelled_column_name   s
   
"r   c                  C   sF   dd t dD } t| j}| }t|tsJ |jtjks!J d S )Nc                 S   s&   i | ]}d |dt jdddqS )col_02dr   i  i0u  )nprandomrandint).0ir   r   r   
<dictcomp>   s   & zBtest_duplicated_do_not_fail_on_wide_dataframes.<locals>.<dictcomp>d   )	ranger   Tr   
isinstancer   dtyper   bool_)datar   resultr   r   r   .test_duplicated_do_not_fail_on_wide_dataframes   s
   
r(   zkeep, expectedfirst)FFTFTlast)TTFFFF)TTTFTc                 C   s2   t g dg dd}|j| d}t|| d S )Nr   r   r      r   r   br.   cr   )r
   r   keep)r   r   tmassert_series_equalr1   expectedr   r'   r   r   r   test_duplicated_keep(   s   	r6   z+GH#21720; nan/None falsely considered equal)reasonc                 C   s<   t tjddd tjgddtd}|j| d}t|| d S )N   r   )r   x)r$   r0   )r   r   nanobjectr   r2   r3   r4   r   r   r   test_duplicated_nan_none7   s    
r<   r
   c                 C   sx   t g dg dtjddd tjgd}| d u rt|j} nt| tr%| g} ||  j|d}|j|| d}t	|| d S )Nr+   r-   r8   r	   r0   )r1   r   )
r   r   r:   listcolumnsr#   strr   r2   r3   )r   r1   r   r5   r'   r   r   r   test_duplicated_subsetG   s   
r@   c                  C   s8   t ddgd} | d}| | }|  }t|| d S )Nr   r.   )r>   )r   r   copyr2   assert_frame_equal)r   dupesr'   r5   r   r   r   test_duplicated_on_empty_frame]   s
   
rD   c                  C   s^   t ddd} td| d}|ddg}|  sJ td| i}|j }|  s-J d S )Nz
2010-07-01z
2010-08-05)endAAA)symboldaterH   rG   )r   r   r   allrH   )dateststr'   r   r   r    test_frame_datetime64_duplicatedh   s   
rL   )r   numpyr   r   pandasr   r   r   pandas._testing_testingr2   markparametrizer   slowr(   r6   xfailr<   r@   rD   rL   r   r   r   r   <module>   s<    
	


