U
|
ý°d& ã @ sr d dl Zd dlZd dlm mZ d dlmZm Z d dl
|
m Z d dl mZ ejZG dd dZdd Zdd ZdS )
|
é N)Ú DataFrameÚ
|
MultiIndex)Ú PandasArrayc @ s e Zd Zdd Zdd ZdS )ÚTestToDictOfBlocksc C s| t |dd}|jd }d }|jdd}| ¡ D ]*}|}||kr.|| d |jd d
|f< q.|d k rt|| || ¡rxtd S )NT©Úcopyr é ©r ÚcolumnsÚ_to_dict_of_blocksÚvaluesÚlocÚequalsÚAssertionError)ÚselfÚ float_frameÚdfÚcolumnÚ_last_dfÚblocksÚ_df© r úhd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_to_dict_of_blocks.pyÚtest_copy_blocks s
|
z#TestToDictOfBlocks.test_copy_blocksc C s t |dd}|jd }d }|jdd}| ¡ D ]*}|}||kr.|| d |jd d
|f< q.|s~|d k rx|| || ¡stn |d k r|| || ¡rtd S )NTr r Fr r )r r Úusing_copy_on_writer r r r r r r r Útest_no_copy_blocks! s
|
z&TestToDictOfBlocks.test_no_copy_blocksN)Ú__name__Ú
|
__module__Ú __qualname__r r r r r r r s r c C sÐ |r| j tjjdd¡ tddddgddd d
|
gd }ttjddd dgt d |d < |j
|
}t |j dksnt |d }| ¡ |r d |jd< |jd dksÌt n,d |jd< |jd d ks¼t |d |ksÌt d S )NzCoW - not yet implemented)Úreasonr é é é ÚaÚbÚcÚd)r# r$ )ZdtypeZfoor )r r$ )ÚnodeZ
|
add_markerÚpytestÚmarkZxfailr r ÚnpÚarrayÚobjectZ_mgrÚlenr r r r r )Úrequestr r ZmgrZserr r r Ú!test_to_dict_of_blocks_item_cache5 s
|
|
r/ c C s´ t dddg¡} tdddgddd gg| d
|
}|d d |d < | ¡ }t| ¡ d dgks\tt |d tddgddgg| d d
d
|
¡ t |d tdgd gg| dd
d
|
¡ d S )N)Z1str# )Ú2ndr$ )Z3rdr% g ð?r r! g @é é )r
|
r0 g @Úfloat64Úint64g $@) r Ú from_tuplesr r ÚsortedÚkeysr ÚtmZassert_frame_equal)Úcolsr r r r r Útest_set_change_dtype_sliceO s ÿr: )Únumpyr* r( Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr r Zpandas._testingZ_testingr8 Zpandas.core.arraysr Z skip_array_manager_invalid_testZ
|
pytestmarkr r/ r: r r r r Ú<module> s %
|
