U }Äúg7Öã @s˜dZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z ddlmZddlmZGd d „d eƒZGd d „d eƒZGd d„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd „d eƒZGd!d"„d"eƒZGd#d$„d$eƒZGd%d&„d&eƒZ d'd(d)d*d+d,d-d.d/d0d1d2g Z!Gd3d4„d4eƒZ"Gd5d6„d6eƒZ#dS)7a8Tests for Beautiful Soup's tree traversal methods. The tree traversal methods are the main advantage of using Beautiful Soup over just using a parser. Different parsers will build different Beautiful Soup trees given the same markup, but all Beautiful Soup trees can be traversed with the methods tested here. éN)Ú BeautifulSoup)Úbuilder_registry)Ú!AttributeResemblesVariableWarningÚCDataÚCommentÚNavigableStringÚTag)Ú SoupStraineré)ÚSoupTestc@sHeZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dS)ÚTestFindz&Basic tests of the find() method. cCs"| d¡}| d¡jdkst‚dS)Nz 1234ÚbÚ2)ÚsoupÚfindÚstringÚAssertionError©Úselfr©rú7/tmp/pip-unpacked-wheel-kgiupv3k/bs4/tests/test_tree.pyÚ test_find_tag"s zTestFind.test_find_tagcCs"| d¡}|jdddkst‚dS)Nu

Räksmörgås

õ RäksmörgÃ¥s©r)rrrrrrrÚtest_unicode_text_find&s zTestFind.test_unicode_text_findcCs,| d¡}t|ƒd|jddjks(t‚dS)Nu&

here it is

z here it isr©Úid)rÚstrrÚtextrrrrrÚtest_unicode_attribute_find*s z$TestFind.test_unicode_attribute_findcCs"| d¡}dt| ¡ƒkst‚dS)z)Test an optimization that finds all tags.úfoobaréN©rÚlenÚfind_allrrrrrÚtest_find_everything/s zTestFind.test_find_everythingcCs$| d¡}dt| d¡ƒks t‚dS)z;Test an optimization that finds all tags with a given name.zfoobarbazr!ÚaNr"rrrrÚtest_find_everything_with_name4s z'TestFind.test_find_everything_with_namecCsP| d¡}d| ¡jkst‚d| d¡ ¡jks4t‚d| d¡ ¡jksLt‚dS)Nz

ÚdivÚp©rrÚnamerÚfind_previous_siblingÚfind_next_siblingrrrrÚtest_find_with_no_arguments9s z$TestFind.test_find_with_no_argumentscCsP| d¡}d| ¡jkst‚d| d¡ ¡jks4t‚d| d¡ ¡jksLt‚dS)Nz"text
text
text

text

r(r)r*rrrrÚ+test_find_with_no_arguments_only_finds_tags?s z4TestFind.test_find_with_no_arguments_only_finds_tagsN) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrr%r'r.r/rrrrr sr c@s@eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dS)Ú TestFindAllz%Basic tests of the find_all() method.cCsX| d¡}dt|j ¡ƒks t‚dt| d¡ ¡ƒks:t‚dt| d¡ ¡ƒksTt‚dS)Nz/text
text
text

text

r!r r)r()rr#Úbodyr$rrÚfind_previous_siblingsÚfind_next_siblingsrrrrÚ/test_find_all_with_no_arguments_only_finds_tagsIs z;TestFindAll.test_find_all_with_no_arguments_only_finds_tagscCsz| d¡}|jdddgks t‚|jddgdddgksFoobar»ÚbarrÚFooz.*õ»TN)rr$rÚreÚcompilerrrrÚtest_find_all_text_nodesOs   z$TestFindAll.test_find_all_text_nodescCs‚| d¡}| |jddddddg¡| |jddddg¡| |jdd ddddd d g¡| |jdd ddddd d g¡d S)z7You can limit the number of items returned by find_all.z(12345r&é©ÚlimitÚ1rÚ3r é Ú4Ú5rN©rÚassert_selectsr$rrrrÚtest_find_all_limit\s   zTestFindAll.test_find_all_limitcCs:| d¡}| |ddddg¡| |jdddg¡dS) Nz!123r&r r@rBÚfoorrC)rrHr rrrrÚ%test_calling_a_tag_is_calling_findallfs z1TestFindAll.test_calling_a_tag_is_calling_findallc Csp| d¡}g}| |¡tjddB}g| |¡ks8t‚|\}|jtksLt‚t|j ƒ}|dksbt‚W5QRXdS)NúT©ÚrecordzLIgnoring nested list [[...]] to avoid the possibility of infinite recursion.) rÚappendÚwarningsÚcatch_warningsr$rÚfilenameÚ__file__rÚmessage)rrÚselfrefÚwÚwarningÚmsgrrrÚTtest_find_all_with_self_referential_data_structure_does_not_cause_infinite_recursionks   ÿÿz`TestFindAll.test_find_all_with_self_referential_data_structure_does_not_cause_infinite_recursioncCsX| d¡}| d¡}t|dƒs"t‚| d¡}t|dƒs:t‚|jdd}t|dƒsTt‚dS)z%All find_all calls return a ResultSetrLr&ÚsourceTrJrN)rr$Úhasattrr©rrÚresultrrrÚtest_find_all_resultsets    z#TestFindAll.test_find_all_resultsetN) r0r1r2r3r8r>rIrKrYr^rrrrr4Fs  r4c@seZdZdd„ZdS)ÚTestFindAllBasicNamespacescCs<| d¡}d| d¡jkst‚d|jddidjks8t‚dS)Nz04rEz mathml:msqrtr&zsvg:fillZred©Úattrs)rrrrr+rrrrÚtest_find_by_namespaced_names z7TestFindAllBasicNamespaces.test_find_by_namespaced_nameN)r0r1r2rbrrrrr_Œsr_c@sneZdZdZddœdd„Zdd„Zdd „Zd d „Zd d „Zdd„Z dd„Z dd„Z dd„Z dd„Z dd„ZdS)ÚTestFindAllByNamez&Test ways of finding tags by tag name.N©ÚreturncCs| d¡|_dS)NzFirst tag. Second tag. Third Nested tag. tag.©rÚtree©rrrrÚ setup_method–szTestFindAllByName.setup_methodcCs| |j d¡ddg¡dS©Nr&ú First tag.ú Nested tag.©rHrgr$rhrrrÚtest_find_all_by_tag_name›sz+TestFindAllByName.test_find_all_by_tag_namecCs\| |jjddddg¡| |jjdddddg¡| |jjdt d¡dddg¡dS)Nr&rkrTrlÚtag©rHrgr$r<r=rhrrrÚtest_find_all_by_name_and_textŸsÿÿþz0TestFindAllByName.test_find_all_by_name_and_textcCs| |jj d¡dg¡dS)Nr&rl)rHrgÚcr$rhrrrÚ!test_find_all_on_non_root_element­sz3TestFindAllByName.test_find_all_on_non_root_elementcCs| | d¡ddg¡dSrj)rHrgrhrrrÚ%test_calling_element_invokes_find_all±sz7TestFindAllByName.test_calling_element_invokes_find_allcCs | |j tdƒ¡ddg¡dSrj)rHrgr$r rhrrrÚtest_find_all_by_tag_strainer´sÿz/TestFindAllByName.test_find_all_by_tag_strainercCs"| |j ddg¡dddg¡dS)Nr&r rkú Second tag.rlrmrhrrrÚtest_find_all_by_tag_names¹sÿz,TestFindAllByName.test_find_all_by_tag_namescCs$| |j dddœ¡dddg¡dS)NT)r&r rkrvrlrmrhrrrÚtest_find_all_by_tag_dict¾sþz+TestFindAllByName.test_find_all_by_tag_dictcCs$| |j t d¡¡dddg¡dS)Nz^[ab]$rkrvrlrprhrrrÚtest_find_all_by_tag_reÄsþz)TestFindAllByName.test_find_all_by_tag_recCs,dd„}| d¡}| | |¡ddg¡dS)NcSs|j| d¡kS©Nr)r+Úget©rorrrÚid_matches_nameÍszRTestFindAllByName.test_find_all_with_tags_matching_method..id_matches_namezƒMatch 1. Does not match. Match 2.zMatch 1.zMatch 2.rG)rr}rgrrrÚ'test_find_all_with_tags_matching_methodÊs z9TestFindAllByName.test_find_all_with_tags_matching_methodcCsx| d¡}| dd¡}| dt d¡¡}| dddg¡\}}d|jksJt‚d|jksXt‚d|jksft‚d|jkstt‚dS)NzH
1
2
3
r(za dza brCrB)rrr<r=r$rr)rrZr1Zr2Zr3Zr4rrrÚ%test_find_with_multi_valued_attributeÖsÿ z7TestFindAllByName.test_find_with_multi_valued_attribute)r0r1r2r3rirnrqrsrtrurwrxryr~rrrrrrc“s rcc@sŒeZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd „Zd!S)"ÚTestFindAllByAttributecCs&| d¡}| |jddddg¡dS)NzÅ Matching a. Non-matching Matching b.a. Úfirstrz Matching a.z Matching b.rG©rrgrrrÚtest_find_all_by_attribute_nameäs z6TestFindAllByAttribute.test_find_all_by_attribute_namecCstd d¡}d d¡}| |¡}|jg|j|dks6t‚|jg|j| d¡dksTt‚|jgspt|j|dgdƒ‚dS)Nu×ולשÚutf8u)Útitlezsomething else)Úencoderr&r$rÚdecode)rZpeaceÚdatarrrrÚ%test_find_all_by_utf8_attribute_valueîs    zName match. Class match. Non-match. A tag called 'name1'. Zname1©r+zA tag called 'name1'.r+r`z Name match.ÚclassZclass2z Class match.rGr‚rrrÚtest_find_all_by_attribute_dictös z6TestFindAllByAttribute.test_find_all_by_attribute_dictcCs°| d¡}| |jddddg¡| |jddddg¡| |jdd ddg¡| | dd¡dg¡| |jdd ddg¡| | dd¡dg¡| | dd ¡dg¡dS) Nzî Class 1. Class 2. Class 1. Class 3 and 4. r&rB©Úclass_zClass 1.rrrCzClass 3 and 4.rEr`rGr‚rrrÚtest_find_all_by_classs z-TestFindAllByAttribute.test_find_all_by_classcCst| d¡}|jdt d¡d}| |dg¡|jdt d¡d}| |dg¡|jdt d¡d}| |dg¡dS)Nz#Found itZgarÚorúFound itr&zo b)rr$r<r=rH)rrgÚfrrrÚ0test_find_by_class_when_multiple_classes_presents zGTestFindAllByAttribute.test_find_by_class_when_multiple_classes_presentcCsd| d¡}| | dt d¡¡dg¡dd„}| | d|¡g¡dd„}| | d|¡dg¡dS) NzFound itr&Úbar‘cSs t|ƒdkS©Nr?©r#©ÚvaluerrrÚbig_attribute_value/sznTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_class..big_attribute_valuecSs t|ƒdkSr•r–r—rrrÚsmall_attribute_value4szpTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_class..small_attribute_value©rrHr$r<r=)rrr™ršrrrÚ:test_find_all_with_non_dictionary_for_attrs_finds_by_class*s  zQTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_classcCsŒ| d¡}| d¡\}}||gs0t| dd¡ƒ‚|gsFt| dd¡ƒ‚|g|jdddks^t‚|g| dd¡kstt‚g| dd¡ksˆt‚dS)Nz*r&rJr9zfoo barrzbar foo©rr$r)rrr&Za2rrrÚ:test_find_all_with_string_for_attrs_finds_multiple_classes9s zQTestFindAllByAttribute.test_find_all_with_string_for_attrs_finds_multiple_classescCs0| d¡}tddid}| | |¡dg¡dS)Nzi Match. Non-match.rrr`zMatch.)rr rHr$)rrgZstrainerrrrÚ'test_find_all_by_attribute_soupstrainerEs z>TestFindAllByAttribute.test_find_all_by_attribute_soupstrainercCs&| d¡}| |jddddg¡dS)NúID present. No ID present. ID is empty.r&rzNo ID present.rGr‚rrrÚ$test_find_all_with_missing_attributeMs z;TestFindAllByAttribute.test_find_all_with_missing_attributecCs&| d¡}| |jddddg¡dS)Nr Trz ID present.z ID is empty.rGr‚rrrÚ$test_find_all_with_defined_attributeUs z;TestFindAllByAttribute.test_find_all_with_defined_attributecCs>| d¡}ddg}| |jdd|¡| |jdd|¡dS)Nz[Unquoted attribute. Quoted attribute.zUnquoted attribute.zQuoted attribute.r rrBrG)rrgÚexpectedrrrÚ$test_find_all_with_numeric_attribute]s z;TestFindAllByAttribute.test_find_all_with_numeric_attributecCs@| d¡}| |jdddgdddg¡| |jgdg¡dS)Nz‘1 2 3 No ID.rBrCrErrGr‚rrrÚ(test_find_all_with_list_attribute_valuesfs z?TestFindAllByAttribute.test_find_all_with_list_attribute_valuescCs,| d¡}| |jt d¡dddg¡dS)NzÞOne a. Two as. Mixed as and bs. One b. No ID.z^a+$rzOne a.zTwo as.r›r‚rrrÚ5test_find_all_with_regular_expression_attribute_valuers zLTestFindAllByAttribute.test_find_all_with_regular_expression_attribute_valuecCsB| d¡}|j}|g|jdddks(t‚g|jdddks>t‚dS)Nzfoobarfoor&rJrr9©rr&r$r©rrr&rrrÚ'test_find_by_name_and_containing_string~s z>TestFindAllByAttribute.test_find_by_name_and_containing_stringcCs*| d¡}| d¡|jdddks&t‚dS)Nz"foofoor&rJrrrrrrÚ=test_find_by_name_and_containing_string_when_string_is_buried…s zTTestFindAllByAttribute.test_find_by_name_and_containing_string_when_string_is_buriedcCsB| d¡}|j}|g|jdddks(t‚g|jdddks>t‚dS)Nz"foofoor!rJ)rrr r9r§r¨rrrÚ,test_find_by_attribute_and_containing_string‰s zCTestFindAllByAttribute.test_find_by_attribute_and_containing_stringN)r0r1r2rƒr‰rŒrr“rœržrŸr¡r¢r¤r¥r¦r©rªr«rrrrr€ãs      r€c@seZdZdZdd„ZdS)Ú TestSmoothzTest Tag.smooth.cCs| d¡}|j}| d¡| d¡| tdƒ¡| tdƒ¡| d¡| ¡}t||dƒ}| d¡| d ¡| |¡d|jjksŒt‚d t |j ƒksžt‚|  ¡d t |j ƒks¸t‚d |j d ksÊt‚d|jjksÚt‚d|j dksìt‚d|j dksþt‚dS)Nz
a
r rrz Comment 1z Comment 2ÚdÚspanrBrééÚabcrZ12r r!) rr(rOrÚdefault_builderrr®rrr#ÚcontentsZsmooth)rrr(Úbuilderr®rrrÚ test_smooth”s(        zTestSmooth.test_smoothN)r0r1r2r3rµrrrrr¬‘sr¬c@seZdZdZdd„ZdS)Ú TestIndexzTest Tag.indexc CsZ| d¡}|j}t|jƒD]\}}|| |¡kst‚qt t¡| d¡W5QRXdS)Nah
Identical Not identical Identical Identical with child Also not identical Identical with child
r ) rr(Ú enumerater³ÚindexrÚpytestÚraisesÚ ValueError)rrgr(ÚiÚelementrrrÚ test_indexÀs   zTestIndex.test_indexN)r0r1r2r3r¾rrrrr¶½sr¶c@sfeZdZdZddœdd„Zdd„Zdd „Zd d „Zd d „Zdd„Z dd„Z dd„Z dd„Z dd„Z dS)ÚTestParentOperationsz;Test navigation and searching through an element's parents.NrdcCs| d¡|_|jj|_dS)Na<