zmc
2023-08-08 e792e9a60d958b93aef96050644f369feb25d61b
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
U
­ý°dã@sJddlZddlZddlZddlmZddlm    Z    dd„Z
Gdd„dƒZ dS)éN)Úread_sascCs8|jD],}||jt d¡kr|| tj¡||<qdS)NÚint64)ÚcolumnsZdtypeÚnpÚastypeÚfloat64)ÚdataÚv©r
úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/sas/test_xport.pyÚnumeric_as_floats
r c@sšeZdZejdd„ƒZejdd„ƒZejdd„ƒZejdd„ƒZejd    d
„ƒZ    ej
j d d „ƒZ d d„Z dd„Zdd„Zdd„Zdd„Zdd„Zdd„ZdS)Ú    TestXportcCs|ddddƒS)NÚioÚsasrz
DEMO_G.xptr
©ÚselfÚdatapathr
r
r Úfile01szTestXport.file01cCs|ddddƒS)Nrrrz SSHSV1_A.xptr
rr
r
r Úfile02szTestXport.file02cCs|ddddƒS)Nrrrz DRXFCD_G.xptr
rr
r
r Úfile03szTestXport.file03cCs|ddddƒS)Nrrrzpaxraw_d_short.xptr
rr
r
r Úfile04"szTestXport.file04cCs|ddddƒS)Nrrrz DEMO_PUF.cptr
rr
r
r Úfile05&szTestXport.file05c    CsVt | dd¡¡}t|ƒt|dd}t ||¡|jd}t|ddd}| |d¡}W5QRX|jd|ksvt    ‚t|ddd}| d    ¡}W5QRXt ||j
dd    …dd…f¡t|dd    d
}|  ¡}W5QRXt ||j
dd    …dd…f¡d}t|ddd
 }|D]}||jd7}qW5QRX||ks>t    ‚t|ƒ}t ||¡dS) Nú.xptú.csvÚxport©ÚformatrT)rÚiteratorédé
)rÚ    chunksize) ÚpdÚread_csvÚreplacer rÚtmÚassert_frame_equalÚshapeÚreadÚAssertionErrorÚilocÚ    get_chunk)rrÚdata_csvrZnum_rowsÚreaderÚmÚxr
r
r Ú test1_basic*s*  
zTestXport.test1_basicc    CsÔt | dd¡¡}| d¡}t|ƒt|ddd}tj||ddt|dddd    }| d
¡}W5QRXtj||j    d d
…dd…fddt|ddd
d }| 
¡}W5QRXtj||j    d d
…dd…fdddS) NrrÚSEQNr)ÚindexrF©Zcheck_index_typeT)r1rrrr)r1rr ) r!r"r#Ú    set_indexr rr$r%r'r)r*)rrr+rr,r
r
r Ú test1_indexQs
"zTestXport.test1_indexc    Csht | dd¡¡}| d¡}t|ƒt|ddd}t|ƒ}W5QRXtj|dd}tj    ||dd    dS)
Nrrr0iè)r1r r)ZaxisFr2)
r!r"r#r3r rÚlistÚconcatr$r%)rrr+r,Zall_datarr
r
r Útest1_incrementalgs
zTestXport.test1_incrementalcCs2t | dd¡¡}t|ƒt|ƒ}t ||¡dS)Nrr)r!r"r#r rr$r%)rrr+rr
r
r Útest2tszTestXport.test2c    CsLt | dd¡¡}t|ƒt|dƒ}t|dd}W5QRXt ||¡dS)NrrÚrbrr)r!r"r#r Úopenrr$r%)rrr+Úfdrr
r
r Ú test2_binary~s
 zTestXport.test2_binarycCs.t | dd¡¡}t|dd}t ||¡dS)Nrrzutf-8)Úencoding)r!r"r#rr$r%)rrr+rr
r
r Útest_multiple_typesŒs zTestXport.test_multiple_typescCs4t | dd¡¡}t|dd}t | d¡|¡dS)Nrrrrr)r!r"r#rr$r%r)rrr+rr
r
r Útest_truncated_float_support•s z&TestXport.test_truncated_float_supportc    Cs.d}tjt|dt|ddW5QRXdS)Nz<Header record indicates a CPORT file, which is not readable.)Úmatchrr)ÚpytestZraisesÚ
ValueErrorr)rrÚmsgr
r
r Útest_cport_header_found_raises¡sz(TestXport.test_cport_header_found_raisesN)Ú__name__Ú
__module__Ú __qualname__rAZfixturerrrrrÚmarkZslowr/r4r7r8r<r>r?rDr
r
r
r r s&
 
 
 
 
 
& 
     r ) ÚnumpyrrAZpandasr!Zpandas._testingZ_testingr$Zpandas.io.sas.sasreaderrr r r
r
r
r Ú<module>s