HR-eDTddlmZddlZddlZddlmZddlmZddl m Z m Z dZ dZ d Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZejeejZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&d Z'd!Z(d"Z)d#Z*d$Z+d%Z,d&Z-d'Z.d(Z/d)Z0d*Z1d+Z2d,Z3dS)-)StringION)ascii)InconsistentTableError)assert_almost_equal assert_equalcnt||dS)N)r splitlines)arg1arg2s Flib/python3.11/site-packages/astropy/io/ascii/tests/test_fixedwidth.pyassert_equal_splitlinesrs,""DOO$5$566666cHd}tjtj}||}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 FixedWidthreadrcolnamesrtablereaderdats r test_read_normalr"s E  U%5 6 6 6F ++e  C/000Aq 3'''QI&&&QK(((((rcd}tjtjd}||}t |jddgt |dddd S) 8Nice, typical fixed format table with col names providedr)name1name2rnamesr%r&rrrNrrs r test_read_normal_namesr)#sj E  U%5=O P P PF ++e  C1222Aq 3'''''rcd}tjtjdd}||}t |jddgt |ddd t |ddd d S) r$zz # comment (with blank line above) | Col1 | Col2 | Col3 | | 1.2 | "hello" | 3 | | 2.4 |'s worlds| 7 | )r%r&name3)r%r+)rr( include_namesr%r+rrrNrrs r test_read_normal_names_includer.1s E  )(F ++e  C1222Aq 3'''QArcd}tjtjd}||}t |jdgt |ddddS) z7Nice, typical fixed format table with col name excludedr)r)r exclude_namesrrrrN)rrrrrrrs r test_read_normal_excluder1Dsf E  U%5Y O O OF ++e  Cx(((QK(((((rcHd}tjtj}||}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 rrrrrrz"helzdf's woNrrs r test_read_weirdr3Rs E  U%5 6 6 6F ++e  C/000Aq 3'''QF###QI&&&&&rcBd}tj|tjd}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)rguessNamePhoneTCPrrMary555-1234 192.168.1.9Nrrrrtupledtyper(rr!s r test_read_doublerBas E *U5#35 A A ACsy'')ABBBQF###QJ'''QM*****rcDd}tj|tjdd}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 )rr5 delimiter)r7z--Phone-z ----TCP-----rrr:r;r<r=Nr>rAs r test_read_space_delimiterrFps E *U5#35C P P PCsy'')MNNNQF###QJ'''QM*****rcFd}tj|tjddd}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)rr5 header_start data_startcol1col2col3rr:r;r<r=r>rAs r test_read_no_header_autocolumnrPs E * e&e$ST   Csy'')ABBBQF###QJ'''QM*****rcHd}tj|tjdddd}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 |.rIFNrr6)rr5rJrKr(rr:r;r<r=r>rAs r test_read_no_header_namesrSs E * &    Csy'')ABBBQF###QJ'''QM*****rc@d}tj|tj}tt |jjdt|dddt|dddt|ddd d S) rHrIrrLrrr:r;r<r=NrrFixedWidthNoHeaderrr?r@r(rAs r 'test_read_no_header_autocolumn_NoHeaderrWs E *U5#; < < <--------><-------------> <== 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 r`)rr[ra)r\!r6r'rrrbr:r<z 192.168.1.123r-z192.255.255.255NrU)rkwargsr!s r #test_read_detect_col_starts_or_endsris E!*- K/HI 3 3j  +*      U39?++-EFFFSVAY ,,,SVAY'''SVAY000SVAY 12222 3 3rz| Col1 | Col2 | Col3 | Col4 | | 1.2 | "hello" | 1 | a | | 2.4 | 's worlds | 2 | 2 | rct}tjt|tjt |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 | Nrrwriter!rrgetvalueouts r test_write_normalrssQ **C KS!12222  rct}tjt|tjdt |ddS)rk)azN/A)rm fill_valueszi| Col1 | Col2 | Col3 | Col4 | | 1.2 | "hello" | 1 | N/A | | 2.4 | 's worlds | 2 | 2 | Nrnrqs r test_write_fill_valuesrw sS **C KS!1|LLLL  rct}tjt|tjdt |ddS)5Write a table as a fixed width table with no padding.Nrm delimiter_padzQ|Col1| Col2|Col3|Col4| | 1.2| "hello"| 1| a| | 2.4|'s worlds| 2| 2| rnrqs r test_write_no_padr|sS **C KS!1FFFF  rct}tjt|tjdt |ddS)5Write a table as a fixed width table with no bookend.Frmbookendz]Col1 | Col2 | Col3 | Col4 1.2 | "hello" | 1 | a 2.4 | 's worlds | 2 | 2 Nrnrqs r test_write_no_bookendr&sS **C KS!15AAAA  rct}tjt|tjddt |ddS)7Write a table as a fixed width table with no delimiter.FNrmrrEzTCol1 Col2 Col3 Col4 1.2 "hello" 1 a 2.4 's worlds 2 2 rnrqs r test_write_no_delimiterr4sU **C KS!15DQQQQ  rct}tjt|tjt |ddS)rkrlz8| 1.2 | "hello" | 1 | a | | 2.4 | 's worlds | 2 | 2 | Nrrror!rVrrprqs r test_write_noheader_normalrBsQ **C KS!9::::  rct}tjt|tjdt |ddS)ryNrzz(|1.2| "hello"|1|a| |2.4|'s worlds|2|2| rrqs r test_write_noheader_no_padrOsS **C KS!9NNNN  rct}tjt|tjdt |ddS)r~Frz01.2 | "hello" | 1 | a 2.4 | 's worlds | 2 | 2 Nrrqs r test_write_noheader_no_bookendr\sS **C KS!95IIII  rct}tjt|tjddt |ddS)rFNrz*1.2 "hello" 1 a 2.4 's worlds 2 2 rrqs r test_write_noheader_no_delimiterris` **C K S15D  rct}tjt|tjdddt |ddS)rz%-8.3fz%-15srr)rmformatsz| Col1 | Col2 | Col3 | Col4 | | 1.200 | "hello" | 1 | a | | 2.400 | 's worlds | 2 | 2 | Nrnrqs r test_write_formatsrxsf **C K S)Hg3V3V  rc&d}tj|tj}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 rrrrrrrNrrFixedWidthTwoLinerr@r(rrAs r test_read_twoline_normalrs E *U5#: ; ; ;C"2333Aq 3'''QI&&&QK(((((rc,d}tj|tjddd}t|jjdt |dddt|ddd t|ddd d S) zRead restructured text tablezn ======= =========== Col1 Col2 ======= =========== 1.2 "hello" 2.4 's worlds ======= =========== rr<)rrJ position_linedata_endrrrrrNrrAs r test_read_twoline_ReSTrs E * &    C"2333Aq 3'''QI&&&QK(((((rc 0d}tj|tjddddd}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-r)rrErJrrKrrrrrNrrAs r test_read_twoline_humanrs E * &   C"2333Aq 3'''QI&&&QK(((((rcd}tjt5}tj|tjdddddn #1swxYwYdt |jvsJdS)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| |FrrEr5NzDPosition line should only contain delimiters and one other characterpytestraisesrrrrstrvaluerexcinfos r test_read_twoline_failrs  E - . .V' 5!8CuUUUUVVVVVVVVVVVVVVV O w}        #A  AAcd}tjt5}tj|tjdddddn #1swxYwYdt |jvsJdS)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| rFrNz(Characters in position line must be partrrs r test_read_twoline_wrong_markerrs  E - . .V' 5!8CuUUUUVVVVVVVVVVVVVVV 5W]9K9K K K K K K Krct}tjt|tjt |ddS)rkrlzdCol1 Col2 Col3 Col4 ---- --------- ---- ---- 1.2 "hello" 1 a 2.4 's worlds 2 2 Nrrror!rrrprqs r test_write_twoline_normalrsQ **C KS!89999  rct}tjt|tjddt |ddS)ryrD=)rmr{ position_charz|Col1 Col2 Col3 Col4 ==== ========= ==== ==== 1.2 "hello" 1 a 2.4 's worlds 2 2 Nrrqs r test_write_twoline_no_padrsa **C K S0SV  rct}tjt|tjddt |ddS)r~Trrzl|Col1| Col2|Col3|Col4| |----|---------|----|----| | 1.2| "hello"| 1| a| | 2.4|'s worlds| 2| 2| Nrrqs r test_write_twoline_no_bookendrsV **C KS!8$RUVVVV  rc,d}gd}tj|ddd|d}|j|ksJtj|ddd gksJtj|d d d gksJtj|d ddgksJdS)z3Test fix in #8511 where data_start is being ignoredzAAA y z 1 2 3 4 5 6 7 8 9 )rubcrr-rDfixed_width_no_header)rKrrEr(formatrurr<r]rN)rrrnpall)tblr(r!s r !test_fixedwidthnoheader_splittingrs C OOE * &    C <5 6#c(q!f$ % %%%% 6#c(q!f$ % %%%% 6#c(q!f$ % %%%%%%rcgd}gd}tj|dd|}t}tj||dd|||ksJdS)N)z&| int16 | float32 | > $ $ & &# - - - - - -rcgd}gd}tj|d|}t}tj||d|||ksJdS)N) zint32 float32 > $ $ & &# - - - - - -rcdg}tjtd5tj|ddgddddS#1swxYwYdS)Nrz)unexpected keyword argument 'header_rows')matchrrr)rr TypeErrorrr)rs r &test_fixed_width_no_header_header_rowsrYs ) *C y(T U U UNN 36VHMMMMNNNNNNNNNNNNNNNNNNsAA  A )4iornumpyrr astropy.iorastropy.io.ascii.corercommonrrrr"r)r.r1r3rBrFrPrSrWrYrcrirrrr!rsrwr|rrrrrrrrrrrrrrrrrrrrr rs 88888855555555777 ) ) ) ( ( (& ) ) ) ' ' ' + + + + + + + + + +++, + + ++++"))).3332  eju/000                               ) ) ) ))).)))4(LLL"   "   &&&....$...&NNNNNr