o iYD@sVddlZddlZddlmZmZddlmZeZeddZ eddZ edd Z ed d Z ed d Z ddZddZddZddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Z d4d5Z!d6d7Z"d8d9Z#d:d;Z$dd?Z&d@dAZ'dS)BN)rowtable) supports_ansicCsddgS)NHelloWorld12344342)zThis is a testr1234r r r L/home/ubuntu/.local/lib/python3.10/site-packages/wasabi/tests/test_tables.pydata r cCgdS)N)zCOL AzCOL BzCOL 3r r r r r headerr rcCr)N)rz 2030203.00r r r r r footerr rcCr)N)ryellow87r r r r r fg_colorsr rcCr)N)green23rr r r r r bg_colorsr rcCst|}|dks JdS)NzE Hello World 12344342 This is a test World 1234 rr resultr r r test_table_default$s rcCst||d}|dks JdS)N)rzg COL A COL B COL 3 Hello World 12344342 This is a test World 1234 r)r rrr r r test_table_header,  rcCs t|||dd}|dksJdS)NT)rrdivider COL A COL B COL 3 -------------- ----- ---------- Hello World 12344342 This is a test World 1234 -------------- ----- ---------- 2030203.00 r)r rrrr r r test_table_header_footer_divider4s r cCt|dd}|dks JdS)N)rclalignszE Hello World 12344342 This is a test World 1234 rrr r r test_table_aligns<rr'cCr!)Nr"r%zE Hello World 12344342 This is a test World 1234 rrr r r test_table_aligns_singleDrr(cCs(ddg}d}t||d}|dksJdS)Nabbccc)deefff widthsz1 a bb ccc d ee fff r)r r5rr r r test_table_widthsLs r6cCs d}t|dd}|dksJdS)Nr)r3r4z$a bb ccc )rrr r r test_row_single_widthsSs r7cCs<dgddfddddgfg}t||ddd}|dksJdS) Nhello)foobarbazworldzworld 1zworld 2T)rr multilinez COL A COL B COL 3 ----- ----- ------- hello foo world bar baz hello world world 1 world 2 r)rr rr r r test_table_multilineYs   r>cC0td|d}tr|dksJdS|dksJdS)Nr)rz3Hello World 12344342Hello World 12344342r SUPPORTS_ANSI)rrr r r test_row_fg_colorse rCcCr?)Nr)rz3Hello World 12344342r@rA)rrr r r test_row_bg_colorsorDrEcCs2td||d}tr|dksJdS|dksJdS)NrrrzHHello World 12344342r@rArrrr r r test_row_fg_colors_and_bg_colorsys rHcCs4d}dtj|<td||d}|dksJtj|=dS)NWASABI_LOG_FRIENDLYTruerrFr@osenvironrrrENV_LOG_FRIENDLYrr r r -test_row_fg_colors_and_bg_colors_log_friendlys   rPcCs6d}dtj|<td||dd}|dksJtj|=dS)NCUSTOM_LOG_FRIENDLYrJrCUSTOM)rr env_prefixr@rKrNr r r 4test_row_fg_colors_and_bg_colors_log_friendly_prefixs   rTcCs0dtjd<td||d}|dksJtjd=dS)NrJANSI_COLORS_DISABLEDrrFr@rKrGr r r 4test_row_fg_colors_and_bg_colors_supports_ansi_falses   rVcCs8t|||d||d}tr|dksJdS|dksJdS)NTrrrrra COL A  COL B COL 3  -------------- ----- ---------- Hello  World 12344342  This is a test World 1234  -------------- ----- ----------     2030203.00 rrrBr rrrrrr r r -test_colors_whole_table_with_automatic_widthss  rZcC6t|||d|d}tr|dksJdS|dksJdS)NT)rrrra{ COL A COL B COL 3  -------------- ----- ---------- Hello World 12344342  This is a test World 1234  -------------- ----- ----------   2030203.00 rrX)r rrrrr r r &test_colors_whole_table_only_fg_colors  r\cCr[)NT)rrrra{ COL A  COL B COL 3 -------------- ----- ---------- Hello  World 12344342 This is a test World 1234 -------------- ----- ----------     2030203.00 rrX)r rrrrr r r &test_colors_whole_table_only_bg_colorsr]r^c C:t|||d||dd}tr|dksJdS|dksJdS)NTr1)rrrrrspacinga COL A  COL B COL 3  -------------- ----- ---------- Hello  World 12344342  This is a test World 1234  -------------- ----- ----------     2030203.00 z COL A COL B COL 3 -------------- ----- ---------- Hello World 12344342 This is a test World 1234 -------------- ----- ---------- 2030203.00 rXrYr r r -test_colors_whole_table_with_supplied_spacing    rac Cr_)NTr0)rrrrrr5a COL A COL B COL 3  ----- -- ---------- Hello World 12344342  This is a test World 1234  ----- -- ----------     2030203.00 z COL A COL B COL 3 ----- -- ---------- Hello World 12344342 This is a test World 1234 ----- -- ---------- 2030203.00 rXrYr r r ,test_colors_whole_table_with_supplied_widthsrbrcc Cr_)NTr"rrrrrr&a  COL A COL B  COL 3 -------------- ----- ----------  Hello World  12344342 This is a test World  1234 -------------- ----- ----------     2030203.00 z COL A COL B COL 3 -------------- ----- ---------- Hello World 12344342 This is a test World 1234 -------------- ----- ---------- 2030203.00 rXrYr r r -test_colors_whole_table_with_single_alignmentrbrec Cr_)NT)r#r"r$rda  COL A  COL B COL 3  -------------- ----- ----------  Hello  World 12344342  This is a test World 1234  -------------- ----- ----------     2030203.00 z COL A COL B COL 3 -------------- ----- ---------- Hello World 12344342 This is a test World 1234 -------------- ----- ---------- 2030203.00 rXrYr r r /test_colors_whole_table_with_multiple_alignment-rbrfcCsBtgdddfdf||dd}tr|dksJdS|dksJdS) N)CharlesQuintonMurphymybrother)123T)r rrr=ad Charles my brother Quinton     Murphy            1  2  3  zt Charles my brother Quinton Murphy 1 2 3 rXrYr r r &test_colors_whole_table_with_multilineEs  rocCs:d}dtj|<t|||d||d}|dksJtj|=dS)NrIrJTrWrrLrMrr rrrrrOrr r r $test_colors_whole_table_log_friendlyXs   rrc Cs<d}dtj|<t|||d||dd}|dksJtj|=dS)NrQrJTrR)rrrrrrSrrprqr r r +test_colors_whole_table_log_friendly_prefixjs   rscCs6dtjd<t|||d||d}|dksJtjd=dS)NrJrUTrWrrprYr r r +test_colors_whole_table_supports_ansi_falses   rtc Cs>t|||d||ddid}tr|dksJdS|dksJdS)NTr )rrrrr color_valuesa COL A  COL B COL 3  -------------- ----- ---------- Hello  World 12344342  This is a test World 1234  -------------- ----- ----------     2030203.00 rrXrYr r r $test_colors_whole_table_color_valuess    rw)(rLpytest wasabi.tablesrr wasabi.utilrrBfixturer rrrrrrr r'r(r6r7r>rCrErHrPrTrVrZr\r^rarcrerfrorrrsrtrwr r r r sR