zmc
2023-08-08 e792e9a60d958b93aef96050644f369feb25d61b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
U
­ý°dã@sNddlmZddlZddlZddlmZmZmZddl    m
Z Gdd„dƒZ dS)é)ÚStringION)Ú    DataFrameÚconcatÚread_csvc@s,eZdZdd„Zdd„Zdd„Zdd„Zd    S)
ÚTestInvalidConcatc
Cs\t dd¡}diddgdfD]:}dt|ƒ›d}tjt|dt||gƒW5QRXqdS)Né
éé)r    rz#cannot concatenate object of type 'z+'; only Series and DataFrame objs are valid©Úmatch)ÚtmÚmakeCustomDataframeÚtypeÚpytestÚraisesÚ    TypeErrorr)ÚselfÚdf1ÚobjÚmsg©rú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/concat/test_invalid.pyÚtest_concat_invalids  ÿz%TestInvalidConcat.test_concat_invalidc    Cs6t dd¡}d}tjt|dt|ƒW5QRXdS)Nrrz^first argument must be an iterable of pandas objects, you passed an object of type "DataFrame"r
)r r rrrr)rrrrrrÚ"test_concat_invalid_first_arguments
 ÿz4TestInvalidConcat.test_concat_invalid_first_argumentcCstdd„tdƒDƒƒdS)Ncss |]}ttj dd¡ƒVqdS)éN)rÚnpÚrandomZrand)Ú.0Ú_rrrÚ    <genexpr>%sz>TestInvalidConcat.test_concat_generator_obj.<locals>.<genexpr>é)rÚrange)rrrrÚtest_concat_generator_obj#sz+TestInvalidConcat.test_concat_generator_objc    CsHd}tt|ƒdd}t|dd}W5QRXtt|ƒƒ}t ||¡dS)Nzäindex,A,B,C,D
                  foo,2,3,4,5
                  bar,7,8,9,10
                  baz,12,13,14,15
                  qux,12,13,14,15
                  foo2,12,13,14,15
                  bar2,12,13,14,15
               r    )Ú    chunksizeT)Z ignore_index)rrrr Zassert_frame_equal)rÚdataÚreaderÚresultÚexpectedrrrÚtest_concat_textreader_obj's
     z,TestInvalidConcat.test_concat_textreader_objN)Ú__name__Ú
__module__Ú __qualname__rrr"r(rrrrrs     r) ÚiorÚnumpyrrZpandasrrrZpandas._testingZ_testingr rrrrrÚ<module>s