
    n9d                     J   d Z ddlmZ ddlZddlZddlZddlmZ ddlm	Z	m
Z
 ddlmZmZmZmZmZmZmZmZmZmZmZ ddlmZ  G d	 d
e          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z  G d de          Z! G d de!          Z" G d de!          Z# G d d e          Z$ G d! d"e$          Z% G d# d$e$          Z& G d% d&e          Z' G d' d(e          Z(dS ))a8  Tests 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_registryHTMLParserTreeBuilder)CDataCommentDeclarationDoctype	FormatterNavigableStringScriptSoupStrainer
StylesheetTagTemplateString   )SoupTestc                   0    e Zd ZdZd Zd Zd Zd Zd ZdS )TestFindzBasic tests of the find() method.

    find() just calls find_all() with limit=1, so it's not tested all
    that thouroughly here.
    c                 p    |                      d          }|                    d          j        dk    sJ d S )Nz <a>1</a><b>2</b><a>3</a><b>4</b>b2)soupfindstringselfr   s     3lib/python3.11/site-packages/bs4/tests/test_tree.pytest_find_tagzTestFind.test_find_tag-   s8    yy;<<yy~~$++++++    c                 h    |                      d          }|                    d          dk    sJ d S )Nu   <h1>Räksmörgås</h1>   Räksmörgåsr   )r   r   r   s     r   test_unicode_text_findzTestFind.test_unicode_text_find1   s9    yy122yyy00OCCCCCCr    c                     |                      d          }t          |           d|                    d          j        k    sJ d S )Nu&   <h1 id="Räksmörgås">here it is</h1>z
here it isr"   id)r   strr   textr   s     r   test_unicode_attribute_findz$TestFind.test_unicode_attribute_find5   sG    yyABBD			tyyOy<<AAAAAAAr    c                 ~    |                      d          }dt          |                                          k    sJ dS )z)Test an optimization that finds all tags.<a>foo</a><b>bar</b>   Nr   lenfind_allr   s     r   test_find_everythingzTestFind.test_find_everything;   s;    yy/00C((((((((r    c                     |                      d          }dt          |                    d                    k    sJ dS )z;Test an optimization that finds all tags with a given name.z<a>foo</a><b>bar</b><a>baz</a>r-   aNr.   r   s     r   test_find_everything_with_namez'TestFind.test_find_everything_with_name@   s?    yy9::Cc**++++++++r    N)	__name__
__module____qualname____doc__r   r$   r*   r1   r4    r    r   r   r   &   so         , , ,D D DB B B) ) )
, , , , ,r    r   c                   0    e Zd ZdZd Zd Zd Zd Zd ZdS )TestFindAllz%Basic tests of the find_all() method.c                 F   |                      d          }|                    d          dgk    sJ |                    ddg          ddgk    sJ |                    t          j        d                    g dk    sJ |                    d          g dk    sJ dS )	z'You can search the tree for text nodes.u   <html>Foo<b>bar</b>»</html>barr#   Fooz.*)r>   r=      »TN)r   r0   recompiler   s     r   test_find_all_text_nodesz$TestFindAll.test_find_all_text_nodesH   s    yy9::}}E}**ug5555 }}UEN}33u~EEEE}}BJt$4$4}559O9O9OOOOO}}D}))-C-C-CCCCCCCr    c                    |                      d          }|                     |                    dd          g d           |                     |                    dd          dg           |                     |                    dd          g d	           |                     |                    dd
          g d	           dS )z7You can limit the number of items returned by find_all.z(<a>1</a><a>2</a><a>3</a><a>4</a><a>5</a>r3      limit)1r   3r   rG   
   )rG   r   rH   45r   Nr   assert_selectsr0   r   s     r   test_find_all_limitzTestFindAll.test_find_all_limitV   s    yyCDDDMM#QM77IIIDMM#QM77#???MM#RM((*C*C*C	E 	E 	E 	MM#QM'')B)B)B	D 	D 	D 	D 	Dr    c                     |                      d          }|                      |dd          dg           |                     |                    d          dg           d S )	Nz!<a>1</a><b>2<a id='foo'>3</a></b>r3   r   rE   rG   foor&   rH   )r   rM   r   r   s     r   %test_calling_a_tag_is_calling_findallz1TestFindAll.test_calling_a_tag_is_calling_findallb   sg    yy<==DDA...666DFFeF,,se44444r    c                     |                      d          }g }|                    |           g |                    |          k    sJ d S )N<a></a>)r   appendr0   )r   r   ls      r   Ttest_find_all_with_self_referential_data_structure_does_not_cause_infinite_recursionz`TestFindAll.test_find_all_with_self_referential_data_structure_does_not_cause_infinite_recursiong   sL    yy##	 T]]1%%%%%%%%r    c                    |                      d          }|                    d          }t          |d          sJ |                    d          }t          |d          sJ |                    d          }t          |d          sJ dS )z%All find_all calls return a ResultSetrS   r3   sourceTrP   r#   N)r   r0   hasattr)r   r   results      r   test_find_all_resultsetz#TestFindAll.test_find_all_resultsetq   s    yy##s##vx(((((t$$vx(((((e,,vx(((((((r    N)	r5   r6   r7   r8   rB   rN   rQ   rV   r[   r9   r    r   r;   r;   E   sk        //D D D
D 
D 
D5 5 5
& & &
) 
) 
) 
) 
)r    r;   c                       e Zd Zd ZdS )TestFindAllBasicNamespacesc                     |                      d          }d|                    d          j        k    sJ d|                    ddi          j        k    sJ d S )Nz0<mathml:msqrt>4</mathml:msqrt><a svg:fill="red">rJ   zmathml:msqrtr3   zsvg:fillredattrs)r   r   r   namer   s     r   test_find_by_namespaced_namez7TestFindAllBasicNamespaces.test_find_by_namespaced_name   sa    yyKLLdii//66666diie'=i>>CCCCCCCr    N)r5   r6   r7   rc   r9   r    r   r]   r]   ~   s(        D D D D Dr    r]   c                   T    e Zd ZdZd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zd Zd ZdS )TestFindAllByNamez&Test ways of finding tags by tag name.c                 :    |                      d          | _        d S )Nz<a>First tag.</a>
                                  <b>Second tag.</b>
                                  <c>Third <a>Nested tag.</a> tag.</c>r   treer   s    r   setup_methodzTestFindAllByName.setup_method   s#    YY  J K K			r    c                 f    |                      | j                            d          ddg           d S Nr3   
First tag.Nested tag.rM   rh   r0   ri   s    r   test_find_all_by_tag_namez+TestFindAllByName.test_find_all_by_tag_name   sC    Is##lM%B	D 	D 	D 	D 	Dr    c                 T   |                      | j                            dd          dg           |                      | j                            dd          ddg           |                      | j                            dt          j        d                    ddg           d S )Nr3   rm   r#   Trn   tagrM   rh   r0   r@   rA   ri   s    r   test_find_all_by_name_and_textz0TestFindAllByName.test_find_all_by_name_and_text   s    Is<88<.	J 	J 	J 	Is400<2O	Q 	Q 	Q 	Is2:e+<+<===)	+ 	+ 	+ 	+ 	+r    c                 n    |                      | j        j                            d          dg           d S )Nr3   rn   )rM   rh   cr0   ri   s    r   !test_find_all_on_non_root_elementz3TestFindAllByName.test_find_all_on_non_root_element   s2    DIK0055GGGGGr    c                 \    |                      |                     d          ddg           d S rl   )rM   rh   ri   s    r   %test_calling_element_invokes_find_allz7TestFindAllByName.test_calling_element_invokes_find_all   s-    DIIcNN\=,IJJJJJr    c                     |                      | j                            t          d                    ddg           d S rl   )rM   rh   r0   r   ri   s    r   test_find_all_by_tag_strainerz/TestFindAllByName.test_find_all_by_tag_strainer   sG    I|C0011=)	+ 	+ 	+ 	+ 	+r    c                 j    |                      | j                            ddg          g d           d S )Nr3   r   rm   zSecond tag.rn   ro   ri   s    r   test_find_all_by_tag_namesz,TestFindAllByName.test_find_all_by_tag_names   sD    ISz**888	: 	: 	: 	: 	:r    c                 l    |                      | j                            ddd          g d           d S )NT)r3   r   r}   ro   ri   s    r   test_find_all_by_tag_dictz+TestFindAllByName.test_find_all_by_tag_dict   sH    Id$7788888	: 	: 	: 	: 	:r    c                     |                      | j                            t          j        d                    g d           d S )Nz^[ab]$r}   rs   ri   s    r   test_find_all_by_tag_rez)TestFindAllByName.test_find_all_by_tag_re   sJ    Irz(3344888	: 	: 	: 	: 	:r    c                     d }|                      d          }|                     |                    |          ddg           d S )Nc                 >    | j         |                     d          k    S Nr'   )rb   getrr   s    r   id_matches_namezRTestFindAllByName.test_find_all_with_tags_matching_method.<locals>.id_matches_name   s    8swwt}},,r    z<a id="a">Match 1.</a>
                            <a id="1">Does not match.</a>
                            <b id="b">Match 2.</a>zMatch 1.zMatch 2.rL   )r   r   rh   s      r   'test_find_all_with_tags_matching_methodz9TestFindAllByName.test_find_all_with_tags_matching_method   sj    	- 	- 	- yy 6 7 7 	MM/**Z,D	F 	F 	F 	F 	Fr    c                 J   |                      d          }|                    dd          }|                    dt          j        d                    }|                    dddg          \  }}d|j        k    sJ d|j        k    sJ d|j        k    sJ d|j        k    sJ d S )NzH<div class='a b'>1</div><div class='a c'>2</div><div class='a d'>3</div>divza dza brH   rG   )r   r   r@   rA   r0   r   )r   r   r1r2r3r4s         r   %test_find_with_multi_valued_attributez7TestFindAllByName.test_find_with_multi_valued_attribute   s    yyV
 
 YYue$$YYubj0011uuen55Bbibibibir    N)r5   r6   r7   r8   rj   rp   rt   rw   ry   r{   r~   r   r   r   r   r9   r    r   re   re      s        00K K K
D D D
	+ 	+ 	+H H HK K K+ + +
: : :
: : :
: : :
F F F
  
  
  
  
 r    re   c                   n    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zd Zd Zd Zd Zd Zd ZdS )TestFindAllByAttributec                     |                      d          }|                     |                    d          ddg           d S )Nz
                         <a id="first">Matching a.</a>
                         <a id="second">
                          Non-matching <b id="first">Matching b.</b>a.
                         </a>firstr&   zMatching a.zMatching b.rL   r   rh   s     r   test_find_all_by_attribute_namez6TestFindAllByAttribute.test_find_all_by_attribute_name   sX     yy ! " "
 	DMMWM55)=9	; 	; 	; 	; 	;r    c                    d                     d          }d                     d          }|                     |          }|j        g|                    |          k    sJ |j        g|                    |                    d                    k    sJ |j        gs J |                    |dg                      d S )Nu   םולשutf8u   <a title="םולש"></a>)titlezsomething else)encoder   r3   r0   decode)r   peacedatar   s       r   %test_find_all_by_utf8_attribute_valuez<TestFindAllByAttribute.test_find_all_by_utf8_attribute_value   s    !!&)))0088yyx4==u=555555x4==u||F/C/C=DDDDDDxGGe5E-FGGGGGGGr    c                 :   |                      d          }|                     |                    d          dg           |                     |                    ddi          dg           |                     |                    dd	i          d
g           d S )Na0  
                         <a name="name1" class="class1">Name match.</a>
                         <a name="name2" class="class2">Class match.</a>
                         <a name="name3" class="class3">Non-match.</a>
                         <name1>A tag called 'name1'.</name1>
                         name1rb   zA tag called 'name1'.rb   r`   zName match.classclass2zClass match.rL   r   s     r   test_find_all_by_attribute_dictz6TestFindAllByAttribute.test_find_all_by_attribute_dict   s     yy    	DMMwM7734	6 	6 	6 	DMM'0BMCC)?	, 	, 	, 	DMM80DMEE*+	- 	- 	- 	- 	-r    c                    |                      d          }|                     |                    dd          dg           |                     |                    dd          dg           |                     |                    dd	          dg           |                     |                    dd          dg           |                     |                    d
          ddg           |                     |                    dd          dg           |                     |                    dd	          dg           d S )Nz
                         <a class="1">Class 1.</a>
                         <a class="2">Class 2.</a>
                         <b class="1">Class 1.</b>
                         <c class="3 4">Class 3 and 4.</c>
                         r3   rG   class_zClass 1.rv   rH   zClass 3 and 4.rJ   r`   rL   r   s     r   test_find_all_by_classz-TestFindAllByAttribute.test_find_all_by_class   s?   yy    	DMM#cM::ZLIIIDMM#cM::=M<NOOODMM#cM::=M<NOOO 	DMM#s33j\BBBDMMM44z:6NOOODMM#s336F5GHHHDMM#s336F5GHHHHHr    c                    |                      d          }|                    dt          j        d                    }|                     |dg           |                    dt          j        d                    }|                     |dg           |                    dt          j        d                    }|                     |dg           d S )Nz#<gar class='foo bar'>Found it</gar>garor   Found itr3   zo b)r   r0   r@   rA   rM   )r   rh   fs      r   0test_find_by_class_when_multiple_classes_presentzGTestFindAllByAttribute.test_find_by_class_when_multiple_classes_present  s    yy>??MM%
3M88A
|,,,MM%
3M88A
|,,, MM%
5(9(9M::A
|,,,,,r    c                 `   |                      d          }|                     |                    dt          j        d                    dg           d }|                     |                    d|          g            d }|                     |                    d|          dg           d S )Nz<a class='bar'>Found it</a>r3   bar   c                 (    t          |           dk    S NrD   r/   values    r   big_attribute_valueznTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_class.<locals>.big_attribute_value%  s    u::>!r    c                 (    t          |           dk    S r   r   r   s    r   small_attribute_valuezpTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_class.<locals>.small_attribute_value*  s    u::?"r    r   rM   r0   r@   rA   )r   r   r   r   s       r   :test_find_all_with_non_dictionary_for_attrs_finds_by_classzQTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_class   s    yy677DMM#rz$/?/?@@:,OOO	" 	" 	" 	DMM#/BCCRHHH	# 	# 	# 	MM#455
|	E 	E 	E 	E 	Er    c                    |                      d          }|                    d          \  }}||gsJ |                    dd                      |gsJ |                    dd                      |g|                    dd          k    sJ |g|                    dd          k    sJ g |                    dd          k    sJ d S )Nz*<a class="foo bar"></a><a class="foo"></a>r3   rP   r=   zfoo barr   zbar foor   r0   )r   r   r3   a2s       r   :test_find_all_with_string_for_attrs_finds_multiple_classeszQTestFindAllByAttribute.test_find_all_with_string_for_attrs_finds_multiple_classes0  s    yyEFFc""22w11c511111s--DMM#u----- sdmmC	m::::::sdmmC333333T]]3	22222222r    c                     |                      d          }t          ddi          }|                     |                    |          dg           d S )Nzi
                         <a id="first">Match.</a>
                         <a id="second">Non-match.</a>r'   r   r`   zMatch.)r   r   rM   r0   )r   rh   strainers      r   'test_find_all_by_attribute_soupstrainerz>TestFindAllByAttribute.test_find_all_by_attribute_soupstrainer<  s[    yy : ; ;  tg&6777DMM(33hZ@@@@@r    c                     |                      d          }|                     |                    dd           dg           d S )N<a id="1">ID present.</a>
                            <a>No ID present.</a>
                            <a id="">ID is empty.</a>r3   r&   zNo ID present.rL   r   s     r   $test_find_all_with_missing_attributez;TestFindAllByAttribute.test_find_all_with_missing_attributeD  sN     yy 9 : : 	DMM#$M77:J9KLLLLLr    c                     |                      d          }|                     |                    d          ddg           d S )Nr   Tr&   zID present.zID is empty.rL   r   s     r   $test_find_all_with_defined_attributez;TestFindAllByAttribute.test_find_all_with_defined_attributeL  s]     yy 9 : : 	MMTM""]N$C	E 	E 	E 	E 	Er    c                     |                      d          }ddg}|                     |                    d          |           |                     |                    d          |           d S )Nz[<a id=1>Unquoted attribute.</a>
                            <a id="1">Quoted attribute.</a>zUnquoted attribute.zQuoted attribute.r   r&   rG   rL   )r   rh   expecteds      r   $test_find_all_with_numeric_attributez;TestFindAllByAttribute.test_find_all_with_numeric_attributeU  sw    yy ? @ @ *+>?DMMQM//:::DMMSM118<<<<<r    c                     |                      d          }|                     |                    g d          ddg           d S )Nz<a id="1">1</a>
                            <a id="2">2</a>
                            <a id="3">3</a>
                            <a>No ID.</a>)rG   rH   rJ   r&   rG   rH   rL   r   s     r   (test_find_all_with_list_attribute_valuesz?TestFindAllByAttribute.test_find_all_with_list_attribute_values^  s[     yy - . . 	DMM___M==:	' 	' 	' 	' 	'r    c                     |                      d          }|                     |                    t          j        d                    ddg           d S )Nz<a id="a">One a.</a>
                            <a id="aa">Two as.</a>
                            <a id="ab">Mixed as and bs.</a>
                            <a id="b">One b.</a>
                            <a>No ID.</a>z^a+$r&   zOne a.zTwo as.r   r   s     r   5test_find_all_with_regular_expression_attribute_valuezLTestFindAllByAttribute.test_find_all_with_regular_expression_attribute_valueh  sb     yy - . . 	DMMRZ-?-?M@@$i0	2 	2 	2 	2 	2r    c                     |                      d          }|j        }|g|                    dd          k    sJ g |                    dd          k    sJ d S )Nz<b>foo</b><b>bar</b><a>foo</a>r3   rP   r#   r=   r   r3   r0   r   r   r3   s      r   'test_find_by_name_and_containing_stringz>TestFindAllByAttribute.test_find_by_name_and_containing_stringu  se    yy9::FsdmmCm666666T]]3u]55555555r    c                     |                      d          }|                    d          |                    dd          k    sJ d S )Nz"<a>foo</a><a><b><c>foo</c></b></a>r3   rP   r#   r   r   s     r   =test_find_by_name_and_containing_string_when_string_is_buriedzTTestFindAllByAttribute.test_find_by_name_and_containing_string_when_string_is_buried|  sG    yy=>>}}S!!T]]3u]%E%EEEEEEEr    c                     |                      d          }|j        }|g|                    dd          k    sJ g |                    dd          k    sJ d S )Nz"<b id="1">foo</b><a id="2">foo</a>r-   rP   )r'   r   r   r=   r   r   s      r   ,test_find_by_attribute_and_containing_stringzCTestFindAllByAttribute.test_find_by_attribute_and_containing_string  se    yy=>>Fsdmmqm777777T]]a]66666666r    N)r5   r6   r7   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r9   r    r   r   r      s       	; 	; 	;H H H- - -*I I I(- - -E E E 
3 
3 
3A A AM M ME E E= = =' ' '2 2 26 6 6F F F7 7 7 7 7r    r   c                       e Zd ZdZd ZdS )
TestSmoothzTest Tag.smooth.c                 V   |                      d          }|j        }|                    d           |                    d           |                    t          d                     |                    t          d                     |                    d           |                                 }t          ||d          }|                    d           |                    d	           |                    |           d |j        j        k    sJ d
t          |j	                  k    sJ |
                                 dt          |j	                  k    sJ d|j	        d         k    sJ d|j        j        k    sJ d|j	        d         k    sJ d|j	        d         k    sJ d S )Nz<div>a</div>r   rv   z	Comment 1z	Comment 2dspanrG   r         abcr   12r   r-   )r   r   rT   r   default_builderr   r   r   r/   contentssmooth)r   r   r   builderr   s        r   test_smoothzTestSmooth.test_smooth  s   yy((h

3

3

7;''(((

7;''(((

3&&((4&))CC

4 sx&&&&C%%%%%%

C%%%%%%
 Q'''' sx&&&&
 cl1o----cl1o------r    N)r5   r6   r7   r8   r   r9   r    r   r   r     s)        &. &. &. &. &.r    r   c                       e Zd ZdZd ZdS )	TestIndexzTest Tag.indexc                 6   |                      d          }|j        }t          |j                  D ] \  }}||                    |          k    sJ !t          j        t                    5  |                    d           d d d            d S # 1 swxY w Y   d S )Nah  <div>
                            <a>Identical</a>
                            <b>Not identical</b>
                            <a>Identical</a>

                            <c><d>Identical with child</d></c>
                            <b>Also not identical</b>
                            <c><d>Identical with child</d></c>
                            </div>r   )r   r   	enumerater   indexpytestraises
ValueError)r   rh   r   ielements        r   
test_indexzTestIndex.test_index  s    yy & ' ' h#CL11 	+ 	+JAw		'*******]:&& 	 	JJqMMM	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	s   +BBBN)r5   r6   r7   r8   r   r9   r    r   r   r     s)            r    r   c                   H    e Zd ZdZd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 ZdS )TestParentOperationsz;Test navigation and searching through an element's parents.c                 \    |                      d          | _        | j        j        | _        d S )Na1  <ul id="empty"></ul>
                                 <ul id="top">
                                  <ul id="middle">
                                   <ul id="bottom">
                                    <b>Start here</b>
                                   </ul>
                                  </ul>)r   rh   r   startri   s    r   rj   z!TestParentOperations.setup_method  s-    II + , ,	 Y[


r    c                     | j         j        d         dk    sJ | j         j        j        d         dk    sJ | j         j        j        j        d         dk    sJ d S )Nr'   bottommiddletop)r   parentri   s    r   test_parentz TestParentOperations.test_parent  sb    z &(2222z '-9999z '.t4======r    c                 N    | j         j        d         }|j        | j         k    sJ d S )Nr   )rh   r   r   )r   top_tags     r   %test_parent_of_top_tag_is_soup_objectz:TestParentOperations.test_parent_of_top_tag_is_soup_object  s,    )$Q'~******r    c                 *    d | j         j        k    sJ d S N)rh   r   ri   s    r   test_soup_object_has_no_parentz3TestParentOperations.test_soup_object_has_no_parent  s    ty'''''''r    c                     |                      | j                            d          g d           |                      | j                            dd          dg           d S )Nul)r   r   r   r   r&   )assert_selects_idsr   find_parentsri   s    r   test_find_parentsz&TestParentOperations.test_find_parents  s|    J##D))+F+F+F	H 	H 	HJ##DX#66
	D 	D 	D 	D 	Dr    c                     | j                             d          d         dk    sJ | j                             dd          d         dk    sJ d S )Nr   r'   r   r   r&   )r   find_parentri   s    r   test_find_parentz%TestParentOperations.test_find_parent  sY    z%%d++D1X====z%%du%55d;uDDDDDDr    c                 `    | j                             d          }|j        j        dk    sJ d S )N
Start herer#   r   )rh   r   r   rb   r   r)   s     r   test_parent_of_text_elementz0TestParentOperations.test_parent_of_text_element  s3    y~~\~22{3&&&&&&r    c                 ~    | j                             d          }|                    d          d         dk    sJ d S )Nr  r#   r   r'   r   )rh   r   r  r	  s     r   test_text_element_find_parentz2TestParentOperations.test_text_element_find_parent  sA    y~~\~22%%d+x777777r    c                 V    d | j         j        D             }|sJ dddk    g            d S )Nc                 4    g | ]}|d|j         v |d         S r   r`   ).0r   s     r   
<listcomp>z>TestParentOperations.test_parent_generator.<locals>.<listcomp>  s:     C C CF(TV\-A-A $<-A-A-Ar    r   r   r   )r   parents)r   r  s     r   test_parent_generatorz*TestParentOperations.test_parent_generator  sM    C Cdj.@ C C C558u#4555555r    N)r5   r6   r7   r8   rj   r   r   r   r  r  r
  r  r  r9   r    r   r   r     s        EE! ! !> > >
+ + +( ( (D D DE E E' ' '8 8 86 6 6 6 6r    r   c                       e Zd Zd ZdS )ProximityTestc                 :    |                      d          | _        d S )Nzg<html id="start"><head></head><body><b id="1">One</b><b id="2">Two</b><b id="3">Three</b></body></html>rg   ri   s    r   rj   zProximityTest.setup_method  s     IIuw w			r    Nr5   r6   r7   rj   r9   r    r   r  r    s(        w w w w wr    r  c                   H     e Zd Z fdZd Zd Zd Zd Zd Zd Z	d Z
 xZS )	TestNextOperationsc                 x    t          t          |                                            | j        j        | _        d S r   )superr  rj   rh   r   r   r   	__class__s    r   rj   zTestNextOperations.setup_method  s/     $''44666Y[


r    c                 d    | j         j        dk    sJ | j         j        j        d         dk    sJ d S )NOner'   r   )r   next_elementri   s    r   	test_nextzTestNextOperations.test_next  s<    z&%////z&3D9S@@@@@@r    c                 V    | j                             d          }|j        d k    sJ d S NThreer#   )rh   r   r  )r   lasts     r   test_next_of_last_item_is_nonez1TestNextOperations.test_next_of_last_item_is_none
  s1    y~~W~-- D((((((r    c                 *    | j         j        d k    sJ d S r   )rh   r  ri   s    r   test_next_of_root_is_nonez,TestNextOperations.test_next_of_root_is_none  s    y%------r    c                     |                      | j                            d          ddg           | j                            d           |                      | j                            d          dg           d S )Nr   Twor#  rD   r&   )rM   r   find_all_nextri   s    r   test_find_all_nextz%TestNextOperations.test_find_all_next  sw    DJ44S99E7;KLLL
  A &&&DJ444::WIFFFFFr    c                     | j                             d          d         dk    sJ | j                             d          dk    sJ d S )Nr   r'   r   r#  r#   )r   	find_nextri   s    r   test_find_nextz!TestNextOperations.test_find_next  sR    z##C((.#5555z##7#33w>>>>>>r    c                     | j                             d          }|                    d          j        dk    sJ |                     |                    d          ddg           d S )Nr  r#   r   r)  r#  )rh   r   r-  r   rM   r*  r	  s     r   test_find_next_for_text_elementz2TestNextOperations.test_find_next_for_text_element  sf    y~~U~++~~c"")U2222D..s33eW5EFFFFFr    c                     | j                             d          }d |j        D             }|\  }}|d         dk    sJ |dk    sJ d S )Nr)  r#   c                     g | ]}|S r9   r9   r  nodes     r   r  z:TestNextOperations.test_next_generator.<locals>.<listcomp>"  s    ;;;td;;;r    r'   rH   r#  )rh   r   next_elements)r   r   
successorsrr   r   s        r   test_next_generatorz&TestNextOperations.test_next_generator   sb    	e,,;;u':;;;
"X4yC7""""""r    )r5   r6   r7   rj   r   r%  r'  r+  r.  r0  r7  __classcell__r  s   @r   r  r     s        ! ! ! ! !A A A) ) ). . .G G G
? ? ?G G G
# # # # # # #r    r  c                   H     e Zd Z fdZd Zd Zd Zd Zd Zd Z	d Z
 xZS )	TestPreviousOperationsc                     t          t          |                                            | j                            d          | _        d S r"  )r  r;  rj   rh   r   endr  s    r   rj   z#TestPreviousOperations.setup_method*  s;    $d++88:::9>>>11r    c                 d    | j         j        d         dk    sJ | j         j        j        dk    sJ d S )Nr'   rH   r)  )r=  previous_elementri   s    r   test_previousz$TestPreviousOperations.test_previous.  s<    x(.#5555x(9UBBBBBBr    c                 T    | j                             d          }|j        d k    sJ d S )Nhtml)rh   r   r?  )r   r   s     r   #test_previous_of_first_item_is_nonez:TestPreviousOperations.test_previous_of_first_item_is_none2  s/    	v&&%------r    c                 *    | j         j        d k    sJ d S r   )rh   r?  ri   s    r   test_previous_of_root_is_nonez4TestPreviousOperations.test_previous_of_root_is_none6  s    y)T111111r    c                     |                      | j                            d          g d           |                      | j                            d          dg           d S )Nr   r#  r)  r  r   r&   r  )rM   r=  find_all_previousri   s    r   test_find_all_previousz-TestPreviousOperations.test_find_all_previous:  sl     	H&&s++-D-D-D	F 	F 	FDH66!6<<ugFFFFFr    c                     | j                             d          d         dk    sJ | j                             d          dk    sJ d S )Nr   r'   rH   r  r#   )r=  find_previousri   s    r   test_find_previousz)TestPreviousOperations.test_find_previousB  sR    x%%c**40C7777x%%U%33u<<<<<<r    c                     | j                             d          }|                    d          j        dk    sJ |                     |                    d          g d           d S )Nr#  r#   r   rG  )rh   r   rK  r   rM   rH  r	  s     r   #test_find_previous_for_text_elementz:TestPreviousOperations.test_find_previous_for_text_elementF  sz    y~~W~--!!#&&-8888""3'')@)@)@	B 	B 	B 	B 	Br    c                     | j                             d          }d |j        D             }|\  }}}}|d         dk    sJ |j        dk    sJ |j        dk    sJ |j        dk    sJ d S )	Nr  r#   c                     g | ]}|S r9   r9   r3  s     r   r  zBTestPreviousOperations.test_previous_generator.<locals>.<listcomp>N  s    AAAAAAr    r'   rG   bodyheadrB  )rh   r   previous_elementsrb   )r   r   predecessorsr   rQ  rR  rB  s          r   test_previous_generatorz.TestPreviousOperations.test_previous_generatorL  s    	e,,AA)@AAA +4tw#~~~~yF""""yF""""yF""""""r    )r5   r6   r7   rj   r@  rC  rE  rI  rL  rN  rU  r8  r9  s   @r   r;  r;  (  s        2 2 2 2 2C C C. . .2 2 2G G G= = =B B B
# 
# 
# 
# 
# 
# 
#r    r;  c                       e Zd Zd ZdS )SiblingTestc                     d}t          j        d                              d|          }|                     |          | _        d S )Na  <html>
                    <span id="1">
                     <span id="1.1"></span>
                    </span>
                    <span id="2">
                     <span id="2.1"></span>
                    </span>
                    <span id="3">
                     <span id="3.1"></span>
                    </span>
                    <span id="4"></span>
                    </html>z\n\s* )r@   rA   subr   rh   )r   markups     r   rj   zSiblingTest.setup_method[  sA     H%%))"f55IIf%%			r    Nr  r9   r    r   rW  rW  Y  s#        & & & & &r    rW  c                   B     e Zd Z fdZd Zd Zd Zd Zd Zd Z	 xZ
S )TestNextSiblingc                     t          t          |                                            | j                            d          | _        d S )NrG   r&   )r  r]  rj   rh   r   r   r  s    r   rj   zTestNextSibling.setup_methodp  s:    ot$$11333Y^^s^++


r    c                 *    | j         j        d k    sJ d S r   )rh   next_siblingri   s    r   !test_next_sibling_of_root_is_nonez1TestNextSibling.test_next_sibling_of_root_is_nonet  s    y%------r    c                     | j         j        d         dk    sJ | j         j        j        d         dk    sJ | j         j        d         dk    sJ d S )Nr'   r   rH   1.1)r   r`  r  ri   s    r   test_next_siblingz!TestNextSibling.test_next_siblingw  s^    z&t,3333z&3D9S@@@@ z&t,555555r    c                     | j         j        j        d k    sJ | j                             d          }|j        d k    sJ | j                             d          }|j        d k    sJ d S )Nrc  r&   rJ   )rh   rB  r`  r   )r   nested_span	last_spans      r   test_next_sibling_may_not_existz/TestNextSibling.test_next_sibling_may_not_exist~  sr    y~*d2222innn..'4////INNcN**	%------r    c                 R    | j                             d          d         dk    sJ d S )Nr   r'   r   )r   find_next_siblingri   s    r   test_find_next_siblingz&TestNextSibling.test_find_next_sibling  s.    z++F33D9S@@@@@@r    c                     |                      | j                            d          g d           |                      | j                            d          dg           d S )Nr   )r   rH   rJ   rH   r&   )r  r   find_next_siblingsri   s    r   test_next_siblingsz"TestNextSibling.test_next_siblings  sf    
 = =f E E-oo	/ 	/ 	/ 	
 = = = E EuMMMMMr    c                 h   |                      d          }|                    d          }|j        j        dk    sJ |j        j        dk    sJ |                     |                    d          dg           |                    d          dk    sJ |                    d          d k    sJ d S )NFoo<b>bar</b>bazr>   r#   r   bazr=   nonesuch)r   r   r`  rb   rM   rm  rj  r   r   r   s      r   "test_next_sibling_for_text_elementz2TestNextSibling.test_next_sibling_for_text_element  s    yy+,,			''!&#----!.%7777E44S99E7CCC&&e&44====&&j&99TAAAAAAr    )r5   r6   r7   rj   ra  rd  rh  rk  rn  rt  r8  r9  s   @r   r]  r]  n  s        , , , , ,. . .6 6 6. . .A A AN N NB B B B B B Br    r]  c                   B     e Zd Z fdZd Zd Zd Zd Zd Zd Z	 xZ
S )TestPreviousSiblingc                     t          t          |                                            | j                            d          | _        d S )NrJ   r&   )r  rv  rj   rh   r   r=  r  s    r   rj   z TestPreviousSibling.setup_method  s;    !4((557779>>S>))r    c                 *    | j         j        d k    sJ d S r   )rh   previous_siblingri   s    r   %test_previous_sibling_of_root_is_nonez9TestPreviousSibling.test_previous_sibling_of_root_is_none  s    y)T111111r    c                     | j         j        d         dk    sJ | j         j        j        d         dk    sJ | j         j        d         dk    sJ d S )Nr'   rH   r   z3.1)r=  ry  r?  ri   s    r   test_previous_siblingz)TestPreviousSibling.test_previous_sibling  s^    x(.#5555x(9$?3FFFF x(.%777777r    c                     | j         j        j        d k    sJ | j                             d          }|j        d k    sJ | j                             d          }|j        d k    sJ d S )Nrc  r&   rG   )rh   rB  ry  r   )r   rf  
first_spans      r   #test_previous_sibling_may_not_existz7TestPreviousSibling.test_previous_sibling_may_not_exist  sr    y~.$6666innn..+t3333Y^^s^++
*d222222r    c                 R    | j                             d          d         dk    sJ d S )Nr   r'   rH   )r=  find_previous_siblingri   s    r   test_find_previous_siblingz.TestPreviousSibling.test_find_previous_sibling  s.    x--f55d;sBBBBBBr    c                     |                      | j                            d          g d           |                      | j                            d          dg           d S )Nr   )rH   r   rG   rG   r&   )r  r=  find_previous_siblingsri   s    r   test_previous_siblingsz*TestPreviousSibling.test_previous_siblings  sf     ? ? G G-oo	/ 	/ 	/ 	 ? ?3 ? G G#OOOOOr    c                 h   |                      d          }|                    d          }|j        j        dk    sJ |j        j        dk    sJ |                     |                    d          dg           |                    d          dk    sJ |                    d          d k    sJ d S )Nrp  rq  r#   r   r>   r=   rr  )r   r   ry  rb   rM   r  r  rs  s      r   &test_previous_sibling_for_text_elementz:TestPreviousSibling.test_previous_sibling_for_text_element  s    yy+,,			''%*c1111%6%????E88==wGGG**%*88EAAAA****==EEEEEEr    )r5   r6   r7   rj   rz  r|  r  r  r  r  r8  r9  s   @r   rv  rv    s        * * * * *2 2 28 8 83 3 3C C CP P PF F F F F F Fr    rv  c                   f   e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zd Zd Zd Zd Zd Zd Zd Zej                            dd d g          d             Zd Zd Zd Zd Zd Zd Zd Zd Zd Z d Z!d  Z"d! Z#d" Z$d# Z%d$ Z&d% Z'd& Z(d' Z)d( Z*d) Z+d* Z,d+ Z-d, Z.d- Z/d. Z0d/ Z1d0 Z2d1S )2TestTreeModificationc                 v   |                      d          }d|j        d<   |                                |                     d          k    sJ |j        d= |                                |                     d          k    sJ d|j        d<   |                                |                     d          k    sJ d S )	Nz<a id="1"></a>r-   r'   z<a id="2"></a>rS   rP   id2z<a id2="foo"></a>)r   r3   r   document_forr   s     r   test_attribute_modificationz0TestTreeModification.test_attribute_modification  s    yy)**t{{}} 1 12B C CCCCCF4L{{}} 1 1) < <<<<<u{{}} 1 12E F FFFFFFFr    c                 d    t          j        d                      }|                     d|          }t          ||d          }t          ||d          }d|d<   |j                            d|           |j                            d	|           |j                                        d
k    sJ d S )NrB  z<body></body>r   r3   olzhttp://foo.com/hrefr   r   s4   <body><a href="http://foo.com/"></a><ol></ol></body>)r   lookupr   r   rQ  insertr   )r   r   r   r3   r  s        r   test_new_tag_creationz*TestTreeModification.test_new_tag_creation  s    1")&1133yy'y::gs##w%%%&		A	By!!%\\\\\\\r    c                 >   d}|                      |          }|                    d          }|j        }|                    d                              |j                   |j        |k    sJ |                                |                     d          k    sJ d S )NzT<p id="1">Don't leave me <b>here</b>.</p>
                <p id="2">Don't leave!</p>r   r&   zD<p id="1">Don't leave me .</p>
<p id="2">Don't leave!<b>here</b></p>)r   r   r   rT   r   r   r  )r   docr   second_parabolds        r   !test_append_to_contents_moves_tagz6TestTreeModification.test_append_to_contents_moves_tag  s    /yy~~ii3i''v 			S	  ((( {k)))){{}} 1 19!
 !
 
 
 
 
 
 
r    c                     d}|                      |          }|j        }|                    |j                  }||k    sJ d S )Nz<a></a><b><c></c></b>)r   r3   replace_withrv   r   r)   r   r3   new_as        r   1test_replace_with_returns_thing_that_was_replacedzFTestTreeModification.test_replace_with_returns_thing_that_was_replaced  sA    &yyFtv&&Ezzzzzzr    c                 z    d}|                      |          }|j        }|                                }||k    sJ d S )N<a><b></b><c></c></a>)r   r3   unwrapr  s        r   +test_unwrap_returns_thing_that_was_replacedz@TestTreeModification.test_unwrap_returns_thing_that_was_replaced  s;    &yyF

Ezzzzzzr    c                    |                      d          }|j        }|                                 d |j        k    sJ t	          j        t                    5  |                                 d d d            n# 1 swxY w Y   t	          j        t                    5  |                    |j	                   d d d            d S # 1 swxY w Y   d S )Nz<a><b>Foo</b></a><c>Bar</c>)
r   r3   extractr   r   r   r   r  r  rv   r   s      r   Itest_replace_with_and_unwrap_give_useful_exception_when_tag_has_no_parentz^TestTreeModification.test_replace_with_and_unwrap_give_useful_exception_when_tag_has_no_parent  s5   yy677F			qx]:&& 	 	HHJJJ	 	 	 	 	 	 	 	 	 	 	 	 	 	 	]:&& 	# 	#NN46"""	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	#s$   A88A<?A<CCCc                     d}|                      |          }|j        }|j                            |           |                                |                     |          k    sJ d S )Nz-<a><b></b><c>Foo<d></d></c></a><a><e></e></a>)r   rv   r  r   r  )r   r)   r   rv   s       r   test_replace_tag_with_itselfz1TestTreeModification.test_replace_tag_with_itself  s`    >yyFA{{}} 1 1$ 7 7777777r    c                     d}|                      |          }t          j        t                    5  |j                            |j                   d d d            d S # 1 swxY w Y   d S N<a><b></b></a>)r   r   r   r   r   r  r3   r   r)   r   s      r   1test_replace_tag_with_its_parent_raises_exceptionzFTestTreeModification.test_replace_tag_with_its_parent_raises_exception  s    yy]:&& 	( 	(F'''	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	(s    AA"%A"c                     d}|                      |          }t          j        t                    5  |j                            d|j                   d d d            d S # 1 swxY w Y   d S )Nr  r   )r   r   r   r   r3   r  r  s      r   ,test_insert_tag_into_itself_raises_exceptionzATestTreeModification.test_insert_tag_into_itself_raises_exception  s    yy]:&& 	% 	%FMM!TV$$$	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	%s   !AA#&A#c                 f   |                      d          }d}|                      |          }|                    d|           |j        D ]}t          |t                    rJ t          |j                  \  }}}}d|j        k    sJ d|j        k    sJ d|j        k    sJ d|j        k    sJ dS )	zInserting one BeautifulSoup object into another actually inserts all
        of its children -- you'll never combine BeautifulSoup objects.
        z-<p>And now, a word:</p><p>And we're back.</p>z<p>p2</p><p>p3</p>r   zAnd now, a word:p2p3zAnd we're back.N)r   r  descendants
isinstancer   listchildrenr   )	r   r   r)   	to_insertr   p1r  r  p4s	            r   1test_insert_beautifulsoup_object_inserts_childrenzFTestTreeModification.test_insert_beautifulsoup_object_inserts_children  s     yyHII#IIdOO	Ay!!!! 	4 	4A!!]333333dm,,BB!RY....ry    ry     BI------r    c                    |                      d          }|j        }|j        d         }|                    dd           |j        \  }}|                    d           |                    d           d|j        j        k    sJ d S )Nz<p><a>one</a><b>three</b></p>r   r   tworY  three)r   r3   r   r  replaceWithr   r   )r   r   r3   r   leftrights         r   3test_replace_with_maintains_next_element_throughoutzHTestTreeModification.test_replace_with_maintains_next_element_throughout.  s    yy899FJqM	E je" $&-''''''r    c                 .   |                      d          }|                    d                              d           |                    d          }|j        }|j        |k    sJ |j        |k    sJ |j        j        |k    sJ |j        d k    sJ d S )Nz<b>Argh!</b>Argh!r#   Hooray!)r   r   r  r   r?  r   r  )r   r   new_textr   s       r   test_replace_final_nodez,TestTreeModification.test_replace_final_node=  s    yy((			!!..y99999I9..F(A----!####(5AAAA$,,,,,,r    c                    |                      d          }|j                            dd           |                                |                     d          k    sJ |                    d          }|j        dk    sJ |j        j        |k    sJ |j        dk    sJ |j        j	        |k    sJ |j	        d k    sJ |j        |j
        k    sJ d S )Nz<a><b>Argh!</b><c></c></a>r   r  z!<a><b>Argh!Hooray!</b><c></c></a>r#   r  )r   r   r  r   r  r   r?  r  ry  r`  rv   )r   r   r  s      r   test_consecutive_text_nodesz0TestTreeModification.test_consecutive_text_nodesG  s     yy566a###{{}} 1 1/!
 !
 
 
 
 
 99I9..(G3333(5AAAA(G3333(5AAAA$,,,,$......r    c                     |                      d          }|j                            dd           |j                            dd           ddg|j        j        k    sJ |j        j        d         j        dk    sJ d S )NrS   r   r=   rP   )r   r3   r  r   r  r   s     r   test_insert_stringz'TestTreeModification.test_insert_string\  s{    yy##aau~0000vq!.%777777r    c                    |                                  }|                     d|          }t          ||d          }|                    dd           |j                            d|           |                                |                     d          k    sJ |j        }|j        |k    sJ |j	        |k    sJ |
                    d	          }|j        |k    sJ |j        |k    sJ |j        }|j        |k    sJ |j	        |k    sJ |
                    d	          }|j        |k    sJ |j        |k    sJ |j        |k    sJ d S )
Nz%<a><b>Find</b><c>lady!</c><d></d></a>r  magictagr   ther   z=<a><b>Find</b><magictag>the</magictag><c>lady!</c><d></d></a>Findr#   )r   r   r   r  r3   r   r  r   r`  ry  r   r  r?  rv   r   )r   r   r   	magic_tagb_tagr   c_tagr  s           r   test_insert_tagz$TestTreeModification.test_insert_tage  s   &&((yy3W  F Fgz22	E"""a###{{}} 1 1O!
 !
 
 
 
 

 !Y....)U2222zzz(( I----)T1111%....%2222nnEn**zY&&&&5((((%,,,,,,r    c                     d}|                      |          }|j                            |j                   ||                                k    sJ d S r  )r   r3   rT   r   r   r   r   r   s      r   *test_append_child_thats_already_at_the_endz?TestTreeModification.test_append_child_thats_already_at_the_end  sK    yydft{{}}$$$$$$r    c                     d}|                      |          }|j        |j        |j        |j        |j        |j        g}|j                            |           d|	                                k    sJ d S )Nz1<a><b><c><d><e><f><g></g></f></e></d></c></b></a>z1<a><g></g><f></f><e></e><d></d><c></c><b></b></a>)
r   gr   er   rv   r   r3   extendr   )r   r   r   rU   s       r   test_extendz TestTreeModification.test_extend  sg    ByyVTVTVTVTVTV<aBdkkmmSSSSSSr    get_tagsc                     | S r   r9   r   s    r   <lambda>zTestTreeModification.<lambda>  s     r    c                     | j         S r   )r   r   s    r   r  zTestTreeModification.<lambda>  s    #, r    c                 8   d}|                      |          }|                    dd          }|                    dd          } ||          }|                    |           d|                                k    sJ d|                                k    sJ d S )NzS<body><div id="d1"><a>1</a><a>2</a><a>3</a><a>4</a></div><div id="d2"></div></body>r   d1r&   d2z<div id="d1"></div>z3<div id="d2"><a>1</a><a>2</a><a>3</a><a>4</a></div>)r   r   r  r   )r   r  r   r   r  r  tagss          r   &test_extend_with_another_tags_contentsz;TestTreeModification.test_extend_with_another_tags_contents  s     eyyYYuY&&YYuY&&x||
		$$		3333D		SSSSSSr    c                     d}|                      |          }|j                            d|j                   d|                                k    sJ d S )Nz<a><b></b><c></c><d></d></a>r   z<a><d></d><b></b><c></c></a>)r   r3   r  r   r   r  s      r   $test_move_tag_to_beginning_of_parentz9TestTreeModification.test_move_tag_to_beginning_of_parent  sM    -yya   ->>>>>>r    c                     |                      d          }|j                            dd           t          |j                  dk    sJ d S )Nz<br/>r   Contentsz<br>Contents</br>)r   brr  r(   r   s     r   &test_insert_works_on_empty_element_tagz;TestTreeModification.test_insert_works_on_empty_element_tag  sK    
 yy!!q*%%%47||2222222r    c                 \   |                      d          }|j                            d           |j                            d           |                                |                     d          k    sJ |j                            |j                   |                                |                     d          k    sJ |j        }t          j        t                    5  |                    |           d d d            n# 1 swxY w Y   |	                                 t          j        t                    5  |                    d           d d d            n# 1 swxY w Y   |                      d          }|j                            |
                    d                     d S )	Nr,   BAZQUUXzQUUX<a>foo</a>BAZ<b>bar</b>QUUX<b>bar</b><a>foo</a>BAZnope<a>r3   )r   r   insert_beforer3   r   r  r   r   r   r  new_tagr   r   r   s      r   test_insert_beforez'TestTreeModification.test_insert_before  s   yy/00U###V$$${{}} 1 1)!
 !
 
 
 
 
 	TV$$${{}} 1 12O P PPPPP F]:&& 	 	OOA	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	
		]:&& 	$ 	$OOF###	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ yyT\\#../////$   #DD	D	=EE#&E#c                    |                      d          }|j                            ddd           |j                            ddd           |                                |                     d          k    sJ |j                            |j        d           |                                |                     d          k    sJ d S )Nr,   r   r  z$QUUX BAZ<a>foo</a>BAZ QUUX<b>bar</b>FOOz'QUUX BAZ<b>bar</b>FOO<a>foo</a>BAZ QUUX)r   r   r  r3   r   r  r   s     r   test_insert_multiple_beforez0TestTreeModification.test_insert_multiple_before  s    yy/00UC000VS%000{{}} 1 12!
 !
 
 
 
 
 	TVU+++{{}} 1 15!
 !
 
 
 
 
 
 
r    c                 \   |                      d          }|j                            d           |j                            d           |                                |                     d          k    sJ |j                            |j                   |                                |                     d          k    sJ |j        }t          j        t                    5  |                    |           d d d            n# 1 swxY w Y   |	                                 t          j        t                    5  |                    d           d d d            n# 1 swxY w Y   |                      d          }|j        
                    |                    d                     d S )	Nr,   r  r  z<a>foo</a>QUUX<b>bar</b>BAZr  r  r  r3   )r   r   insert_afterr3   r   r  r   r   r   r  r  r  r  s      r   test_insert_afterz&TestTreeModification.test_insert_after  s   yy/00E"""F###{{}} 1 1)!
 !
 
 
 
 
 	DF###{{}} 1 12O P PPPPP F]:&& 	 	NN1	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	
		]:&& 	# 	#NN6"""	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# yyT\\#../////r  c                    |                      d          }|j                            ddd           |j                            ddd           |                                |                     d          k    sJ |j                            |j        d           |                                |                     d          k    sJ d S )Nr,   r  r  r  z$<a>foo</a>QUUX BAZ<b>bar</b>BAZ QUUXzFOO z(QUUX BAZ<b>bar</b><a>foo</a>FOO BAZ QUUX)r   r   r  r3   r   r  r   s     r   test_insert_multiple_afterz/TestTreeModification.test_insert_multiple_after  s    yy/00E3///FC///{{}} 1 12!
 !
 
 
 
 
 	DFF+++{{}} 1 16!
 !
 
 
 
 
 
 
r    c                 *   |                      d          }|                    d          }|                    d          }t          j        t
                    5  |                    |           d d d            n# 1 swxY w Y   t          j        t                    5  |                    |           d d d            n# 1 swxY w Y   t          j        t
                    5  |                    |           d d d            d S # 1 swxY w Y   d S NrY  r3   )r   r  
new_stringr   r   r   r  NotImplementedErrorr   r   rr   r   s       r   :test_insert_after_raises_exception_if_after_has_no_meaningzOTestTreeModification.test_insert_after_raises_exception_if_after_has_no_meaning  s   yy}}ll3$$]:&& 	% 	%$$$	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	%].// 	# 	#c"""	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	#]:&& 	" 	"S!!!	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	"6   A;;A?A?CCC%DDDc                 *   |                      d          }|                    d          }|                    d          }t          j        t
                    5  |                    |           d d d            n# 1 swxY w Y   t          j        t                    5  |                    |           d d d            n# 1 swxY w Y   t          j        t
                    5  |                    |           d d d            d S # 1 swxY w Y   d S r  )r   r  r  r   r   r   r  r   r  s       r   Ftest_insert_before_raises_notimplementederror_if_before_has_no_meaningz[TestTreeModification.test_insert_before_raises_notimplementederror_if_before_has_no_meaning  s   yy}}ll3$$]:&& 	& 	&  %%%	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	&].// 	$ 	$s###	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$]:&& 	# 	#c"""	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	#r  c                 V   |                      d          }|                    d          \  }}|                    |           |                                |                     d          k    sJ |j        d k    sJ |j        |j        k    sJ |j        dk    sJ |j        dk    sJ d S )Nz;<p>There's <b>no</b> business like <b>show</b> business</p>r   z0<p>There's  business like <b>no</b> business</p>noz	 business)	r   r0   r  r   r  r   pr  r`  )r   r   r  shows       r   test_replace_withz&TestTreeModification.test_replace_with  s    yyMO O==%%D"{{}} 1 1>!
 !
 
 
 
 
 {d""""yDF""""$&&&&+------r    c                    t          d          }t          j        t                    5  |                    d           d d d            n# 1 swxY w Y   |                     d          j        }t          j        t                    5  |j                            |           d d d            n# 1 swxY w Y   t          j        t                    5  |j                            d|d           d d d            d S # 1 swxY w Y   d S )Nr3   r   z
won't workr  string1string2)r   r   r   r   r  r   r3   r   )r   a_tags     r   test_replace_with_errorsz-TestTreeModification.test_replace_with_errors  s   ]:&& 	- 	-|,,,	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 		*++-]:&& 	( 	(G  '''	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( ]:&& 	> 	>G  E9===	> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	>s5   AAA
B11B58B5C??DDc                    d}|                      |          }|                    d          }d|_        |                    d          }|                    d          }d}|j                            ||||           |                                dk    sJ |j        j        |k    sJ |j        j        |k    sJ |j        j        |k    sJ |j        j        |k    sJ d S )Nr  r   zText In D Tagr  r   zRandom Textz;<a><b></b><d>Text In D Tag</d><e></e>Random Text<f></f></a>)r   r  r   rv   r  r   r   r  )r   r   r   d_tage_tagf_taga_strings          r   test_replace_with_multiplez/TestTreeModification.test_replace_with_multiple-  s    &yyS!!&S!!S!! E5(E:::{{}} ]]]]]v"e++++|(%////!(H4444!.%777777r    c                     d}|                      |          }|j                            |j                   d|                                k    sJ d S )Nr  z<a><c></c></a>)r   r   r  rv   r   r  s      r   test_replace_first_childz-TestTreeModification.test_replace_first_child<  M    &yyDF###4;;==000000r    c                     d}|                      |          }|j                            |j                   d|                                k    sJ d S )Nr  r  )r   rv   r  r   r   r  s      r   test_replace_last_childz,TestTreeModification.test_replace_last_childB  r  r    c                    |                      d          }|j        }|j        }|                    |           |                                |                     d          k    sJ |j        d k    sJ |                    d          j        d k    sJ |j	        d k    sJ |j
        d k    sJ |j        d k    sJ |j        |j        k    sJ |j	        dk    sJ |j        j        |j        k    sJ |j
        d k    sJ |                    d          }|j        }|j        |k    sJ |j
        |k    sJ |j	        |k    sJ |j        |k    sJ d S )NzQ<a>We<b>reserve<c>the</c><d>right</d></b></a><e>to<f>refuse</f><g>service</g></e>z-<a>We<f>refuse</f></a><e>to<g>service</g></e>r  r#   Weto)r   r   r   r  r   r  r   r   r  r?  r`  ry  r3   r  r  )r   r   
remove_tagmove_tagto_textg_tags         r   test_nested_tag_replace_withz1TestTreeModification.test_nested_tag_replace_withH  s   yyce e
 V
6))){{}} 1 1;!
 !
 
 
 
 

  D((((g..;tCCCC*d2222&$....*d2222 $&(((((D0000$1TV;;;;$,,,, ))4)((#u,,,,#u,,,,%0000%000000r    c                     |                      d          }|j                                         |j        d k    sJ |j        j        dk    sJ d S )NzI
            <p>Unneeded <em>formatting</em> is unneeded</p>
            zUnneeded formatting is unneeded)r   emr  r  r)   r   s     r   test_unwrapz TestTreeModification.test_unwrapl  sY    yy    	w$v{???????r    c                    |                      d          }|j                            |                    d                    }|                                dk    sJ |                                |                     d          k    sJ d S )NzI wish I was bold.r   <b>I wish I was bold.</b>)r   r   wrapr  r   r  )r   r   r   s      r   	test_wrapzTestTreeModification.test_wrapt  s}    yy-..  c!2!233||~~!<<<<<{{}} 1 12M N NNNNNNNr    c                     |                      d          }|j        j                            |j                   |                                |                     d          k    sJ d S )Nz<b></b>I wish I was bold.r'  )r   r   r`  r(  r   r  r   s     r   %test_wrap_extracts_tag_from_elsewherez:TestTreeModification.test_wrap_extracts_tag_from_elsewherez  s]    yy455  ((({{}} 1 12M N NNNNNNNr    c                    |                      d          }|j        j                            |j                   dt	          |j        j                  k    sJ |                                |                     d          k    sJ d S )Nz+<b>I like being bold.</b>I wish I was bold.r-   z+<b>I like being bold.I wish I was bold.</b>)r   r   r`  r(  r/   r   r   r  r   s     r   &test_wrap_puts_new_contents_at_the_endz;TestTreeModification.test_wrap_puts_new_contents_at_the_end  s    yyFGG  (((C(((((({{}} 1 19!
 !
 
 
 
 
 
 
r    c                 |   |                      d          }t          |j        j                  dk    sJ |                    d                                          }|                                dk    sJ |                                dk    sJ t          |j        j                  dk    sJ |j        d k    sJ |j        d k    sJ |j	        j	        d k    sJ |                    d	          }|                    d
	          }|j	        |k    sJ |j
        |k    sJ |j        |k    sJ |j        |k    sJ d S )NzR<html><body>Some content. <div id="nav">Nav crap</div> More content.</body></html>rD   navr&   z6<html><body>Some content.  More content.</body></html>z<div id="nav">Nav crap</div>r-   zSome content. r#   z More content.)r   r/   rQ  r   r   r  r   r   r?  r  r`  ry  )r   r   	extracted	content_1	content_2s        r   test_extractz!TestTreeModification.test_extract  sp   yy`b b 49%&&!++++III''//11	{{}} XXXXX!!%CCCCC 49%&&!++++4'''')T1111%2d:::: II%5I66	II%5I66	%2222%2222)Y6666)Y666666r    c                    |                      d          }|j        j        }|j        j        }|                    d          }|                    d          }|j                            |           |j                            |           |                                 |                                 ||j        j        k    sJ ||j        j        k    sJ d S )Nr,   rP   r=   )r   r3   r   r   r  rT   r  )r   r   foo_1bar_1foo_2bar_2s         r   4test_extract_distinguishes_between_identical_stringszITestTreeModification.test_extract_distinguishes_between_identical_strings  s    yy/00&&&&ee
 	%%%%%%%%%%r    c                     |                      d          fd                    d          D              dt          j                  k    sJ d S )Nzv
<html>
<head>
<script>foo</script>
</head>
<body>
 <script>bar</script>
 <a></a>
</body>
<script>baz</script>
</html>c                 B    g | ]}j                                         S r9   )scriptr  )r  r   r   s     r   r  zKTestTreeModification.test_extract_multiples_of_same_tag.<locals>.<listcomp>  s'    @@@1				@@@r    r<  z<body>

<a></a>
</body>)r   r0   r(   rQ  r   s    @r   "test_extract_multiples_of_same_tagz7TestTreeModification.test_extract_multiples_of_same_tag  sc    yy 
 
 
 	A@@@h(?(?@@@@+s49~~======r    c                     |                      d          }|                    d                                           d |                    d          k    sJ d S )Nz<html>
<body>hi</body>
</html>rQ  )r   r   r  r   s     r   Btest_extract_works_when_element_is_surrounded_by_identical_stringszWTestTreeModification.test_extract_works_when_element_is_surrounded_by_identical_strings  s]    yy  			&!!###tyy((((((((r    c                 @   |                      d          }|j        }|j                                         t	          |j        j                  dk    sJ t          |d          sJ |j        }|                    d           dt	          |j                  k    sJ dS )zTag.clear()z4<p><a>String <em>Italicized</em></a> and another</p>r   r   T)	decomposeN)r   r3   r  clearr/   r   rY   r$  )r   r   r3   r$  s       r   
test_clearzTestTreeModification.test_clear  s    yyOPPF46?##q((((q*%%%%% T	$C$$$$$$$$r    c                    |                      d          }|                    d          \  }}|j        }|j        j        }||||fD ]}d|j        k    sJ |                                 |||fD ]}d|j        k    sJ d|j        k    sJ d S )Nz;<p><a>String <em>Italicized</em></a></p><p>Another para</p>r  FT)r   r0   r3   r$  r   
decomposedrA  )r   r   r  r  r3   r)   r   s          r   test_decomposez#TestTreeModification.test_decompose  s    yyVWWs##BDu|b!T" 	) 	)AAL((((( 	a 	( 	(A1<'''''%%%%%%r    c                     |                      d          }d|j        _        |j        j        dgk    sJ d|j        _        |j        j        dgk    sJ dS )zTag.string = 'string'z<a></a> <b><c></c></b>rP   r=   N)r   r3   r   r   r   r   s     r   test_string_setz$TestTreeModification.test_string_set  sZ    yy122v5'))))v5'))))))r    c                     |                      d          }|j        j        |j        _        |j                                        dk    sJ d S )Nz<a><b>foo</b><c>bar</c>s   <a><b>bar</b><c>bar</c></a>)r   rv   r   r   r3   r   r   s     r   /test_string_set_does_not_affect_original_stringzDTestTreeModification.test_string_set_does_not_affect_original_string  sC    yy233v}}"@@@@@@@r    c                     |                      d          }t          d          }||j        _        t	          |j        j        t                    sJ d S )NrS   rP   )r   r   r3   r   r  )r   r   cdatas      r   )test_set_string_preserves_class_of_stringz>TestTreeModification.test_set_string_preserves_class_of_string  sG    yy##e$&-///////r    N)3r5   r6   r7   r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   markparametrizer  r  r  r  r  r  r  r  r  r
  r  r  r  r  r"  r%  r)  r+  r-  r3  r9  r=  r?  rC  rF  rH  rJ  rM  r9   r    r   r  r    s       G G G] ] ]
 
 
$    # # #8 8 8( ( (% % %. . .(( ( (- - -/ / /*8 8 8- - -<% % %T T T [__&>&>? T T T? ? ?3 3 30 0 06
 
 
0 0 04

 

 

	" 	" 	"	# 	# 	#. . .> > >8 8 81 1 11 1 1"1 "1 "1H@ @ @O O OO O O

 
 
7 7 70& & &"> > > ) ) )% % %& & & * * *A A A
0 0 0 0 0r    r  c                   `    e Zd Zej                            dg d          d             Zd ZdS )TestDeprecatedArgumentsmethod_name)r   r0   r  r  r-  r*  rK  rH  rj  rm  r  r  c                 4   |                      d          }t          |j        |          }t          j        d          5 } |d           |\  }|j        t          k    sJ t          |j                  }|dk    sJ 	 d d d            d S # 1 swxY w Y   d S )Nz<a>some</a><b>markup</b>Trecordr[  r)   zOThe 'text' argument to find()-type methods is deprecated. Use 'string' instead.)	r   getattrr   warningscatch_warningsfilename__file__r(   message)r   rR  r   methodwwarningmsgs          r   test_find_type_method_stringz4TestDeprecatedArguments.test_find_type_method_string  s    yy344--$D111 	lQF!!!!IW#x////go&&Ckkkkkk	l 	l 	l 	l 	l 	l 	l 	l 	l 	l 	l 	l 	l 	l 	l 	l 	l 	ls    ?BBBc                    t          j        d          5 }t          d          }|j        dk    sJ |\  }t	          |j                  }|j        t          k    sJ |dk    sJ 	 d d d            d S # 1 swxY w Y   d S )NTrT  r)   rV  zXThe 'text' argument to the SoupStrainer constructor is deprecated. Use 'string' instead.)rX  rY  r   r)   r(   r\  rZ  r[  )r   r^  r   r_  r`  s        r   $test_soupstrainer_constructor_stringz<TestDeprecatedArguments.test_soupstrainer_constructor_string  s    $D111 	uQ#000H=F****IWgo&&C#x////tttttt	u 	u 	u 	u 	u 	u 	u 	u 	u 	u 	u 	u 	u 	u 	u 	u 	u 	us   AA44A8;A8N)r5   r6   r7   r   rN  rO  ra  rc  r9   r    r   rQ  rQ    sk        [ 
 
 
 l l lu u u u ur    rQ  ))r8   pdbr   r   r@   rX  bs4r   bs4.builderr   r   bs4.elementr   r   r	   r
   r   r   r   r   r   r   r   rY  r   r   r;   r]   re   r   r   r   r   r  r  r;  rW  r]  rv  r  rQ  r9   r    r   <module>rh     s           				                                            , , , , ,x , , ,>6) 6) 6) 6) 6)( 6) 6) 6)rD D D D D D D DK  K  K  K  K  K  K  K \q7 q7 q7 q7 q7X q7 q7 q7h). ). ). ). ). ). ). ).X       &/6 /6 /6 /6 /68 /6 /6 /6dw w w w wH w w w&# &# &# &# &# &# &# &#P.# .# .# .# .#] .# .# .#b& & & & &( & & &**B *B *B *B *Bk *B *B *BZ*F *F *F *F *F+ *F *F *FZq0 q0 q0 q0 q08 q0 q0 q0hu u u u uh u u u u ur    