U
|
ý°dé ã @ sæ d dl Zd dlZd dlmZmZmZ d dlmZ d dl
|
m Z ej j ddd dd gdd gd
|
d d Zej j dd d dd gdd gd
|
dd Zdd Zej dddg¡dd Zdd Zdd Zd d Zd d Zd d! Zd"d# ZdS )$é N)Ú DataFrameÚSeriesÚ
|
date_range)Ú get_arrayÚmethodc C s | j S ©N©Úvalues©Úser© r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/copy_view/test_array.pyÚ<lambda> ó r c C s
|
t | ¡S r ©ÚnpÚasarrayr
|
r r r r r r r )Úidsc C sÆ t dddgdd}| ¡ }||}| rt |t|d¡s<t|jjdksLttj t
|
dd d |d < W 5 Q R X t ||¡ d |j d < |jd d ksÂtn*|jjd
|
ks¨td |d < |j d d ksÂtd S ) Né é é Úname©r Fú read-only©Úmatchr T)r Úcopyr Ú shares_memoryr ÚAssertionErrorÚflagsÚ writeableÚpytestÚraisesÚ
|
ValueErrorÚtmÚassert_series_equalÚilocr )Úusing_copy_on_writer r Úser_origÚarrr r r Útest_series_values s
|
r* c C s | j S r r ©Údfr r r r 0 r c C s
|
t | ¡S r r r+ r r r r 0 r c C sà t dddgdddgd}| ¡ }||}| r t |t|d¡sDt|jjd ksTttj t
|
d
|
d d |d < W 5 Q R X t ||¡ d |j d < |jd d ksÜtn<|jjdks°td |d < |sÐ|j d d ksÜtn t ||¡ d S )Nr r r é é é ©ÚaÚbr1 Fr r r )r r T)r r r r r r r r r! r" r# r$ Zassert_frame_equalr& r )r' Úusing_array_managerr r, Zdf_origr) r r r Útest_dataframe_values. s
|
r4 c C sT t dddgdd}| ¡ }| ¡ }| rt |t|d¡s<t|jjdksLtt j
|
t dd d |d < W 5 Q R X t ||¡ d |jd < |jd d ksÂtn*|jjd
|
ks¨td |d < |jd d ksÂtt dddgdd}|jd
|
d }t |t|d¡röt|jjd
|
kstt dddgdd}|jd d }t |t|d¡r>t|jjd
|
ksPtd S )Nr r r r r Fr r r T)r Úfloat64©Zdtype)r r Zto_numpyr r r r r r r! r" r# r$ r% r& r )r' r r( r) r r r Útest_series_to_numpyO s,
|
r7 ÚorderÚFÚCc C sF t dddg}|j|d}| r.|jjdks.tt t||¡sBtd S )Nr r r )r8 F)r Zravelr r r r r r )r' r8 r r) r r r Útest_ravel_read_onlyt s
|
r; c C s t dddgdd}tj|dd}t |t|¡s4t| rJ|jjdksZtn|jjdksZtt |¡}t |t|¡rxt|jjdkstd S ) Nr r r ÚInt64r6 Úint64FT)r r r r r r r r )r' r r) r r r Útest_series_array_ea_dtypes} s
|
r> c C s~ t ddddgidd}tj|dd}t |t|d¡r:t|jjdksJtt |¡}| rj|jjd ksztn|jjdksztd S )
|
Nr1 r r r r< r6 r= TF©r r r r r r r r ©r' r, r) r r r Ú test_dataframe_array_ea_dtypes s
|
rA c C sb t dddgidd}t |¡}|s8t |t|d¡s8t| rN|jjdks^tn|jjdks^td S )Nr1 r2 Ústringr6 FTr? )r' r3 r, r) r r r Ú!test_dataframe_array_string_dtype s
|
rC c C sH t dddgdd} t | ¡}t |t| d¡r4t|jjdksDtd S )Nr r r g ø?r0 r1 Tr? ©r, r) r r r Ú$test_dataframe_multiple_numpy_dtypes¦ s
|
rE c C sH t dtdddi}t |¡}| r4|jjdksDtn|jjdksDtd S )Nr1 z
|
2012-01-01r )ZperiodsFT)r r r r r r r r@ r r r Útest_values_is_ea s
|
|
rF c C s$ t } t | ¡}|jjdks td S )NT)r r r r r r rD r r r Útest_empty_dataframe¶ s
|
rG )Únumpyr r! Zpandasr r r Zpandas._testingZ_testingr$ Zpandas.tests.copy_view.utilr ÚmarkZ parametrizer* r4 r7 r; r> rA rC rE rF rG r r r r Ú<module> s2 ý
|
ý
|
%
|
|
