o iIP@sdZddlZddlZddlmZmZmZmZGdddZGdddZ Gd d d Z Gd d d Z Gd ddZ GdddZ GdddZdZGdddZdS)z comment testing is all about roundtrips these can be done in the "old" way by creating a file.data and file.roundtrip but there is little flexibility in doing that but some things are not easily tested, eog. how a roundtrip changes N) round_tripdedentround_trip_loadround_trip_dumpc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3S)4 TestCommentscCsLd}t|ddttt|ddWddS1swYdS)z~not excluding comments caused some problems if at the end of the file without a newline. First error, then included z - europe: 10 # abc )extraza N)rpytestraisesAssertionError)selfxrY/home/ubuntu/.local/lib/python3.10/site-packages/srsly/tests/ruamel_yaml/test_comments.pytest_no_end_of_file_eols   "z$TestComments.test_no_end_of_file_eolcC tddS)Nz[ - europe: 10 - usa: - ohio: 2 - california: 9 rr rrrtest_no_commentszTestComments.test_no_commentscCr)Nzs a: 1 b: 2 c: 3 b1: 2 b2: 2 d: 4 e: 5 f: 6 rrrrrtest_round_trip_ordering'rz%TestComments.test_round_trip_orderingcCr)Nze - europe: 10 # top - usa: - ohio: 2 - california: 9 # o rrrrr test_complex5rzTestComments.test_complexcCsd}t|ddS)Nz5 # comment scalar ... z scalar ... rr srrr test_dropped?szTestComments.test_droppedcCr)Nz # C start a # C start b abc: 1 ghi: 2 klm: 3 # C end a # C end b rrrrrtest_main_mapping_begin_endGrz(TestComments.test_main_mapping_begin_endcCs,d}t|}t|dd}|tdksJdS)NzM a: b: # comment 1 c: 1 # comment 2 indentzQ a: b: # comment 1 c: 1 # comment 2 )rrr)r rdyrrr test_reindentTs  zTestComments.test_reindentcCr)Nz # C start a # C start b abc: 1 # abc comment ghi: 2 klm: 3 # klm comment # C end a # C end b rrrrr&test_main_mapping_begin_end_items_postdrz3TestComments.test_main_mapping_begin_end_items_postcCr)Nz # C start a # C start b - abc - ghi - klm # C end a # C end b rrrrrtest_main_sequence_begin_endqrz)TestComments.test_main_sequence_begin_endcCr)Nz # C start a # C start b - abc # abc comment - ghi - klm # klm comment # C end a # C end b rrrrr'test_main_sequence_begin_end_items_post~rz4TestComments.test_main_sequence_begin_end_items_postcCr)Nz # C start a # C start b abc: 1 ghi: 2 klm: 3a: alpha 3b: beta # it is all greek to me # C end a # C end b rrrrr#test_main_mapping_begin_end_complexrz0TestComments.test_main_mapping_begin_end_complexcCd}t|ddddS)Nz hr: # 1998 hr ranking - Mark McGwire - Sammy Sosa rbi: # 1998 rbi ranking - Sammy Sosa - Ken Griffey rrblock_seq_indentrrrrrtest_09s zTestComments.test_09cCr)Nz hr: # 1998 hr ranking - Mark McGwire - Sammy Sosa rbi: # 1998 rbi ranking - Sammy Sosa - Ken Griffey rrrrrtest_09arzTestComments.test_09acCr)NzE abc: 1 # C 3a # C 3b ghi: 2 rrrrrtest_simple_map_middle_commentrz+TestComments.test_simple_map_middle_commentcCr)Nz` map1: # comment 1 # comment 2 map2: key1: val1 rrrrrtest_map_in_map_0rzTestComments.test_map_in_map_0cCr)NzT map1: # comment 1 map2: key1: val1 rrrrrtest_map_in_map_1zTestComments.test_map_in_map_1cCr)Nz args: username: anthon passwd: secret fullname: Anthon van der Neut tmux: session-name: test loop: wait: 10 rrrrrtest_application_argumentsr0z'TestComments.test_application_argumentscCs<d}t|}d|dd<|dd}t|t|ksJdS)Nz args: username: anthon # name passwd: secret # password fullname: Anthon van der Neut tmux: session-name: test loop: wait: 10 zdeleted passwordargspasswdz: secret z: deleted password)rreplacerr)r rdatarrrtest_substitutes   zTestComments.test_substitutecCr)Nz !!set # the beginning ? a # next one is B (lowercase) ? b # You see? Promised you. ? c # this is the end rrrrrtest_set_commentrzTestComments.test_set_commentcCr)Nzd !!omap - a: 1 - b: 2 # two - c: 3 # three - d: 4 rrrrrtest_omap_comment_roundtriprz(TestComments.test_omap_comment_roundtripcCr)Nzw !!omap - a: 1 - b: 2 # two - c: 3 # three # last one - d: 4 rrrrr'test_omap_comment_roundtrip_pre_comment rz4TestComments.test_omap_comment_roundtrip_pre_commentcCr)Nu verbosity: 1 # 0 is minimal output, -1 none base_url: http://gopher.net special_indices: [1, 5, 8] also_special: - a - 19 - 32 asia and europe: &asia_europe Turkey: Ankara Russia: Moscow countries: Asia: <<: *asia_europe Japan: Tokyo # 東京 Europe: <<: *asia_europe Spain: Madrid Italy: Rome rrrrrtest_non_asciirzTestComments.test_non_asciicCNddl}td}t|}|jj}dD]}|jj|d||d}||ks$JqdSNrzG ab: - x # comment - y # more comment )TFF)default_flow_styleDumper allow_unicodesrsly.ruamel_yamlrr ruamel_yamlRoundTripDumperdumpr srslyrr5dumperutfr!rrrtest_dump_utf8.zTestComments.test_dump_utf8cCr;r<r@rErrrtest_dump_unicode_utf8@rJz#TestComments.test_dump_unicode_utf8cCshd}t|}|jgddddksJtt|gddkWddS1s-wYdS)Nzq a: - b: c: 42 - d: f: 196 e: g: 3.14 )arr fT)list_ok)rmlgetr r r )r rr rrr test_mlget_00Rs   "zTestComments.test_mlget_00N)__name__ __module__ __qualname__rrrrrrr"r#r$r%r&r+r,r-r.r/r1r6r7r8r9r:rIrKrQrrrrrs4                rc@sTeZdZdZeddZddZddZdd Zd d Z d d Z ddZ ddZ dS)TestInsertPopListzslist insertion is more complex than dict insertion, as you need to move the values to subsequent keys on insertcCdS)Nz ab: - a # a - b # b - c - d # d de: - 1 - 2 rrrrrinsfzTestInsertPopList.inscC:t|j}|dddt|dd}|tdksJdS)Nabrxyzr(rz ab: - xyz - a # a - b # b - c - d # d de: - 1 - 2 rrWinsertrrr r r!rrr test_insert_0t  zTestInsertPopList.test_insert_0cCrY)NrZrr[r(rz ab: - a # a - b # b - c - d # d - xyz de: - 1 - 2 r\r^rrr test_insert_1r`zTestInsertPopList.test_insert_1cCrY)NrZrr[r(rz ab: - a # a - xyz - b # b - c - d # d de: - 1 - 2 r\r^rrr test_insert_2r`zTestInsertPopList.test_insert_2cC@t|j}|ddt|dd}t||tdksJdS)NrZrr(rzo ab: - b # b - c - d # d de: - 1 - 2 rrWpoprprintrr^rrr test_pop_0  zTestInsertPopList.test_pop_0cCrc)NrZrr(rzo ab: - a # a - c - d # d de: - 1 - 2 rdr^rrr test_pop_1rhzTestInsertPopList.test_pop_1cCs@t|j}|ddt|dd}t||tdksJdS)NrZr(rzx ab: - a # a - b # b - d # d de: - 1 - 2 rdr^rrr test_pop_2rhzTestInsertPopList.test_pop_2cCrc)NrZr(rzn ab: - a # a - b # b - c de: - 1 - 2 rdr^rrr test_pop_3rhzTestInsertPopList.test_pop_3N) rRrSrT__doc__propertyrWr_rarbrgrirjrlrrrrrUbs   rUc@s0eZdZeddZddZddZddZd S) TestInsertInMappingcCrV)Nz first_name: Art occupation: Architect # This is an occupation comment about: Art Vandelay is a fictional character that George invents... rrrrrrWrXzTestInsertInMapping.inscC@t|j}|jdddddt|}t||tdksJdS)Nr last nameVandelaynew keycommentz first_name: Art last name: Vandelay # new key occupation: Architect # This is an occupation comment about: Art Vandelay is a fictional character that George invents... rrWr]rrfrr^rrrtest_insert_at_pos_1 z(TestInsertInMapping.test_insert_at_pos_1cCrp)Nrrqrrrsrtz last name: Vandelay # new key first_name: Art occupation: Architect # This is an occupation comment about: Art Vandelay is a fictional character that George invents... rvr^rrrtest_insert_at_pos_0rxz(TestInsertInMapping.test_insert_at_pos_0cCrp)Nrkrqrrrsrtz first_name: Art occupation: Architect # This is an occupation comment about: Art Vandelay is a fictional character that George invents... last name: Vandelay # new key rvr^rrrtest_insert_at_pos_3s z(TestInsertInMapping.test_insert_at_pos_3N)rRrSrTrnrWrwryrzrrrrros   roc@s$eZdZddZddZddZdS)TestCommentedMapMergecCsTtd}|dddksJd|dvsJ|dddks Jd|dvs(JdS)Nz x: &base a: 1 b: 2 c: 3 y: <<: *base k: 4 l: 5 rrLrr!)rr r5rrrtest_in_operator-s z&TestCommentedMapMerge.test_in_operatorcCjtd}|dddksJ|dddksJtjdkr)t|ddks'JdSt|ddks3JdS) NzH x: &base a: 1 y: <<: *base rrLrr!rk zordereddict({'a': 1})zordereddict([('a', 1)])rsys version_infostrr|rrr test_issue_60?s z#TestCommentedMapMerge.test_issue_60cCr~) NzW x: &base a: 1 y: <<: *base b: 2 rrLrr!rzordereddict({'b': 2, 'a': 1})z!ordereddict([('b', 2), ('a', 1)])rr|rrrtest_issue_60_1Os  z%TestCommentedMapMerge.test_issue_60_1N)rRrSrTr}rrrrrrr{,s r{c@seZdZddZddZejjddddZd d Z d d Z d dZ ddZ ddZ ddZddZddZddZddZddZdd Zd!S)"TestEmptyLinescCs0td}t|dd}t|dd}||ksJdS)Na --- # Please add key/value pairs in alphabetical order aws_s3_bucket: 'mys3bucket' jenkins_ad_credentials: bind_name: 'CN=svc-AAA-BBB-T,OU=Example,DC=COM,DC=EXAMPLE,DC=Local' bind_pass: 'xxxxyyyy{' Tpreserve_quotesexplicit_startrrrr yaml_strr r!rrr test_issue_46cs  zTestEmptyLines.test_issue_46cCr)NzH a: 1x b: 2x c: 3x d: 4x rrrrrtest_multispace_maptrz"TestEmptyLines.test_multispace_mapT)strictcCr)NzI a: 1x b: 2x c: 3x d: 4x rrrrrtest_multispace_map_initialr0z*TestEmptyLines.test_multispace_map_initialcCr)Nz: - a: 1y b: 2y c: 3y rrrrrtest_embedded_maprz TestEmptyLines.test_embedded_mapcCr)Nz. - 1 - 2 - 3 rrrrrtest_toplevel_seqrz TestEmptyLines.test_toplevel_seqcCr)NzN a: b: - 1 - 2 - 3 rrrrrtest_embedded_seqrz TestEmptyLines.test_embedded_seqcCZd}t|dd}t|dd}d}|D]}||d7}t|dq||ks+JdS)Nz--- a: 'x' b: y Trrr$rr splitlinesrstriprfr rr r!strippedlinerrrtest_line_with_only_spaces   z)TestEmptyLines.test_line_with_only_spacescCr)Nz--- a: "x" b: y Trrrrrrrrrrtest_some_eol_spacesrz#TestEmptyLines.test_some_eol_spacescCsFtd}t|}t|jt|dd}t|dd||ks!JdS)NzP toplevel: # some comment sublevel: 300 rrrz$ )rrrfcarr4rrrrtest_issue_54_not_oks  z#TestEmptyLines.test_issue_54_not_okcCs,td}t|}t|dd}||ksJdS)NzO toplevel: # some comment sublevel: 300 rrrrrrrtest_issue_54_oks  zTestEmptyLines.test_issue_54_okcCr)Nzz a: b: - c1: cat # a1 # my comment on catfish - c2: catfish # a2 rrrrr test_issue_93rzTestEmptyLines.test_issue_93cCr)Nzn a: - - c1: cat # a1 # my comment on catfish - c2: catfish # a2 rrrrrtest_issue_93_00rzTestEmptyLines.test_issue_93_00cCr)Nzc - - c1: cat # a1 # my comment on catfish - c2: catfish # a2 rrrrrtest_issue_93_01rzTestEmptyLines.test_issue_93_01cCr)NzP - c1: cat # my comment on catfish - c2: catfish rrrrrtest_issue_93_02 r0zTestEmptyLines.test_issue_93_02cCr)Nzg a: b: c: c_val d: e: g: g_val rrrrr test_issue_96r0zTestEmptyLines.test_issue_96N)rRrSrTrrr markxfailrrrrrrrrrrrrrrrrrras"          rc@s*eZdZejjejdkddddZdS)TestUnicodeComments)r(z wide unicode)reasoncCr)Nu name: TEST description: test using author: Harguroicha sql: command: |- select name from testtbl where no = :no ci-test: - :no: 04043709 # 小花 - :no: 05161690 # 茶 - :no: 05293147 # 〇𤋥川 - :no: 05338777 # 〇〇啓 - :no: 05273867 # 〇 - :no: 05205786 # 〇𤦌 rrrrr test_issue_55$r0z!TestUnicodeComments.test_issue_55N) rRrSrTr rskipifrrrrrrrr#src@sdeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ dS)TestEmptyValueBeforeCommentscCr)NzK - a: b c: d d: # foo - e: f rrrrrtest_issue_25a;rz+TestEmptyValueBeforeComments.test_issue_25acCr)NzK - a: b c: d d: # foo e: f rrrrrtest_issue_25a1Erz,TestEmptyValueBeforeComments.test_issue_25a1cCr)Nz? var1: #empty var2: something #notempty rrrrrtest_issue_25bOrz+TestEmptyValueBeforeComments.test_issue_25bcCr)Nzi params: a: 1 # comment a b: # comment b c: 3 # comment c rrrrrtest_issue_25cWrz+TestEmptyValueBeforeComments.test_issue_25ccCr)Nz{ params: a: 1 # comment a b: # comment b # extra c: 3 # comment c rrrrrtest_issue_25c1arz,TestEmptyValueBeforeComments.test_issue_25c1cCr)NzN params: a: 1 # comment a b: # comment b rrrrrtest_issue_25_00lrz-TestEmptyValueBeforeComments.test_issue_25_00cCr)Nz a: # comment 1 # comment 2 - b: # comment 3 c: 1 # comment 4 rrrrrtest_issue_25_01urz-TestEmptyValueBeforeComments.test_issue_25_01cCr)Nze a: # comment 1 # comment 2 - b: 2 # comment 3 rrrrrtest_issue_25_02rz-TestEmptyValueBeforeComments.test_issue_25_02cCr')Nze a: # comment 1 # comment 2 - b: 2 # comment 3 rr(r)rrrrrtest_issue_25_03sz-TestEmptyValueBeforeComments.test_issue_25_03cCr)Nze a: # comment 1 # comment 2 b: 1 # comment 3 rrrrrtest_issue_25_04rz-TestEmptyValueBeforeComments.test_issue_25_04cCr)Nz # comment 1 - a - b # comment 2 - c - d # comment 3 - [e] - f # comment 4 - [] rrrrrtest_flow_seq_within_seqrz5TestEmptyValueBeforeComments.test_flow_seq_within_seqN)rRrSrTrrrrrrrrrrrrrrrr:s        rz%y: p # Some comment a: | x {}b: y c@seZdZddZdS)TestBlockScalarWithCommentscCs@ddl}dD]}t|}|j|}|j||ksJqdS)Nr)rrz # Another comment z # abc #xyz rz # abc #xyz z # abc #xyz )rA)test_block_scalar_commented_line_templateformatrBrr)r rFrcommented_liner5rrrtest_scalar_with_commentss  z5TestBlockScalarWithComments.test_scalar_with_commentsN)rRrSrTrrrrrrs r)rmr r roundtriprrrrrrUror{rrrrrrrrrs" Q55Cq