1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
| U
| ý°d» ã @ sä d Z ddlmZ ddlZddlmZ ddlmZ ej
| d¡Z e dd Z e dd Z d
| d Ze ej d d edddggdddgdfdeddddddggddddddgdfd eddddddd ggdd d ddd d!gdfg¡d"d# Ze ej d$d%dddgedd&d&gd'd(d)ggdddgdfd*ddddddgedd&d+d,d-d.gd'd(d)d/d0d1ggdddddd2gdfd3ddd ddddgedd&d+d,d-d.d4gd'd(d)d/d0d1d5ggddd ddd6d7gdfg¡d8d9 Ze d:d; Ze d<d= Ze d>d? ZdS )@zÖ
| Tests that duplicate columns are handled appropriately when parsed by the
| CSV engine. In general, the expected result is that they are either thoroughly
| de-duplicated (if mangling requested) or ignored otherwise.
| é )ÚStringION)Ú DataFrameZ pyarrow_skipc C sL | }d}|j t|dd}tdddddggd d
| d d d gd}t ||¡ d S )Nza,a,b,b,b
| 1,2,3,4,5ú,)Úsepé é é é é Úaúa.1Úbzb.1zb.2©Úcolumns©Úread_csvr r ÚtmÚassert_frame_equal©Ú all_parsersÚparserÚdataÚresultÚexpected© r ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/parser/test_mangle_dupes.pyÚ
| test_basic s
| "r c C sH | }d}t dddgdddggdd d
| gd }| t|¡}t ||¡ d S ) Nza,b,a
| 0,1,2
| 3,4,5r r r r r r
| r r r r )r r r r r )r r r r r r r r Útest_basic_names s
| "r c C s> | }d}t jtdd |jt|dddgd W 5 Q R X d S )Nz 0,1,2
| 3,4,5úDuplicate names©Úmatchr r ©Únames©ÚpytestZraisesÚ
| ValueErrorr r )r r r r r r Útest_basic_names_raise' s r& z data,expectedz a,a,a.1
| 1,2,3r r r r úa.2r r z+a,a,a.1,a.1.1,a.1.1.1,a.1.1.1.1
| 1,2,3,4,5,6r r
| é úa.1.1za.1.1.1z a.1.1.1.1z!a,a,a.3,a.1,a.2,a,a
| 1,2,3,4,5,6,7é úa.4úa.3za.5za.6c C s" | }| t|¡}t ||¡ d S )N)r r r r )r r r r r r r r Ú test_thorough_mangle_columns0 s r- zdata,names,expectedz a,b,b
| 1,2,3r Ú1Ú2Ú3za,b,c,d,e,f
| 1,2,3,4,5,6ÚcÚdÚeÚfÚ4Ú5Ú6z a.1.1.1.1.1za,b,c,d,e,f,g
| 1,2,3,4,5,6,7ÚgÚ7za.2.1za.3.1c C s4 | }t jtdd |jt||d W 5 Q R X d S )Nr r r! r# )r r r" r r r r r Útest_thorough_mangle_namesM s "r: c C s d}| }dddg}t ||i}tdD ]t}t }t|d D ]6}ddd| t|d }|jd|dddgd q<|||< | t| ¡ ¡}t ||¡ q&d S ) NÚ0r r r ú
| Unnamed: 0Ú.r )ÚlocÚcolumnÚvalue) r ÚrangeÚminÚinsertr r Zto_csvr r ) r Zorig_keyr Z
| orig_valueZdfÚir ÚjZcol_namer r r Ú!test_mangled_unnamed_placeholdersu s
| rF c C sP | }d}| t|¡}tdddddddggd d
| d d d ddgd}t ||¡ d S )Nz%a,a,a.1,a,a.3,a.1,a.1.1
| 1,2,3,4,5,6,7r r r r r
| r( r* r r' r r+ r, za.1.2r) r r r r r r Ú$test_mangle_dupe_cols_already_exists s þrG c C sD | }d}| t|¡}tddddggdddd gd
| }t ||¡ d S ) Nz ,Unnamed: 0,,Unnamed: 2
| 1,2,3,4r r r r z Unnamed: 0.1r< z Unnamed: 2.1z
| Unnamed: 2r r r r r r Ú0test_mangle_dupe_cols_already_exists_unnamed_col s
| þrH )Ú__doc__Úior r$ Zpandasr Zpandas._testingZ_testingr ÚmarkZ usefixturesZ skip_pyarrowr r r& Z parametrizer- r: rF rG rH r r r r Ú<module> s
|
|
| þþþþ÷þ ÿý þý þûýðþ
|
|
|
|
