
    PiU              
          d Z ddlZddlZddlZddlmZ ddlZddl	m
Z ddlmZ ddlmZ ddlmZ d Zd Zej                            d	d
dg          d             Zd Zd Zd Zd Zd Zd Zej                            d ej        dd           ej        ddd           ej        ddd           ej        dd          g          d             Z dS )z_
Tests for the pseudo-public API implemented in internals/api.py and exposed
in core.internals
    N)Pandas4Warning)create_dataframe_from_blocks)	internals)apic                  :    t           j        t          j        u sJ d S )N)r   
make_blockr        s/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/internals/test_api.pytest_internals_apir      s    3>111111r
   c                      g d} g d}d t          t                    D             }t          |          t          || z             k    s*J t          |          t          || z             z              d S )N)blocksconcatmanagersconstructionr   ops)r   BlockManagerSingleBlockManagerconcatenate_managersc                 <    g | ]}|                     d           |S )__)
startswith).0xs     r   
<listcomp>z"test_namespace.<locals>.<listcomp>*   s)    BBBAq||D/A/ABaBBBr
   )dirr   set)modulesexpectedresults      r   test_namespacer!      s      G  H CBYBBBFv;;#h0111113v;;XPWEWAXAX3X11111r
   nameBlockExtensionBlockc                     |  d}t          j        t          |          5  t          t          |            d d d            d S # 1 swxY w Y   d S )Nz( is deprecated.* Use public APIs insteadmatch)tmassert_produces_warningr   getattrr   )r"   msgs     r   test_deprecationsr,   .   s     
;
;
;C		#N#	>	>	> ! !	4   ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !s   AAAc                  
   t          j        ddd          } d}t          j        t          |          5  t          j        | dg          }d d d            n# 1 swxY w Y   |j        d	k    sJ |j        j        d	k    sJ d S )
N2012   UTCperiodstzzmake_block is deprecatedr&   r   )	placement)   r/   )	pd
date_ranger(   r)   r   r   r   shapevalues)dtir+   blks      r   test_make_block_2d_with_dtir<   <   s    
-e
4
4
4C
$C		#N#	>	>	> 1 1nSQC0001 1 1 1 1 1 1 1 1 1 1 1 1 1 1 9:v%%%%%%s   AA Ac                      d} t          j        t          |           5  t          j         d d d            d S # 1 swxY w Y   d S )Nzocreate_block_manager_from_blocks is deprecated and will be removed in a future version. Use public APIs insteadr&   )r(   r)   r   r    create_block_manager_from_blocks)r+   s    r   0test_create_block_manager_from_blocks_deprecatedr?   H   s    
	?  
	#N#	>	>	> 3 3223 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3s   8<<c                  &   d} t          j        d          }t          j        j                            dg          }t          j        t          |           5  t          j	        
                    ||d           d d d            d S # 1 swxY w Y   d S )NzGmaybe_infer_ndim is deprecated and will be removed in a future version.   r5   r&   )nparanger6   _libsr   BlockPlacementr(   r)   DeprecationWarningr   maybe_infer_ndim)r+   arrbps      r    test_maybe_infer_ndim_deprecatedrJ   T   s    
SC
)A,,C			*	*A3	/	/B		#$6c	B	B	B 3 3&&sB2223 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3s   "BB
B
c                     | j         j        d         }| j                                        }| j                                        }t          |j        |j        j        fg||          }t          j
        ||            d S )Nr   indexcolumns)_mgrr   rM   copyrN   r   r9   mgr_locsas_arrayr(   assert_frame_equal)float_frameblockrM   rN   r    s        r   !test_create_dataframe_from_blocksrV   ]   s    #A&E""$$E!&&((G)
,/	01  F &+.....r
   c                  
   t          j        t          t          dd                    t	          j        dd                              d          dt          j        dgt	          j        g d          t          j        g d	d
          t          d          t          j	        dd          t          j	        dd          
                    d          t          j        dd          t          j        ddd          t          j        g d          t           j                            g d          d          } t!          d | j        j        D             | j        | j                  }t+          j        ||            d S )Nr5      r/      uint8g       @g      @)TFT)TFNboolean)dtypeabc20130101r2   zEurope/Brussels1 day
2012-01-01Dr2   freq)abre   ))r   r5   )r5      )r/   rX   )intuintfloatboolr[   stringdatetime
datetimetz	timedeltaperiodcategoricalintervalc                 4    g | ]}|j         |j        j        fS r	   )r9   rQ   rR   )r   rU   s     r   r   z;test_create_dataframe_from_blocks_types.<locals>.<listcomp>}   s$    MMMU%,/	0MMMr
   rL   )r6   	DataFramelistrangerB   rC   astypenanarrayr7   tz_localizetimedelta_rangeperiod_rangeCategoricalIntervalIndexfrom_tuplesr   rO   r   rM   rN   r(   rS   )dfr    s     r   'test_create_dataframe_from_blocks_typesr   h   si   	a$$IaOO**733263'H00011x 3 3 39EEE5kkj!<<<-
A>>>JJ!  +GQ???olACHHH>///::(445M5M5MNN	
 	

 
B& *MMbgnMMMh
  F
 &"%%%%%r
   c            
      f   t          j        d          t          j        g d          }} t          j        t          j        ddd          t          j        ddd          t          j        d                                        d          }t          j        t          j        d          t          j        d          t          j        d                                        d          }t          |t          j	        dd	g          f|t          j	        ddg          fg| |
          }t          j
        t          j        ddd          t          j        ddd          t          j        ddd          t          j        ddd          d          }t          j        ||           d S )Nr/   )re   rf   cdi  r5      )step)rg   r/   r   rg   rL   
2020-01-01us)r2   unitz1 daysz
2020-01-04z4 days)r6   
RangeIndexIndexrB   rC   rm   ro   reshaper   ry   rt   r7   r{   r(   rS   )rM   rN   block_array1block_array2r    r   s         r   .test_create_dataframe_from_blocks_datetimeliker      s    ]1%%rx0D0D0D'E'E7E9$1%%$1%%""   gfoo	 
 91x1!448;Ma;P;P  gfoo  *
!Q((	)L"(Aq6:J:J+KL  F
 ||QTBBB#HadCCC|QTBBB#HadCCC		
 	
 H &(+++++r
   ry   r   r/   r_   r0   r1   ra   rb   rc   r`   c                     t          j        d| i          }|j        j        d         }t	          |j        d         |j        j        fg|j        |j	                  }t          j        ||           d S )Nre   r   rL   )r6   rt   rO   r   r   r9   rQ   rR   rM   rN   r(   rS   )ry   r   rU   r    s       r   &test_create_dataframe_from_blocks_1dEAr      sr     
sEl	#	#BGN1E)
,q/5>2	34BHbj  F &"%%%%%r
   )!__doc__rm   numpyrB   pytestpandas.errorsr   pandasr6   pandas._testing_testingr(   pandas.api.internalsr   pandas.corer   pandas.core.internalsr   r   r!   markparametrizer,   r<   r?   rJ   rV   r   r   r7   r|   r{   r   r	   r
   r   <module>r      s   
       ( ( ( ( ( (           = = = = = = ! ! ! ! ! ! % % % % % %2 2 2Y Y Y, 
 ! ! !	& 	& 	&	3 	3 	33 3 3/ / /& & &8, , ,: lA...lA%888ac:::7A...	 & & & & &r
   