U e{jD@s|ddlmZddlZddlZddlmZddlmZddl m Z m Z ddZ d d Z d d Zd dZddZddZddZddZddZddZddZddZdd Zd!d"Zd#Zejeejd$Zd%d&Zd'd(Zd)d*Z d+d,Z!d-d.Z"d/d0Z#d1d2Z$d3d4Z%d5d6Z&d7d8Z'd9d:Z(d;d<Z)d=d>Z*d?d@Z+dAdBZ,dCdDZ-dEdFZ.dGdHZ/dIdJZ0dKdLZ1dMdNZ2dOdPZ3dS)Q)StringION)ascii)InconsistentTableError)assert_almost_equal assert_equalcCst||dS)N)r splitlines)Zarg1Zarg2r e/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/io/ascii/tests/test_fixedwidth.pyassert_equal_splitlinessr cCsfd}tjtjd}||}t|jddgt|dddt|dddt|ddd d S) z Nice, typical fixed format tableb # comment (with blank line above) | Col1 | Col2 | | 1.2 | "hello" | | 2.4 |'s worlds| ReaderCol1Col2rr333333@"hello" 's worldsNr get_reader FixedWidthreadrcolnamesrtablereaderdatr r r test_read_normals rcCsDd}tjtjdd}||}t|jddgt|dddd S) 8Nice, typical fixed format table with col names providedr )name1name2rnamesrr rrrNrrr r r test_read_normal_names#s  r#cCsXd}tjtjddd}||}t|jddgt|ddd t|ddd d S) rzz # comment (with blank line above) | Col1 | Col2 | Col3 | | 1.2 | "hello" | 3 | | 2.4 |'s worlds| 7 | )rr name3)rr$)rr"Z include_namesrr$rrrNrrr r r test_read_normal_names_include1s r&cCsBd}tjtjdd}||}t|jdgt|ddddS) z7Nice, typical fixed format table with col name excludedr )r)rZ exclude_namesrrrrN)rrrrrrrr r r test_read_normal_excludeDs  r'cCsfd}tjtjd}||}t|jddgt|dddt|dddt|ddd d S) z8Weird input table with data values chopped by col extentz< Col1 | Col2 | 1.2 "hello" 2.4 sdf's worlds r rrrrrz"helzdf's woNrrr r r test_read_weirdRs r(cCsbd}tj|tjdd}tt|jjdt|dddt|dddt|d d d d S) zTable with double delimitersz || Name || Phone || TCP|| | John | 555-1234 |192.168.1.10X| | Mary | 555-2134 |192.168.1.12X| | Bob | 555-4527 | 192.168.1.9X| F)rguessNameZPhoneZTCPrrMary555-1234 192.168.1.9Nrrrrtupledtyper"rrr r r test_read_doubleas r4cCsdd}tj|tjddd}tt|jjdt|dddt|ddd t|d d d d S) zTable with space delimiterz Name --Phone- ----TCP----- John 555-1234 192.168.1.10 Mary 555-2134 192.168.1.12 Bob 555-4527 192.168.1.9 F )rr) delimiter)r+z--Phone-z ----TCP-----rrr,r-r.r/Nr0r3r r r test_read_space_delimiterps r7cCsfd}tj|tjdddd}tt|jjdt|dddt|ddd t|d d d dS) /Table with no header row and auto-column namingj | John | 555-1234 |192.168.1.10| | Mary | 555-2134 |192.168.1.12| | Bob | 555-4527 | 192.168.1.9| FNr)rr) header_start data_startZcol1Zcol2Zcol3rr,r-r.r/r0r3r r r test_read_no_header_autocolumnsr=cCshd}tj|tjddddd}tt|jjdt|dddt|ddd t|d d d dS) xTable with no header row and with col names provided. Second and third rows also have hanging spaces after final |.r9FNrr*)rr)r:r;r"rr,r-r.r/r0r3r r r test_read_no_header_namessr?cCs`d}tj|tjd}tt|jjdt|dddt|dddt|ddd d S) r8r9r r<rrr,r-r.r/NrrFixedWidthNoHeaderrr1r2r"r3r r r 'test_read_no_header_autocolumn_NoHeaders rBcCsbd}tj|tjdd}tt|jjdt|dddt|dddt|ddd d S) r>r9r*r!rrr,r-r.r/Nr@r3r r r "test_read_no_header_names_NoHeadersrCcCsxd}tj|tjdddd}tt|jjdt|dddt|ddd t|dd d t|d d d d S)zCTable with no delimiter with column start and end values specified.z # 5 9 17 18 28 # | | || | John 555- 1234 192.168.1.10 Mary 555- 2134 192.168.1.12 Bob 555- 4527 192.168.1.9 r*)r ))rr" col_startscol_endsrr 555- 1234r,r.z 192.168.1.z 192.168.1Nr@r3r r r test_read_col_startssrLcCsd}ddiddifD]x}tj|ftjdd|}tt|jjdt|dd d t|d dd t|d d d t|dd dqdS)zFTable with no delimiter with only column start or end values specifiedaC #1 9 19 <== Column start indexes #| | | <== Column start positions #<------><--------><-------------> <== Inferred column positions John 555- 1234 192.168.1.10 Mary 555- 2134 192.168.1.123 Bob 555- 4527 192.168.1.9 Bill 555-9875 192.255.255.255 rI)rrDrJ)rE!r*r!rrrKr,r.z 192.168.1.123r%z192.255.255.255Nr@)rkwargsrr r r #test_read_detect_col_starts_or_endss rQz| Col1 | Col2 | Col3 | Col4 | | 1.2 | "hello" | 1 | a | | 2.4 | 's worlds | 2 | 2 | r cCs*t}tjt|tjdt|ddS),Write a table as a normal fixed width table.Writerzi| Col1 | Col2 | Col3 | Col4 | | 1.2 | "hello" | 1 | a | | 2.4 | 's worlds | 2 | 2 | Nrrwriterrr getvalueoutr r r test_write_normals rZcCs,t}tjt|tjddt|ddS)rR)azN/A)rTZ fill_valueszi| Col1 | Col2 | Col3 | Col4 | | 1.2 | "hello" | 1 | N/A | | 2.4 | 's worlds | 2 | 2 | NrUrXr r r test_write_fill_values s r\cCs,t}tjt|tjddt|ddS)5Write a table as a fixed width table with no padding.NrT delimiter_padzQ|Col1| Col2|Col3|Col4| | 1.2| "hello"| 1| a| | 2.4|'s worlds| 2| 2| rUrXr r r test_write_no_pads r`cCs,t}tjt|tjddt|ddS)5Write a table as a fixed width table with no bookend.FrTbookendz]Col1 | Col2 | Col3 | Col4 1.2 | "hello" | 1 | a 2.4 | 's worlds | 2 | 2 NrUrXr r r test_write_no_bookend&s rdcCs.t}tjt|tjdddt|ddS)7Write a table as a fixed width table with no delimiter.FNrTrcr6zTCol1 Col2 Col3 Col4 1.2 "hello" 1 a 2.4 's worlds 2 2 rUrXr r r test_write_no_delimiter4s rgcCs*t}tjt|tjdt|ddS)rRrSz8| 1.2 | "hello" | 1 | a | | 2.4 | 's worlds | 2 | 2 | NrrrVrrAr rWrXr r r test_write_noheader_normalBs ricCs,t}tjt|tjddt|ddS)r]Nr^z(|1.2| "hello"|1|a| |2.4|'s worlds|2|2| rhrXr r r test_write_noheader_no_padOs rjcCs,t}tjt|tjddt|ddS)raFrbz01.2 | "hello" | 1 | a 2.4 | 's worlds | 2 | 2 NrhrXr r r test_write_noheader_no_bookend\s rkcCs.t}tjt|tjdddt|ddS)reFNrfz*1.2 "hello" 1 a 2.4 's worlds 2 2 rhrXr r r test_write_noheader_no_delimiterisrlcCs2t}tjt|tjddddt|ddS)rez%-8.3fz%-15srr)rTformatsz| Col1 | Col2 | Col3 | Col4 | | 1.200 | "hello" | 1 | a | | 2.400 | 's worlds | 2 | 2 | NrUrXr r r test_write_formatsxsrocCs\d}tj|tjd}t|jjdt|dddt|dddt|dddd S) zjTypical fixed format table with two header lines (with some cruft thrown in to test column positioningzD Col1 Col2 ---- --------- 1.2xx"hello" 2.4 's worlds r rmrrrrrNrrFixedWidthTwoLinerr2r"rr3r r r test_read_twoline_normals rrcCsbd}tj|tjdddd}t|jjdt|dddt|ddd t|ddd d S) zRead restructured text tablezn ======= =========== Col1 Col2 ======= =========== 1.2 "hello" 2.4 's worlds ======= =========== rr.)rr: position_linedata_endrmrrrrNrpr3r r r test_read_twoline_ReSTsrvc Csfd}tj|tjdddddd}t|jjdt|ddd t|ddd t|ddd d S) z\Read text table designed for humans and test having position line before the header linezy +------+----------+ | Col1 | Col2 | +------|----------+ | 1.2 | "hello" | | 2.4 | 's worlds| +------+----------+ +rrr%rs)rr6r:rtr;rurmrrrNrpr3r r r test_read_twoline_humans rxc CsDd}tt}tj|tjdddW5QRXdt|jks@tdS)zTest failure if too many different character are on position line. The position line shall consist of only one character in addition to the delimiter. zQ | Col1 | Col2 | |------|==========| | 1.2 | "hello" | | 2.4 | 's worlds| |Frr6r)zDPosition line should only contain delimiters and one other characterN pytestraisesrrrrqstrvalueAssertionErrorrexcinfor r r test_read_twoline_fails rc CsDd}tt}tj|tjdddW5QRXdt|jks@tdS)zTest failure when position line uses characters prone to ambiguity Characters in position line must be part an allowed set because normal letters or numbers will lead to ambiguous tables. zQ | Col1 | Col2 | |aaaaaa|aaaaaaaaaa| | 1.2 | "hello" | | 2.4 | 's worlds| ryFrzz(Characters in position line must be partNr{rr r r test_read_twoline_wrong_markers rcCs*t}tjt|tjdt|ddS)rRrSzdCol1 Col2 Col3 Col4 ---- --------- ---- ---- 1.2 "hello" 1 a 2.4 's worlds 2 2 NrrrVrrqr rWrXr r r test_write_twoline_normals rcCs.t}tjt|tjdddt|ddS)r]r5=)rTr_Z position_charz|Col1 Col2 Col3 Col4 ==== ========= ==== ==== 1.2 "hello" 1 a 2.4 's worlds 2 2 NrrXr r r test_write_twoline_no_padsrcCs.t}tjt|tjdddt|ddS)raTryrfzl|Col1| Col2|Col3|Col4| |----|---------|----|----| | 1.2| "hello"| 1| a| | 2.4|'s worlds| 2| 2| NrrXr r r test_write_twoline_no_bookends rcCsd}dddg}tj|ddd|dd }|j|ks2tt|ddd gksLtt|dd d gksftt|ddd gkstdS)z3Test fix in #8511 where data_start is being ignoredzAAA y z 1 2 3 4 5 6 7 8 9 r[bcrr%r5fixed_width_no_header)r;rur6r"formatr.rFN)rrrrnpall)tblr"rr r r !test_fixedwidthnoheader_splittings rcCsfddddddddg}d d d d d g}tj|dd|d}t}tj||dd|d||ksbtdS)Nz&| int16 | float32 | sV