a
    j=icß	  ã                   @   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@0    Y  t| jtƒsZJ ‚t  dd¡. t  g d¢g d¢g d¢dœ¡}W d   ƒ n1 sš0    Y  t|jtƒs´J ‚t | |¡ |  	d¡}t|jtƒsÚJ ‚|  	d¡}t|jtƒsôJ ‚t || ¡ t
dd	„ t|jj|jjƒD ƒƒs&J ‚| 	d¡}t|jtƒsBJ ‚| 	d¡}t|jtƒs^J ‚t ||¡ t|jjƒd
ks€J ‚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   úu/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/pandas/tests/internals/test_managers.pyÚ	<genexpr>#   s   ÿz*test_dataframe_creation.<locals>.<genexpr>r   )ÚpdÚoption_contextZ	DataFrameÚ
isinstanceÚ_mgrr   r   ÚtmZassert_frame_equalÚ_as_managerÚallÚzipZarraysÚlenÚblocks)Zdf_blockZdf_arrayÚresultr   r   r   Útest_dataframe_creation   s*    <<

þ

r'   c                  C   s<  t  dd¡( t jg d¢dg d¢d} W d   ƒ n1 s:0    Y  t| jtƒsTJ ‚t  dd¡( t jg d¢dg d¢d}W d   ƒ n1 sŽ0    Y  t|jtƒs¨J ‚t | |¡ |  	d¡}t|jtƒsÎJ ‚|  	d¡}t|jtƒsèJ ‚t || ¡ | 	d¡}t|jtƒsJ ‚| 	d¡}t|jtƒs,J ‚t ||¡ d S )Nr   r	   r
   ÚAr   )ÚnameÚindexr   )
r   r   ZSeriesr   r   r   r   r    Zassert_series_equalr!   )Zs_blockZs_arrayr&   r   r   r   Útest_series_creation0   s"    66



r+   )Ú__doc__Zpandas.core.dtypes.missingr   Zpandasr   Zpandas._testingZ_testingr    Zpandas.core.internalsr   r   r   r   r'   r+   r   r   r   r   Ú<module>   s    