o
    ҷh	                     @   sP   d Z ddlmZ ddlZddlmZ ddlm	Z	m
Z
mZmZ dd Zdd ZdS )	zQ
Testing interaction between the different managers (BlockManager, ArrayManager)
    array_equivalentN)ArrayManagerBlockManagerSingleArrayManagerSingleBlockManagerc                  C   s|  t dd t g dg dg dd} W d    n1 s w   Y  t| jts-J t dd t g dg dg dd}W d    n1 sMw   Y  t|jtsZJ t| | | 	d}t|jtsmJ | 	d}t|jtszJ t||  t
dd	 t|jj|jjD sJ |	d}t|jtsJ |	d}t|jtsJ t|| t|jjd
ksJ d S )Nmode.data_managerblock         )g?g?g333333?)         abcarrayc                 s   s    | ]
\}}t ||V  qd S )Nr   ).0leftright r   W/var/www/html/venv/lib/python3.10/site-packages/pandas/tests/internals/test_managers.py	<genexpr>"   s
    
z*test_dataframe_creation.<locals>.<genexpr>r   )pdoption_context	DataFrame
isinstance_mgrr   r   tmassert_frame_equal_as_managerallziparrayslenblocks)df_blockdf_arrayresultr   r   r   test_dataframe_creation   s.     



r,   c                  C   s8  t dd t jg ddg dd} W d    n1 sw   Y  t| jts*J t dd t jg ddg dd}W d    n1 sGw   Y  t|jtsTJ t| | | 	d}t|jtsgJ | 	d}t|jtstJ t||  |	d}t|jtsJ |	d}t|jtsJ t|| d S )Nr   r	   r
   Ar   )nameindexr   )
r   r   Seriesr   r    r   r   r!   assert_series_equalr#   )s_blocks_arrayr+   r   r   r   test_series_creation/   s&   



r4   )__doc__pandas.core.dtypes.missingr   pandasr   pandas._testing_testingr!   pandas.core.internalsr   r   r   r   r,   r4   r   r   r   r   <module>   s    