U
|
ý°d# ã @ s0 d dl Z d dlZd dlZd dlZd dlmZ dd Z ej
|
de ddfg dfe ddfg¡dd Z ej
|
d
|
d d d e d ¡e d¡g¡dd Zdd Zdd Zdd Zej
|
dddgddggd d gd d d gddddgd d d d gd d d d d gfddgddggd d gd ejd d gd d ddddgd d d d gd ejd d d d gd d fddgddggd d gd ejjd d gddggd!d"gd#ddddgd d d d gd ejjd d d d gddddggd!d"gd#fddgddggd d gd ejjd d gddggd dgd#ddddgd d d d gd ejjd d d d gddddggd dgd#fg¡d$d% Zd&d' Zd(d) Zej
|
d*e dejd ddejejddejge d+e d,dd d-d.d ejd/d0ejgd1e d+fe d2ejd ddejejddejge d+e d,dd d-d.gd d-d.gd d-d.gd g d/d0gd/d0gejgd1e d+fg¡d3d4 Z d5d6 Z dS )7é Nc C s¸ t t jdddgtjg dgtdddd¡} tjtdd | td
|
g¡ W 5 Q R X tjtd d | td
|
¡ W 5 Q R X td
|
| _
|
tjtt d ¡d | d ¡ W 5 Q R X d S )Nr é é ©é é Úabcd©Úindex©ÚAÚBz/column must be a scalar, tuple, or list thereof©ÚmatchZAAzcolumn must be uniquez:DataFrame columns must be unique. Duplicate columns: ['A']r ) ÚpdÚ DataFrameÚSeriesÚnpÚnanÚlistÚpytestÚraisesÚ
|
ValueErrorÚexplodeÚcolumnsÚreÚescape)Údf© r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_explode.pyÚ
|
test_error
|
s &ÿ ÿ
|
þr zinput_subset, error_messageZACz)columns must have matching element countszcolumn must be nonemptyc C sf t jdddgtjg dgddddgdg d d
|
d ggd td d}tjt|d | | ¡ W 5 Q R X d S )Nr r r r ÚaÚbÚcÚfooÚdÚeÚf©r r ÚCr r r ) r r r r r r r r r )Ú input_subsetÚ error_messager r r r Útest_error_multi_columns s ýúr+ Úscalarr g ø?z1 daysz
|
2019-12-31c C s~ t | t jdddgtjg dgtddddi¡}| | ¡}t | t jdddtjtjdd gtd
|
td ddi¡}t ||¡ d S ) Nr r r r r r r r r Zaaabcdd©r Údtype)
|
r r r r r r r ÚobjectÚtmÚassert_frame_equal)r, r ÚresultÚexpectedr r r Ú
|
test_basic= s (ÿ
|
ÿ üÿr4 c C s t jtjdddgtjg dgtdddt j ddd d
|
g¡d } | d ¡}t t j dddtjtjd dgt j ddddd d
|
d
|
g¡tddd¡}t
|
||¡ d S )Nr r r r ©r. r
|
)r r )r r )r! r )r! r r r r r r- ) r r r Úarrayr r/ Ú
|
MultiIndexÚ from_tuplesr r r0 r1 ©r r2 r3 r r r Útest_multi_index_rowsQ s0 "þ
|
ùÿ óðÿr: c C s t tjdddgtjg dgtddd¡} | d¡}t t jdddtjtjdd gt dddddddg¡td
|
dd¡}t
|
||¡ d S ) Nr r r r r5 )©r r )r r r; r r r- ) r r r r6 r r/ r r ÚIndexr0 r1 r9 r r r Útest_multi_index_columnso s "ÿ
|
ýúÿ
|
r= c C s. t jdtddgdtddggtdd d ¡} | d
|
¡}t jddddddddgtjd d d ddd d d gtdddddddddgdtdd d ¡}t
|
||¡ t jdddgdddggdddgd} | j | j j d¡d d¡}t jddd gddd
|
gddd gddd ggdddgd d d d gd }t
|
||¡ d S ) Né é é
|
é r é ÚABC)r r( r r r r r r5 r' z
|
2014-01-01ZAlicezA Bz
|
2014-01-02ZBobzC DÚdtÚnameÚtextú )rF r ÚD)r r )r r Úranger Z set_indexr r r6 r/ r0 r1 ZassignrF ÚstrÚsplitr9 r r r Ú test_usecase s@ ÿþ
|
ýúù þ ü
|
ø
|
rL z6input_dict, input_index, expected_dict, expected_indexr r r r r# Úbar)Úcol1Zcol2Zmy_index)rE Zmy_first_indexZmy_second_index)Únamesc C s8 t j| |d}| d¡}t j||td}t ||¡ d S )Nr rN r- )r r r r/ r0 r1 )Z
|
input_dictZ input_indexÚ expected_dictÚexpected_indexr r2 r3 r r r Útest_duplicate_index¨ s &
|
rR c C sj t tdddtdtdgd¡} | jddd }t jddddgtd
|
ddd d d gd}t ||¡ d S )Nr rB r@ ÚabZcd)ÚidÚvaluesrU T)Z ignore_indexr r r r r )r r rI r r r0 r1 r9 r r r Útest_ignore_indexÔ s $
|
ÿrV c C sb t jddhgdgddgd} | jddjdd}t jddgddgdddgd}t ||¡ d S ) NÚxÚyr )r r! r r )Úcolumn)Zby)r r r Z sort_valuesr0 r1 r9 r r r Útest_explode_setsÞ s rZ z+input_subset, expected_dict, expected_indexZaaabcdder- r! r" r$ r% r' r c C sj t jdddgtjg dtjgddddgdg d d
|
gtjgd td d }| | ¡}t ||¡}t ||¡ d S )Nr r r r r r! r" r# r$ r% r' Zabcder )r r r r r r r0 r1 )r) rP rQ r r2 r3 r r r Útest_multi_columnsæ s *ýú
|
r[ c
|
C s¼ t ddgdgg ddggdddd gddgtjg dd
|
ggd ¡} | d d g¡}t jtjdddtjddgtdddddd d gtjddtjtjdd
|
gtdd ddddddgd}t ||¡ d S )Nr r r? r r é é é é r r' r r( r5 r ) r r r r r r6 r/ r0 r1 r9 r r r Ú test_multi_columns_nan_empty s
|
ýÿ ýúr` ) r Únumpyr r Zpandasr Zpandas._testingZ_testingr0 r ÚmarkZ parametrizer r+ Z TimedeltaÚ Timestampr4 r: r= rL r< r7 Z from_arraysrR rV rZ r r r/ r[ r` r r r r Ú<module> sÆ þþþ÷þ
|
þ
|
&
|
üü ÿþú ÿüèþ$
|
|
ýù õýøùìòþ(
|
|
