U lu f@sdZddlmZddlZddlZddlZddlmZddlm Z m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZddlmZGd d d eZGd d d eZGd ddeZGdddeZGdddeZGdddeZGdddeZGdddeZ GdddeZ!Gddde!Z"Gddde!Z#Gdd d eZ$Gd!d"d"e$Z%Gd#d$d$e$Z&Gd%d&d&eZ'Gd'd(d(eZ(dS))a8Tests 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. ) set_traceN) BeautifulSoup)builder_registryHTMLParserTreeBuilder) CDataComment DeclarationDoctype FormatterNavigableStringScript SoupStrainer StylesheetTagTemplateString)SoupTestc@s8eZdZdZddZddZddZdd Zd d Zd S) TestFindzBasic tests of the find() method. find() just calls find_all() with limit=1, so it's not tested all that thouroughly here. cCs"|d}|djdkstdS)Nz 1234b2)soupfindstringAssertionErrorselfrr7/tmp/pip-unpacked-wheel-pg4vfqbd/bs4/tests/test_tree.py test_find_tag-s zTestFind.test_find_tagcCs"|d}|jdddkstdS)Nu

Räksmörgås

Räksmörgåsr)rrrrrrrtest_unicode_text_find1s zTestFind.test_unicode_text_findcCs,|d}t|d|jddjks(tdS)Nu&

here it is

z here it isrid)rstrrtextrrrrrtest_unicode_attribute_find5s z$TestFind.test_unicode_attribute_findcCs"|d}dt|kstdS)z)Test an optimization that finds all tags.foobarNrlenfind_allrrrrrtest_find_everything;s zTestFind.test_find_everythingcCs$|d}dt|dks tdS)z;Test an optimization that finds all tags with a given name.zfoobarbazr(aNr)rrrrtest_find_everything_with_name@s z'TestFind.test_find_everything_with_nameN) __name__ __module__ __qualname____doc__rr!r&r,r.rrrrr&s rc@s8eZdZdZddZddZddZdd Zd d Zd S) TestFindAllz%Basic tests of the find_all() method.cCsz|d}|jdddgks t|jddgdddgksFoobar»barr Fooz.*»TN)rr+rrecompilerrrrtest_find_all_text_nodesHs   z$TestFindAll.test_find_all_text_nodescCs|d}||jddddddg||jddddg||jdd ddddd d g||jdd ddddd d gd S)z7You can limit the number of items returned by find_all.z(12345r-limit1r3r 45rNrassert_selectsr+rrrrtest_find_all_limitVs   zTestFindAll.test_find_all_limitcCs:|d}||ddddg||jdddgdS) Nz!123r-rr;r=foor"r>)rrCrrrrr%test_calling_a_tag_is_calling_findallbs z1TestFindAll.test_calling_a_tag_is_calling_findallcCs.|d}g}||g||ks*tdS)N)rappendr+r)rrlrrrTtest_find_all_with_self_referential_data_structure_does_not_cause_infinite_recursiongs  z`TestFindAll.test_find_all_with_self_referential_data_structure_does_not_cause_infinite_recursioncCsX|d}|d}t|ds"t|d}t|ds:t|jdd}t|dsTtdS)z%All find_all calls return a ResultSetrGr-sourceTrEr N)rr+hasattrr)rrresultrrrtest_find_all_resultsetqs    z#TestFindAll.test_find_all_resultsetN) r/r0r1r2r9rDrFrJrNrrrrr3Es   r3c@seZdZddZdS)TestFindAllBasicNamespacescCs<|d}d|djkstd|jddidjks8tdS)Nz04r@z mathml:msqrtr-zsvg:fillZredattrs)rrrrnamerrrrtest_find_by_namespaced_names z7TestFindAllBasicNamespaces.test_find_by_namespaced_nameN)r/r0r1rSrrrrrO~srOc@sheZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZdS)TestFindAllByNamez&Test ways of finding tags by tag name.cCs|d|_dS)NzFirst tag. Second tag. Third Nested tag. tag.rtreerrrr setup_methodszTestFindAllByName.setup_methodcCs||jdddgdSNr- First tag. Nested tag.rCrVr+rWrrrtest_find_all_by_tag_names z+TestFindAllByName.test_find_all_by_tag_namecCs\||jjddddg||jjdddddg||jjdtddddgdS)Nr-rZr Tr[tagrCrVr+r7r8rWrrrtest_find_all_by_name_and_textsz0TestFindAllByName.test_find_all_by_name_and_textcCs||jjddgdS)Nr-r[)rCrVcr+rWrrr!test_find_all_on_non_root_elementsz3TestFindAllByName.test_find_all_on_non_root_elementcCs||dddgdSrY)rCrVrWrrr%test_calling_element_invokes_find_allsz7TestFindAllByName.test_calling_element_invokes_find_allcCs ||jtdddgdSrY)rCrVr+r rWrrrtest_find_all_by_tag_strainersz/TestFindAllByName.test_find_all_by_tag_strainercCs"||jddgdddgdS)Nr-rrZ Second tag.r[r\rWrrrtest_find_all_by_tag_namessz,TestFindAllByName.test_find_all_by_tag_namescCs$||jddddddgdS)NT)r-rrZrer[r\rWrrrtest_find_all_by_tag_dictsz+TestFindAllByName.test_find_all_by_tag_dictcCs$||jtddddgdS)Nz^[ab]$rZrer[r_rWrrrtest_find_all_by_tag_resz)TestFindAllByName.test_find_all_by_tag_recCs,dd}|d}|||ddgdS)NcSs|j|dkSNr#)rRgetr^rrrid_matches_nameszRTestFindAllByName.test_find_all_with_tags_matching_method..id_matches_namezMatch 1. Does not match. Match 2.zMatch 1.zMatch 2.rB)rrlrVrrr'test_find_all_with_tags_matching_methods  z9TestFindAllByName.test_find_all_with_tags_matching_methodcCsx|d}|dd}|dtd}|dddg\}}d|jksJtd|jksXtd|jksftd|jksttdS)NzH
1
2
3
divza dza br>r=)rrr7r8r+rr)rrZr1Zr2Zr3Zr4rrr%test_find_with_multi_valued_attributes z7TestFindAllByName.test_find_with_multi_valued_attributeN)r/r0r1r2rXr]r`rbrcrdrfrgrhrmrorrrrrTs  rTc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!S)"TestFindAllByAttributecCs&|d}||jddddgdS)Nz Matching a. Non-matching Matching b.a. firstr"z Matching a.z Matching b.rBrrVrrrtest_find_all_by_attribute_names z6TestFindAllByAttribute.test_find_all_by_attribute_namecCstdd}dd}||}|jg|j|dks6t|jg|j|ddksTt|jgspt|j|dgddS)Nuםולשutf8u)titlezsomething else)encoderr-r+rdecode)rZpeacedatarrrr%test_find_all_by_utf8_attribute_values    zName match. Class match. Non-match. A tag called 'name1'. Zname1rRzA tag called 'name1'.rRrPz Name match.classZclass2z Class match.rBrrrrrtest_find_all_by_attribute_dicts z6TestFindAllByAttribute.test_find_all_by_attribute_dictcCs|d}||jddddg||jddddg||jdd ddg||dddg||jdd ddg||dddg||dd dgdS) Nz Class 1. Class 2. Class 1. Class 3 and 4. r-r=class_zClass 1.rar>zClass 3 and 4.r@rPrBrrrrrtest_find_all_by_classs z-TestFindAllByAttribute.test_find_all_by_classcCst|d}|jdtdd}||dg|jdtdd}||dg|jdtdd}||dgdS)Nz#Found itZgaror}Found itr-zo b)rr+r7r8rC)rrVfrrr0test_find_by_class_when_multiple_classes_presents zGTestFindAllByAttribute.test_find_by_class_when_multiple_classes_presentcCsd|d}||dtddgdd}||d|gdd}||d|dgdS) NzFound itr-barcSs t|dkSNr:r*valuerrrbig_attribute_value%sznTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_class..big_attribute_valuecSs t|dkSrrrrrrsmall_attribute_value*szpTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_class..small_attribute_valuerrCr+r7r8)rrrrrrr: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|ddksttg|ddkstdS)Nz*r-rEr4zfoo barr}zbar foorr+r)rrr-Za2rrr:test_find_all_with_string_for_attrs_finds_multiple_classes0s zQTestFindAllByAttribute.test_find_all_with_string_for_attrs_finds_multiple_classescCs0|d}tddid}|||dgdS)Nzi Match. Non-match.r#rqrPzMatch.)rr rCr+)rrVstrainerrrr'test_find_all_by_attribute_soupstrainer<s z>TestFindAllByAttribute.test_find_all_by_attribute_soupstrainercCs&|d}||jddddgdS)NID present. No ID present. ID is empty.r-r"zNo ID present.rBrrrrr$test_find_all_with_missing_attributeDs z;TestFindAllByAttribute.test_find_all_with_missing_attributecCs&|d}||jddddgdS)NrTr"z ID present.z ID is empty.rBrrrrr$test_find_all_with_defined_attributeLs   z;TestFindAllByAttribute.test_find_all_with_defined_attributecCs>|d}ddg}||jdd|||jdd|dS)Nz[Unquoted attribute. Quoted attribute.zUnquoted attribute.zQuoted attribute.rr"r=rB)rrVexpectedrrr$test_find_all_with_numeric_attributeUs z;TestFindAllByAttribute.test_find_all_with_numeric_attributecCs,|d}||jdddgdddgdS)Nz1 2 3 No ID.r=r>r@r"rBrrrrr(test_find_all_with_list_attribute_values^s z?TestFindAllByAttribute.test_find_all_with_list_attribute_valuescCs,|d}||jtddddgdS)NzOne a. Two as. Mixed as and bs. One b. No ID.z^a+$r"zOne a.zTwo as.rrrrrr5test_find_all_with_regular_expression_attribute_valuehs zLTestFindAllByAttribute.test_find_all_with_regular_expression_attribute_valuecCsB|d}|j}|g|jdddks(tg|jdddks>tdS)Nzfoobarfoor-rEr r4rr-r+rrrr-rrr'test_find_by_name_and_containing_stringus z>TestFindAllByAttribute.test_find_by_name_and_containing_stringcCs*|d}|d|jdddks&tdS)Nz"foofoor-rEr rrrrr=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(tg|jdddks>tdS)Nz"foofoor(rE)r#rrr4rrrrr,test_find_by_attribute_and_containing_strings zCTestFindAllByAttribute.test_find_by_attribute_and_containing_stringN)r/r0r1rsryr|rrrrrrrrrrrrrrrrrrps       rpc@seZdZdZddZdS) TestSmoothzTest Tag.smooth.cCs|d}|j}|d|d|td|td|d|}t||d}|d|d ||d|jjkstd t |j kst| d t |j kstd |j d kstd|jjkstd|j dkstd|j dkstdS)Nz
a
rraz Comment 1z Comment 2dspanr=rabcrZ12rr() rrnrHrdefault_builderrrrrr*contentsZsmooth)rrrnbuilderrrrr test_smooths(        zTestSmooth.test_smoothN)r/r0r1r2rrrrrrsrc@seZdZdZddZdS) TestIndexzTest Tag.indexc CsZ|d}|j}t|jD]\}}|||kstqtt|dW5QRXdS)Nah
Identical Not identical Identical Identical with child Also not identical Identical with child
r) rrn enumeraterindexrpytestraises ValueError)rrVrnielementrrr test_indexs   zTestIndex.test_indexN)r/r0r1r2rrrrrrsrc@sXeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS)TestParentOperationsz;Test navigation and searching through an element's parents.cCs|d|_|jj|_dS)Na1