File: //lib/python3/dist-packages/bs4/tests/__pycache__/test_tree.cpython-38.pyc
U
t�^�P � @ sH d Z ddlmZ ddlZddlZddlZddlZddlmZ ddl m
Z
m Z ddl m
Z
mZmZmZmZmZmZmZmZ ddlmZmZ e
�d�dk Ze
�d �dk ZG d
d � d e�Z G d d
� d
e �Z G dd� de �Z G dd� de �Z G dd� de �Z G dd� de �Z!G dd� de �Z"G dd� de �Z#G dd� de �Z$G d d � d e �Z%G d d � d e%�Z&G d d!� d!e%�Z'G d"d#� d#e �Z(G d$d%� d%e(�Z)G d&d'� d'e(�Z*G d(d)� d)e�Z+G d*d+� d+e�Z,G d,d-� d-e�Z-G d.d/� d/e�Z.G d0d1� d1e�Z/G d2d3� d3e�Z0G d4d5� d5e�Z1G d6d7� d7e�Z2G d8d9� d9e�Z3G d:d;� d;e�Z4G d<d=� d=e �Z5dS )>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) �PY3K�CData�Comment� Declaration�Doctype� Formatter�NavigableString� SoupStrainer�Tag)�SoupTest�skipIfZxmlZlxmlc @ s e Zd Zdd� Zdd� ZdS )�TreeTestc C s | � dd� |D �|� dS )z�Make sure that the given tags have the correct text.
This is used in tests that define a bunch of tags, each
containing a single string, and then select certain strings by
some mechanism.
c S s g | ]
}|j �qS � ��string��.0�tagr r �5/usr/lib/python3/dist-packages/bs4/tests/test_tree.py�
<listcomp>2 s z*TreeTest.assertSelects.<locals>.<listcomp>N�� assertEqual��selfZtagsZ should_matchr r r �
assertSelects+ s zTreeTest.assertSelectsc C s | � dd� |D �|� dS )z�Make sure that the given tags have the correct IDs.
This is used in tests that define a bunch of tags, each
containing a single string, and then select certain strings by
some mechanism.
c S s g | ] }|d �qS ��idr r r r r r ; s z-TreeTest.assertSelectsIDs.<locals>.<listcomp>Nr r r r r �assertSelectsIDs4 s zTreeTest.assertSelectsIDsN)�__name__�
__module__� __qualname__r r! r r r r r ) s r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�TestFindz�Basic tests of the find() method.
find() just calls find_all() with limit=1, so it's not tested all
that thouroughly here.
c C s" | � d�}| �|�d�jd� d S )Nz <a>1</a><b>2</b><a>3</a><b>4</b>�b�2)�soupr �findr �r r( r r r �
test_find_tagE s
zTestFind.test_find_tagc C s" | � d�}| �|jdd�d� d S )Nu <h1>Räksmörgås</h1>�
Räksmörgåsr )r( r r) r* r r r �test_unicode_text_findI s
z TestFind.test_unicode_text_findc C s, | � d�}t|� | �d|jdd�j� d S )Nu& <h1 id="Räksmörgås">here it is</h1>z
here it isr, r )r( �strr r) �textr* r r r �test_unicode_attribute_findM s
z$TestFind.test_unicode_attribute_findc C s" | � d�}| �dt|�� �� dS )z)Test an optimization that finds all tags.�<a>foo</a><b>bar</b>� N�r( r �len�find_allr* r r r �test_find_everythingS s
z TestFind.test_find_everythingc C s$ | � d�}| �dt|�d��� dS )z;Test an optimization that finds all tags with a given name.z <a>foo</a><b>bar</b><a>baz</a>r2 �aNr3 r* r r r � test_find_everything_with_nameX s
z'TestFind.test_find_everything_with_nameN) r" r# r$ �__doc__r+ r- r0 r6 r8 r r r r r% > s r% c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� TestFindAllz%Basic tests of the find_all() method.c C s� | � d�}| �|jdd�dg� | �|jdd�dg� | �|jddgd�ddg� | �|jt�d�d�dddg� | �|jdd�dddg� d S )
z'You can search the tree for text nodes.u <html>Foo<b>bar</b>»</html>�barr �r/ �Fooz.*� »TN)r( r r5 �re�compiler* r r r �test_find_all_text_nodes` s
���z$TestFindAll.test_find_all_text_nodesc C s� | � d�}| �|jddd�dddg� | �|jddd�dg� | �|jdd d�dddd
d g� | �|jdd d�dddd
d g� 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>r7 � ��limit�1r' �3� �
�4�5r N�r( r r5 r* r r r �test_find_all_limitp s
� �z TestFindAll.test_find_all_limitc C s: | � d�}| �|ddd�dg� | �|jdd�dg� d S ) Nz!<a>1</a><b>2<a id='foo'>3</a></b>r7 rG rC rE �foor rF )r( r r& r* r r r �%test_calling_a_tag_is_calling_findall| s
z1TestFindAll.test_calling_a_tag_is_calling_findallc C s. | � d�}g }|�|� | �g |�|�� d S �N�<a></a>)r( �appendr r5 )r r( �lr r r �Ttest_find_all_with_self_referential_data_structure_does_not_cause_infinite_recursion� s
z`TestFindAll.test_find_all_with_self_referential_data_structure_does_not_cause_infinite_recursionc C s^ | � d�}|�d�}| �t|d�� |�d�}| �t|d�� |jdd�}| �t|d�� dS )z%All find_all calls return a ResultSetrP r7 �sourceTrM r< N)r( r5 �
assertTrue�hasattr)r r( �resultr r r �test_find_all_resultset� s
z#TestFindAll.test_find_all_resultsetN) r" r# r$ r9 rA rL rN rS rX r r r r r: ] s
r: c @ s e Zd Zdd� ZdS )�TestFindAllBasicNamespacesc C s< | � d�}| �d|�d�j� | �d|jddid�j� d S )Nz0<mathml:msqrt>4</mathml:msqrt><a svg:fill="red">rI z mathml:msqrtr7 zsvg:fillZred��attrs)r( r r) r �namer* r r r � test_find_by_namespaced_name� s
z7TestFindAllBasicNamespaces.test_find_by_namespaced_nameN)r" r# r$ r] r r r r rY � s rY c sp e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Z� ZS )�TestFindAllByNamez&Test ways of finding tags by tag name.c s t t| ��� | �d�| _d S )Nz�<a>First tag.</a>
<b>Second tag.</b>
<c>Third <a>Nested tag.</a> tag.</c>��superr �setUpr( �tree�r �� __class__r r ra � s zTestFindAllByName.setUpc C s | � | j�d�ddg� d S �Nr7 �
First tag.� Nested tag.�r rb r5 rc r r r �test_find_all_by_tag_name� s
�z+TestFindAllByName.test_find_all_by_tag_namec C s\ | � | jjddd�dg� | � | jjddd�ddg� | � | jjdt�d�d�ddg� d S )Nr7 rg r< Trh r �r rb r5 r? r@ rc r r r � test_find_all_by_name_and_text� s � ��z0TestFindAllByName.test_find_all_by_name_and_textc C s | � | jj�d�dg� d S )Nr7 rh )r rb �cr5 rc r r r �!test_find_all_on_non_root_element� s z3TestFindAllByName.test_find_all_on_non_root_elementc C s | � | �d�ddg� d S rf )r rb rc r r r �%test_calling_element_invokes_find_all� s z7TestFindAllByName.test_calling_element_invokes_find_allc C s | � | j�td��ddg� d S rf )r rb r5 r
rc r r r � test_find_all_by_tag_strainer� s �z/TestFindAllByName.test_find_all_by_tag_strainerc C s" | � | j�ddg�dddg� d S )Nr7 r& rg � Second tag.rh ri rc r r r �test_find_all_by_tag_names� s �z,TestFindAllByName.test_find_all_by_tag_namesc C s$ | � | j�ddd��dddg� d S )NT)r7 r& rg rq rh ri rc r r r �test_find_all_by_tag_dict� s �z+TestFindAllByName.test_find_all_by_tag_dictc C s$ | � | j�t�d��dddg� d S )Nz^[ab]$rg rq rh rk rc r r r �test_find_all_by_tag_re� s �z)TestFindAllByName.test_find_all_by_tag_rec C s, dd� }| � d�}| �|�|�ddg� d S )Nc S s | j | �d�kS �Nr )r\ �get)r r r r �id_matches_name� s zRTestFindAllByName.test_find_all_with_tags_matching_method.<locals>.id_matches_namez�<a id="a">Match 1.</a>
<a id="1">Does not match.</a>
<b id="b">Match 2.</a>zMatch 1.zMatch 2.rK )r rw rb r r r �'test_find_all_with_tags_matching_method� s
�z9TestFindAllByName.test_find_all_with_tags_matching_methodc C sx | � d�}|�dd�}|�dt�d��}|�dddg�\}}| �d|j� | �d|j� | �d|j� | �d|j� d S )NzH<div class='a b'>1</div><div class='a c'>2</div><div class='a d'>3</div>�divza dza brF rE )r( r) r? r@ r5 r r )r r( Zr1Zr2Zr3Zr4r r r �%test_find_with_multi_valued_attribute� s � z7TestFindAllByName.test_find_with_multi_valued_attribute)r" r# r$ r9 ra rj rl rn ro rp rr rs rt rx rz �
__classcell__r r rd r r^ � s
r^ c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!S )"�TestFindAllByAttributec C s& | � d�}| �|jdd�ddg� 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.rK �r rb r r r � test_find_all_by_attribute_name� s
�z6TestFindAllByAttribute.test_find_all_by_attribute_namec C st d� d�}d� d�}| �|�}| �|jg|j|d�� | �|jg|j|�d�d�� | �|jg|j|dgd�� d S )Nu םולש�utf8u <a title="םולש"></a>)�titlezsomething else)�encoder( r r7 r5 �decode)r Zpeace�datar( r r r �%test_find_all_by_utf8_attribute_value� s
z<TestFindAllByAttribute.test_find_all_by_utf8_attribute_valuec C sX | � d�}| �|jdd�dg� | �|jddid�dg� | �|jdd id�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>
Zname1�r\ zA tag called 'name1'.r\ rZ z Name match.�classZclass2z Class match.rK r~ r r r � test_find_all_by_attribute_dict s
���z6TestFindAllByAttribute.test_find_all_by_attribute_dictc C s� | � d�}| �|jddd�dg� | �|jddd�dg� | �|jdd d�dg� | �|�dd�dg� | �|jdd
�ddg� | �|�dd�dg� | �|�dd �dg� 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>
r7 rE �Zclass_zClass 1.rm rF zClass 3 and 4.rI rZ rK r~ r r r �test_find_all_by_class s
z-TestFindAllByAttribute.test_find_all_by_classc C st | � d�}|jdt�d�d�}| �|dg� |jdt�d�d�}| �|dg� |jdt�d�d�}| �|dg� d S )Nz#<gar class='foo bar'>Found it</gar>Zgar�or� �Found itr7 zo b)r( r5 r? r@ r )r rb �fr r r �0test_find_by_class_when_multiple_classes_present- s
zGTestFindAllByAttribute.test_find_by_class_when_multiple_classes_presentc C sd | � d�}| �|�dt�d��dg� dd� }| �|�d|�g � dd� }| �|�d|�dg� d S ) Nz<a class='bar'>Found it</a>r7 Zbar� c S s t | �dkS �NrB �r4 ��valuer r r �big_attribute_value@ s znTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_class.<locals>.big_attribute_valuec S s t | �dkS r� r� r� r r r �small_attribute_valueE s zpTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_class.<locals>.small_attribute_value�r( r r5 r? r@ )r r( r� r� r r r �:test_find_all_with_non_dictionary_for_attrs_finds_by_class; s
�zQTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_classc C s� | � d�}|�d�\}}| �||g|�dd�� | �|g|�dd�� | �|g|jddd�� | �|g|�dd�� | �g |�dd�� d S )Nz*<a class="foo bar"></a><a class="foo"></a>r7 rM r; zfoo barr� zbar foo)r( r5 r )r r( r7 Za2r r r �:test_find_all_with_string_for_attrs_finds_multiple_classesK s
zQTestFindAllByAttribute.test_find_all_with_string_for_attrs_finds_multiple_classesc C s0 | � d�}tddid�}| �|�|�dg� d S )Nzi
<a id="first">Match.</a>
<a id="second">Non-match.</a>r r} rZ zMatch.)r( r
r r5 )r rb �strainerr r r �'test_find_all_by_attribute_soupstrainerW s
z>TestFindAllByAttribute.test_find_all_by_attribute_soupstrainerc C s& | � d�}| �|jdd d�dg� d S )N��<a id="1">ID present.</a>
<a>No ID present.</a>
<a id="">ID is empty.</a>r7 r zNo ID present.rK r~ r r r �$test_find_all_with_missing_attribute_ s
z;TestFindAllByAttribute.test_find_all_with_missing_attributec C s& | � d�}| �|jdd�ddg� d S )Nr� Tr z ID present.z ID is empty.rK r~ r r r �$test_find_all_with_defined_attributeg s
�z;TestFindAllByAttribute.test_find_all_with_defined_attributec C s> | � d�}ddg}| �|jdd�|� | �|jdd�|� d S )Nz[<a id=1>Unquoted attribute.</a>
<a id="1">Quoted attribute.</a>zUnquoted attribute.zQuoted attribute.rG r rE rK )r rb Zexpectedr r r �$test_find_all_with_numeric_attributep s
z;TestFindAllByAttribute.test_find_all_with_numeric_attributec C s, | � d�}| �|jdddgd�ddg� d S )Nz�<a id="1">1</a>
<a id="2">2</a>
<a id="3">3</a>
<a>No ID.</a>rE rF rI r rK r~ r r r �(test_find_all_with_list_attribute_valuesy s
�z?TestFindAllByAttribute.test_find_all_with_list_attribute_valuesc C s, | � d�}| �|jt�d�d�ddg� 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 zOne a.zTwo as.r� r~ r r r �5test_find_all_with_regular_expression_attribute_value� s
�zLTestFindAllByAttribute.test_find_all_with_regular_expression_attribute_valuec C sX | � d�}|j}| �|g|jddd�� | �g |jddd�� | �g |jddd�� d S )Nz <b>foo</b><b>bar</b><a>foo</a>r7 rM r< r; �r( r7 r r5 �r r( r7 r r r �'test_find_by_name_and_containing_string� s
z>TestFindAllByAttribute.test_find_by_name_and_containing_stringc C s* | � d�}| �|�d�|jddd�� d S )Nz"<a>foo</a><a><b><c>foo</c></b></a>r7 rM r< )r( r r5 r* r r r �=test_find_by_name_and_containing_string_when_string_is_buried� s
zTTestFindAllByAttribute.test_find_by_name_and_containing_string_when_string_is_buriedc C sB | � d�}|j}| �|g|jddd�� | �g |jddd�� d S )Nz"<b id="1">foo</b><a id="2">foo</a>r2 rM )r r/ rG r; r� r� r r r �,test_find_by_attribute_and_containing_string� s
zCTestFindAllByAttribute.test_find_by_attribute_and_containing_stringN)r" r# r$ r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r| � s
r| c @ s e Zd ZdZdd� ZdS )�
TestSmoothzTest Tag.smooth.c C s | � d�}|j}|�d� |�d� |�td�� |�td�� |�d� | �� }t||d�}|�d� |�d � |�|� | �d |jj� | �d
t |j
�� |� � | �d t |j
�� | �d |j
d
� | �d|jj� | �d|j
d � | �d|j
d � d S )Nz <div>a</div>r& rm z Comment 1z Comment 2�d�spanrE r' � � �abcr Z12rG r2 ) r( ry rQ r �default_builderr r r� r r4 �contentsZsmooth)r r( ry �builderr� r r r � test_smooth� s(
zTestSmooth.test_smoothN)r" r# r$ r9 r� r r r r r� � s r� c @ s e Zd ZdZdd� ZdS )� TestIndexzTest Tag.indexc C sJ | � d�}|j}t|j�D ]\}}| �||�|�� q| �t|jd� 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>rG )r( ry � enumerater� r �index� assertRaises�
ValueError)r rb ry �i�elementr r r �
test_index� s
zTestIndex.test_indexN)r" r# r$ r9 r� r r r r r� � s r� c s` e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z � Z
S )�TestParentOperationsz;Test navigation and searching through an element's parents.c s( t t| ��� | �d�| _| jj| _d S )Na1 <ul id="empty"></ul>
<ul id="top">
<ul id="middle">
<ul id="bottom">
<b>Start here</b>
</ul>
</ul>)r` r� ra r( rb r& �startrc rd r r ra � s zTestParentOperations.setUpc C sF | � | jjd d� | � | jjjd d� | � | jjjjd d� d S )Nr �bottom�middle�top)r r� �parentrc r r r � test_parent� s z TestParentOperations.test_parentc C s | j jd }| �|j| j � d S )Nr )rb r� r r� )r Ztop_tagr r r �%test_parent_of_top_tag_is_soup_object� s z:TestParentOperations.test_parent_of_top_tag_is_soup_objectc C s | � d | jj� d S �N)r rb r� rc r r r � test_soup_object_has_no_parent� s z3TestParentOperations.test_soup_object_has_no_parentc C s8 | � | j�d�dddg� | � | jjddd�dg� d S )N�ulr� r� r� r )r! r� Z find_parentsrc r r r �test_find_parents� s
� �z&TestParentOperations.test_find_parentsc C s8 | � | j�d�d d� | � | jjddd�d d� d S )Nr� r r� r� r )r r� � find_parentrc r r r �test_find_parent s z%TestParentOperations.test_find_parentc C s" | j jdd�}| �|jjd� d S )N�
Start herer< r& )rb r) r r� r\ �r r/ r r r �test_parent_of_text_element s z0TestParentOperations.test_parent_of_text_elementc C s( | j jdd�}| �|�d�d d� d S )Nr� r< r� r r� )rb r) r r� r� r r r � test_text_element_find_parent s z2TestParentOperations.test_text_element_find_parentc C s( dd� | j jD �}| �|dddg� d S )Nc S s& g | ] }|d k rd|j kr|d �qS ru rZ )r r� r r r r s
�z>TestParentOperations.test_parent_generator.<locals>.<listcomp>r� r� r� )r� �parentsr )r r� r r r �test_parent_generator s z*TestParentOperations.test_parent_generator)r" r# r$ r9 ra r� r� r� r� r� r� r� r� r{ r r rd r r� � s r� c s e Zd Z� fdd�Z� ZS )�
ProximityTestc s t t| ��� | �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>r_ rc rd r r ra s �zProximityTest.setUp�r" r# r$ ra r{ r r rd r r� s r� c sT e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
� Z S )�TestNextOperationsc s t t| ��� | jj| _d S r� )r` r� ra rb r& r� rc rd r r ra s zTestNextOperations.setUpc C s* | � | jjd� | � | jjjd d� d S )N�Oner r' )r r� � next_elementrc r r r � test_next# s z TestNextOperations.test_nextc C s | j jdd�}| �|jd � d S �N�Threer< )rb r) r r� )r Zlastr r r � test_next_of_last_item_is_none' s z1TestNextOperations.test_next_of_last_item_is_nonec C s | � | jjd � d S r� )r rb r� rc r r r �test_next_of_root_is_none+ s z,TestNextOperations.test_next_of_root_is_nonec C sB | � | j�d�ddg� | jjdd� | � | jjdd�dg� d S )Nr& �Twor� rB r )r r� �
find_all_nextrc r r r �test_find_all_next/ s z%TestNextOperations.test_find_all_nextc C s2 | � | j�d�d d� | � | jjdd�d� d S )Nr& r r' r� r< )r r� � find_nextrc r r r �test_find_next4 s z!TestNextOperations.test_find_nextc C s< | j jdd�}| �|�d�jd� | �|�d�ddg� d S )Nr� r< r& r� r� )rb r) r r� r r r� r� r r r � test_find_next_for_text_element8 s z2TestNextOperations.test_find_next_for_text_elementc C sF | j jdd�}dd� |jD �}|\}}| �|d d� | �|d� d S )Nr� r< c S s g | ]}|�qS r r �r Znoder r r r ? s z:TestNextOperations.test_next_generator.<locals>.<listcomp>r rF r� )rb r) Z
next_elementsr )r r� Z
successorsr r� r r r �test_next_generator= s
z&TestNextOperations.test_next_generator) r" r# r$ ra r� r� r� r� r� r� r� r{ r r rd r r� s r� c sT e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
� Z S )�TestPreviousOperationsc s" t t| ��� | jjdd�| _d S r� )r` r� ra rb r) �endrc rd r r ra G s z TestPreviousOperations.setUpc C s* | � | jjd d� | � | jjjd� d S )Nr rF r� )r r� �previous_elementrc r r r �
test_previousK s z$TestPreviousOperations.test_previousc C s | j �d�}| �|jd � d S )N�html)rb r) r r� )r r} r r r �#test_previous_of_first_item_is_noneO s z:TestPreviousOperations.test_previous_of_first_item_is_nonec C s d S r� r rc r r r � test_previous_of_root_is_noneS s z4TestPreviousOperations.test_previous_of_root_is_nonec C s6 | � | j�d�dddg� | � | jjdd�dg� d S )Nr& r� r� r� rG r )r r� �find_all_previousrc r r r �test_find_all_previousY s
�z-TestPreviousOperations.test_find_all_previousc C s2 | � | j�d�d d� | � | jjdd�d� d S )Nr& r rF r� r< )r r� �
find_previousrc r r r �test_find_previousa s z)TestPreviousOperations.test_find_previousc C s> | j jdd�}| �|�d�jd� | �|�d�dddg� d S )Nr� r< r& r� r� )rb r) r r� r r r� r� r r r �#test_find_previous_for_text_elemente s �z:TestPreviousOperations.test_find_previous_for_text_elementc C sh | j jdd�}dd� |jD �}|\}}}}| �|d d� | �|jd� | �|jd� | �|jd � d S )
Nr� r< c S s g | ]}|�qS r r r� r r r r m s zBTestPreviousOperations.test_previous_generator.<locals>.<listcomp>r rE �body�headr� )rb r) Zprevious_elementsr r\ )r r� Z predecessorsr& r� r� r� r r r �test_previous_generatork s z.TestPreviousOperations.test_previous_generator) r" r# r$ ra r� r� r� r� r� r� r� r{ r r rd r r� E s r� c s e Zd Z� fdd�Z� ZS )� SiblingTestc s4 t t| ��� d}t�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` r� ra r? r@ �subr( rb �r �markuprd r r ra z s zSiblingTest.setUpr� r r rd r r� x s r� c sL e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z � Z
S )�TestNextSiblingc s" t t| ��� | jjdd�| _d S )NrE r )r` r� ra rb r) r� rc rd r r ra � s zTestNextSibling.setUpc C s | � | jjd � d S r� )r rb � next_siblingrc r r r �!test_next_sibling_of_root_is_none� s z1TestNextSibling.test_next_sibling_of_root_is_nonec C sB | � | jjd d� | � | jjjd d� | � | jjd d� d S )Nr r' rF �1.1)r r� r� r� rc r r r �test_next_sibling� s z!TestNextSibling.test_next_siblingc C sN | � | jjjd � | jjdd�}| � |jd � | jjdd�}| � |jd � d S )Nr� r rI )r rb r� r� r) )r � nested_spanZ last_spanr r r � test_next_sibling_may_not_exist� s
z/TestNextSibling.test_next_sibling_may_not_existc C s | � | j�d�d d� d S )Nr� r r' )r r� �find_next_siblingrc r r r �test_find_next_sibling� s z&TestNextSibling.test_find_next_siblingc C s6 | � | j�d�dddg� | � | jjdd�dg� d S )Nr� r' rF rI r )r! r� �find_next_siblingsrc r r r �test_next_siblings� s �z"TestNextSibling.test_next_siblingsc C sv | � d�}|jdd�}| �|jjd� | �|jjd� | �|�d�dg� | �|jdd�d� | �|jdd�d � d S )N�Foo<b>bar</b>bazr= r< r& �bazr; �nonesuch)r( r) r r� r\ r r� r� �r r( r� r r r �"test_next_sibling_for_text_element� s
z2TestNextSibling.test_next_sibling_for_text_element) r" r# r$ ra r� r� r� r� r� r r{ r r rd r r� � s r� c sL e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z � Z
S )�TestPreviousSiblingc s" t t| ��� | jjdd�| _d S )NrI r )r` r ra rb r) r� rc rd r r ra � s zTestPreviousSibling.setUpc C s | � | jjd � d S r� )r rb �previous_siblingrc r r r �%test_previous_sibling_of_root_is_none� s z9TestPreviousSibling.test_previous_sibling_of_root_is_nonec C sB | � | jjd d� | � | jjjd d� | � | jjd d� d S )Nr rF r' z3.1)r r� r r� rc r r r �test_previous_sibling� s z)TestPreviousSibling.test_previous_siblingc C sN | � | jjjd � | jjdd�}| � |jd � | jjdd�}| � |jd � d S )Nr� r rE )r rb r� r r) )r r� Z
first_spanr r r �#test_previous_sibling_may_not_exist� s
z7TestPreviousSibling.test_previous_sibling_may_not_existc C s | � | j�d�d d� d S )Nr� r rF )r r� �find_previous_siblingrc r r r �test_find_previous_sibling� s z.TestPreviousSibling.test_find_previous_siblingc C s6 | � | j�d�dddg� | � | jjdd�dg� d S )Nr� rF r' rE r )r! r� �find_previous_siblingsrc r r r �test_previous_siblings� s �z*TestPreviousSibling.test_previous_siblingsc C sv | � d�}|jdd�}| �|jjd� | �|jjd� | �|�d�dg� | �|jdd�d� | �|jdd�d � d S )Nr� r� r< r& r= r; r� )r( r) r r r\ r r r r� r r r �&test_previous_sibling_for_text_element� s
z:TestPreviousSibling.test_previous_sibling_for_text_element) r" r# r$ ra r r r r r r
r{ r r rd r r � s r c @ s e Zd Zdd� ZdS )�TestTagc C st | � d��d�}d |_| �d|�d�� dg|_| �d|�d�� | �d|�d �� ddg|_| �d|�d�� d S )Nr� Za_tagTr Zsome_other_tagrG F)r( �new_tagZ_preserve_whitespace_tagsr Z_should_pretty_print�preserve_whitespace_tags)r r r r r �test__should_pretty_print� s
z!TestTag.test__should_pretty_printN)r" r# r$ r r r r r r � s r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �TestTagCreationz$Test the ability to create new tags.c C sd | � d�}|jddddid�}| �t|t�� | �d|j� | �tddd�|j� | �d |j � d S )Nr� rM r� r\ za name)r; r[ )r; r\ )
r( r rU �
isinstancer r r\ �dictr[ r� )r r( r r r r � test_new_tag s
z TestTagCreation.test_new_tagc C s� t rBtdd�}|�d�}|�d�}| �d|�� � | �d|�� � tdd�}|�d�}|�d�}| �d|�� � | �d|�� � d S ) Nr� zlxml-xml�br�ps <br/>s <p/>� html.parsers <p></p>)�XML_BUILDER_PRESENTr r r r� )r Zxml_soupZxml_brZxml_pZ html_soupZhtml_brZhtml_pr r r �1test_tag_inherits_self_closing_rules_from_builder
s
zATestTagCreation.test_tag_inherits_self_closing_rules_from_builderc C s4 | � d�}|�d�}| �d|� | �t|t�� d S �Nr� rM )r( �
new_stringr rU r r �r r( �sr r r �'test_new_string_creates_navigablestring s
z7TestTagCreation.test_new_string_creates_navigablestringc C s6 | � d�}|�dt�}| �d|� | �t|t�� d S r )r( r r r rU r r r r r �3test_new_string_can_create_navigablestring_subclass$ s
zCTestTagCreation.test_new_string_can_create_navigablestring_subclassN)r" r# r$ r9 r r r r r r r r r s
r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIdJ� Z'dKdL� Z(dMdN� Z)dOdP� Z*dQdR� Z+dSS )T�TestTreeModificationc C sl | � d�}d|jd< | �|�� | �d�� |jd= | �|�� | �d�� d|jd< | �|�� | �d�� d S ) Nz<a id="1"></a>r2 r z<a id="2"></a>rP rM Zid2z<a id2="foo"></a>)r( r7 r r� � document_forr* r r r �test_attribute_modification, s
z0TestTreeModification.test_attribute_modificationc C sl t �d�� }| jd|d�}t||d�}t||d�}d|d< |j�d|� |j�d |� | �|j�� d
� d S ) Nr� z
<body></body>�r� r7 �olzhttp://foo.com/Zhrefr rG s4 <body><a href="http://foo.com/"></a><ol></ol></body>)r �lookupr( r r� �insertr r� )r r� r( r7 r" r r r �test_new_tag_creation5 s �z*TestTreeModification.test_new_tag_creationc C s\ d}| � |�}|jdd�}|j}|jdd��|j� | �|j|� | �|�� | �d�� 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& rQ r r� r� r )r �docr( Z second_paraZboldr r r �!test_append_to_contents_moves_tagA s
��z6TestTreeModification.test_append_to_contents_moves_tagc C s0 d}| � |�}|j}|�|j�}| �||� d S )Nz<a></a><b><c></c></b>)r( r7 � replace_withrm r �r r/ r( r7 Znew_ar r r �1test_replace_with_returns_thing_that_was_replacedS s
zFTestTreeModification.test_replace_with_returns_thing_that_was_replacedc C s, d}| � |�}|j}|�� }| �||� d S )N�<a><b></b><c></c></a>)r( r7 �unwrapr r) r r r �+test_unwrap_returns_thing_that_was_replacedZ s
z@TestTreeModification.test_unwrap_returns_thing_that_was_replacedc C sJ | � d�}|j}|�� | �d |j� | �t|j� | �t|j|j � d S )Nz<a><b>Foo</b></a><c>Bar</c>)
r( r7 �extractr r� r� r� r, r( rm r� r r r �Itest_replace_with_and_unwrap_give_useful_exception_when_tag_has_no_parenta s
z^TestTreeModification.test_replace_with_and_unwrap_give_useful_exception_when_tag_has_no_parentc C s: d}| � |�}|j}|j�|� | �|�� | �|�� d S )Nz-<a><b></b><c>Foo<d></d></c></a><a><e></e></a>)r( rm r( r r� r )r r/ r( rm r r r � test_replace_tag_with_itselfi s
z1TestTreeModification.test_replace_tag_with_itselfc C s& d}| � |�}| �t|jj|j� d S �N�<a><b></b></a>)r( r� r� r& r( r7 �r r/ r( r r r �1test_replace_tag_with_its_parent_raises_exceptionp s
zFTestTreeModification.test_replace_tag_with_its_parent_raises_exceptionc C s( d}| � |�}| �t|jjd|j� d S )Nr2 r )r( r� r� r7 r$ r3 r r r �,test_insert_tag_into_itself_raises_exceptionu s
zATestTreeModification.test_insert_tag_into_itself_raises_exceptionc C s� | � d�}d}| � |�}|�d|� |jD ]}t|t�r*t�q*t|j�\}}}}| �d|j � | �d|j � | �d|j � | �d|j � dS ) z�Inserting 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>rG zAnd now, a word:�p2�p3zAnd we're back.N)
r( r$ Z descendantsr r �AssertionError�listZchildrenr r ) r r( r/ Z to_insertr� �p1r6 r7 Zp4r r r �1test_insert_beautifulsoup_object_inserts_childrenz s
zFTestTreeModification.test_insert_beautifulsoup_object_inserts_childrenc C sX | � d�}|j}|jd }|�dd� |j\}}|�d� |�d� | �d|jj� d S )Nz <p><a>one</a><b>three</b></p>r rG Ztwor� Zthree)r( r7 r� r$ Z replaceWithr r& r )r r( r7 r& �left�rightr r r �3test_replace_with_maintains_next_element_throughout� s
zHTestTreeModification.test_replace_with_maintains_next_element_throughoutc C sl | � d�}|jdd��d� |jdd�}|j}| �|j|� | �|j|� | �|jj|� | �|jd � d S )Nz <b>Argh!</b>�Argh!r< �Hooray!)r( r) r( r& r r� r� r� )r r( �new_textr& r r r �test_replace_final_node� s
z,TestTreeModification.test_replace_final_nodec C s� | � d�}|j�dd� | �|�� | �d�� |jdd�}| �|jd� | �|jj|� | �|j d� | �|j j
|� | �|j
d � | �|j|j � d S )Nz<a><b>Argh!</b><c></c></a>rG r@ z!<a><b>Argh!Hooray!</b><c></c></a>r< r? ) r( r& r$ r r� r r) r� r� r r� rm )r r( rA r r r �test_consecutive_text_nodes� s
�� z0TestTreeModification.test_consecutive_text_nodesc C sT | � d�}|j�dd� |j�dd� | �ddg|jj� | �|jjd jd� d S )NrP r r; rM )r( r7 r$ r r� r� r* r r r �test_insert_string� s
z'TestTreeModification.test_insert_stringc C s� | � � }| jd|d�}t||d�}|�dd� |j�d|� | �|�� | �d�� |j}| �|j |� | �|j
|� |j dd �}| �|j |� | �|j
|� |j}| �|j |� | �|j
|� |j dd �}| �|j|� | �|j |� | �|j
|� d S )
Nz%<a><b>Find</b><c>lady!</c><d></d></a>r! Zmagictagr �therG z=<a><b>Find</b><magictag>the</magictag><c>lady!</c><d></d></a>ZFindr< )r� r( r r$ r7 r r� r r& r� r r) r� r� rm r� )r r� r( Z magic_tagZb_tagr) Zc_tagrE r r r �test_insert_tag� s6 � �� z$TestTreeModification.test_insert_tagc C s0 d}| � |�}|j�|j� | �||�� � d S r1 )r( r7 rQ r& r r� �r r� r( r r r �*test_append_child_thats_already_at_the_end� s
z?TestTreeModification.test_append_child_thats_already_at_the_endc C sJ d}| � |�}|j|j|j|j|j|jg}|j�|� | � d|�
� � 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� rm r& r7 �extendr r� )r r� r( rR r r r � test_extend� s
z TestTreeModification.test_extendc C s2 d}| � |�}|j�d|j� | �d|�� � d S )Nz <a><b></b><c></c><d></d></a>r z <a><d></d><b></b><c></c></a>)r( r7 r$ r� r r� rG r r r �$test_move_tag_to_beginning_of_parent� s
z9TestTreeModification.test_move_tag_to_beginning_of_parentc C s. | � d�}|j�dd� | �t|j�d� d S )Nz<br/>rG ZContentsz<br>Contents</br>)r( r r$ r r. r* r r r �&test_insert_works_on_empty_element_tag� s
z;TestTreeModification.test_insert_works_on_empty_element_tagc C s� | � d�}|j�d� |j�d� | �|�� | �d�� |j�|j� | �|�� | �d�� |j}| �t|j|� |� � | �t|jd� | � d�}|j�|�
d�� d S ) Nr1 �BAZ�QUUXzQUUX<a>foo</a>BAZ<b>bar</b>�QUUX<b>bar</b><a>foo</a>BAZ�nope�<a>r7 ) r( r& �
insert_beforer7 r r� r r� r� r. r �r r( r& r r r �test_insert_before� s$
� �
z'TestTreeModification.test_insert_beforec C sj | � d�}|j�ddd� |j�ddd� | �|�� | �d�� |j�|jd� | �|�� | �d�� d S )Nr1 rO � rP 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& rT r7 r r� r r* r r r �test_insert_multiple_before s
� �z0TestTreeModification.test_insert_multiple_beforec C s� | � d�}|j�d� |j�d� | �|�� | �d�� |j�|j� | �|�� | �d�� |j}| �t|j|� |� � | �t|jd� | � d�}|j�
|� d�� d S ) Nr1 rO rP z<a>foo</a>QUUX<b>bar</b>BAZrQ rR rS r7 ) r( r& � insert_afterr7 r r� r r� r� r. rT r rU r r r �test_insert_after! s$
� �
z&TestTreeModification.test_insert_afterc C sj | � d�}|j�ddd� |j�ddd� | �|�� | �d�� |j�|jd� | �|�� | �d�� d S )Nr1 rO rW rP z$<a>foo</a>QUUX BAZ<b>bar</b>BAZ QUUXzFOO z(QUUX BAZ<b>bar</b><a>foo</a>FOO BAZ QUUX)r( r& rZ r7 r r� r r* r r r �test_insert_multiple_after7 s
� �z/TestTreeModification.test_insert_multiple_afterc C sR | � d�}|�d�}|�d�}| �t|j|� | �t|j|� | �t|j|� d S �Nr� r7 )r( r r r� r� rZ �NotImplementedError�r r( r r r r r �:test_insert_after_raises_exception_if_after_has_no_meaningA s
zOTestTreeModification.test_insert_after_raises_exception_if_after_has_no_meaningc C sR | � d�}|�d�}|�d�}| �t|j|� | �t|j|� | �t|j|� d S r] )r( r r r� r� rT r^ r_ r r r �Ftest_insert_before_raises_notimplementederror_if_before_has_no_meaningI s
z[TestTreeModification.test_insert_before_raises_notimplementederror_if_before_has_no_meaningc C sv | � d�}|�d�\}}|�|� | �|�� | �d�� | �|jd � | �|j|j� | �|jd� | �|j d� 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( r5 r( r r� r r� r r� r� )r r( rb Zshowr r r �test_replace_withQ s �
��z&TestTreeModification.test_replace_withc C s0 d}| � |�}|j�|j� | �d|�� � d S )Nr+ z<a><c></c></a>)r( r& r( rm r r� rG r r r �test_replace_first_child` s
z-TestTreeModification.test_replace_first_childc C s0 d}| � |�}|j�|j� | �d|�� � d S )Nr+ r2 )r( rm r( r& r r� rG r r r �test_replace_last_childf s
z,TestTreeModification.test_replace_last_childc C s | � d�}|j}|j}|�|� | �|�� | �d�� | �|jd � | �|jdd�j d � | �|j
d � | �|j d � | �|j d � | �|j|j
� | �|j
d� | �|j j |j� | �|j d � |jdd�}|j}| �|j |� | �|j |� | �|j
|� | �|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< ZWe�to)r( r& r� r( r r� r r� r) r� r� r� r r7 rJ rI )r r( Z
remove_tagZmove_tagZto_textZg_tagr r r � test_nested_tag_replace_withl s6 �
�� z1TestTreeModification.test_nested_tag_replace_withc C s6 | � d�}|j�� | �|jd � | �|jjd� d S )NzI
<p>Unneeded <em>formatting</em> is unneeded</p>
z Unneeded formatting is unneeded)r( �emr, r r r/ r~ r r r � test_unwrap� s
z TestTreeModification.test_unwrapc C sF | � d�}|j�|�d��}| �|�� d� | �|�� | �d�� d S )NzI wish I was bold.r& �<b>I wish I was bold.</b>)r( r �wrapr r r� r )r r( r� r r r � test_wrap� s
�z TestTreeModification.test_wrapc C s4 | � d�}|jj�|j� | �|�� | �d�� d S )Nz<b></b>I wish I was bold.rj )r( r& r� rk r r� r r* r r r �%test_wrap_extracts_tag_from_elsewhere� s
�z:TestTreeModification.test_wrap_extracts_tag_from_elsewherec C sH | � d�}|jj�|j� | �dt|jj�� | �|�� | �d�� d S )Nz+<b>I like being bold.</b>I wish I was bold.r2 z+<b>I like being bold.I wish I was bold.</b>) r( r& r� rk r r4 r� r� r r* r r r �&test_wrap_puts_new_contents_at_the_end� s
��z;TestTreeModification.test_wrap_puts_new_contents_at_the_endc C s� | � d�}| �t|jj�d� |jdd��� }| �|�� d� | �|�� d� | �t|jj�d� | �|jd � | �|j d � | �|j
j
d � |jdd �}|jd
d �}| �|j
|� | �|j |� | �|j |� | �|j |� d S ) NzR<html><body>Some content. <div id="nav">Nav crap</div> More content.</body></html>rB Znavr z6<html><body>Some content. More content.</body></html>z <div id="nav">Nav crap</div>r2 zSome content. r< z More content.)
r( r r4 r� r� r) r. r� r� r� r� r� r )r r( Z extractedZ content_1Z content_2r r r � test_extract� s( � � z!TestTreeModification.test_extractc C sz | � d�}|jj}|jj}|�d�}|�d�}|j�|� |j�|� |�� |�� | �||jj� | �||jj� d S )Nr1 rM r; )r( r7 r r& r rQ r. r )r r( Zfoo_1Zbar_1Zfoo_2Zbar_2r r r �4test_extract_distinguishes_between_identical_strings� s
zITestTreeModification.test_extract_distinguishes_between_identical_stringsc s8 | � d�� � fdd�� �d�D � | �dt� j�� d S )Nzv
<html>
<head>
<script>foo</script>
</head>
<body>
<script>bar</script>
<a></a>
</body>
<script>baz</script>
</html>c s g | ]}� j �� �qS r )�scriptr. )r r� �r( r r r � s zKTestTreeModification.test_extract_multiples_of_same_tag.<locals>.<listcomp>rq z<body>
<a></a>
</body>)r( r5 r r. r� rc r rr r �"test_extract_multiples_of_same_tag� s
z7TestTreeModification.test_extract_multiples_of_same_tagc C s. | � d�}|�d��� | �d |�d�� d S )Nz <html>
<body>hi</body>
</html>r� )r( r) r. r r* r r r �Btest_extract_works_when_element_is_surrounded_by_identical_strings� s
�zWTestTreeModification.test_extract_works_when_element_is_surrounded_by_identical_stringsc C sf | � d�}|j}|j�� | �t|jj�d� | �t|d�� |j }|jdd� | �dt|j�� dS )z Tag.clear()z4<p><a>String <em>Italicized</em></a> and another</p>r r� T)Z decomposeN)
r( r7 r �clearr r4 r� rU rV rh )r r( r7 rh r r r �
test_clear� s
z TestTreeModification.test_clearc C sB | � d�}d|j_| �|jjdg� d|j_| �|jjdg� dS )zTag.string = 'string'z<a></a> <b><c></c></b>rM r; N)r( r7 r r r� r& r* r r r �test_string_set� s
z$TestTreeModification.test_string_setc C s, | � d�}|jj|j_| �|j�� d� d S )Nz<a><b>foo</b><c>bar</c>s <a><b>bar</b><c>bar</c></a>)r( rm r r& r r7 r� r* r r r �/test_string_set_does_not_affect_original_string s
zDTestTreeModification.test_string_set_does_not_affect_original_stringc C s2 | � d�}td�}||j_| �t|jjt�� d S )NrP rM )r( r r7 r rU r �r r( �cdatar r r �)test_set_string_preserves_class_of_string s
z>TestTreeModification.test_set_string_preserves_class_of_stringN),r" r# r$ r r% r' r* r- r/ r0 r4 r5 r; r> rB rC rD rF rH rL rM rN rV rY r[ r\ r` ra rc rd re rg ri rl rm rn ro rp rs rt rv rw rx r{ r r r r r * sR
$ r c @ sx e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd S ) �TestElementObjectsz)Test various features of element objects.c C sV | � d�}| �t|j�d� | �t|�d� | �t|j�d� | �t|jj�d� dS )z3The length of an element is its number of children.z<top>1<b>2</b>3</top>rG rB N)r( r r4 r� r� r* r r r �test_len s
zTestElementObjects.test_lenc C sL | � d�}| �|j|�d�� | �|jj|�d��d�� | �|jd� dS )z2Accessing a Python member .foo invokes find('foo')�<b><i></i></b>r& r� N)r( r r& r) r� r7 r* r r r � test_member_access_invokes_find" s
z2TestElementObjects.test_member_access_invokes_findc C sP | � d�}tjdd�� }|j}W 5 Q R X | �|j|� | �dt|d j�� d S )Nr~ T)�recordzp.bTag is deprecated, use .find("b") instead. If you really were looking for a tag called bTag, use .find("bTag")r )r( �warnings�catch_warningsZbTagr r& r. �message)r r( �wr r r r � test_deprecated_member_access) s
�z0TestElementObjects.test_deprecated_member_accessc C s2 | � d�}| �|j�d�� | �|j�d�� dS )z�has_attr() checks for the presence of an attribute.
Please note note: has_attr() is different from
__in__. has_attr() checks the tag's attributes and __in__
checks the tag's chidlren.
z<foo attr='bar'>�attrZattr2N)r( rU rM �has_attr� assertFalser* r r r �
test_has_attr2 s
z TestElementObjects.test_has_attrc C s d}| � |d� d S )Nz%<b a="1" z="5" m="3" f="2" y="4"></b>z%<b a="1" f="2" m="3" y="4" z="5"></b>)ZassertSoupEqualsr� r r r �.test_attributes_come_out_in_alphabetical_order> s zATestElementObjects.test_attributes_come_out_in_alphabetical_orderc C s | � d�}| �|jjd� d S )Nz
<b>foo</b>rM �r( r r& r r* r r r � test_stringB s
z TestElementObjects.test_stringc C s | � d�}| �|jjd � d S )N�<b></b>r� r* r r r � test_empty_tag_has_no_stringH s
z/TestElementObjects.test_empty_tag_has_no_stringc C s` | � d�}| �|jjd � | � d�}| �|jjd � | � d�}|j�dd� | �|jjd � d S )Nz<a>foo<b></b><b></b></b>z<a>foo<b></b>bar</b>z
<a>foo</b>rG r; )r( r r& r r7 r$ r* r r r �-test_tag_with_multiple_children_has_no_stringM s
z@TestElementObjects.test_tag_with_multiple_children_has_no_stringc C s, | � d�}| �|jjd� | �|jd� d S )Nz<a><b>foo</b></a>rM )r( r r7 r r* r r r �)test_tag_with_recursive_string_has_string[ s
z<TestElementObjects.test_tag_with_recursive_string_has_stringc C s4 | � d�}| �|jj� | � d�}| �|jj� dS )z7Only a tag containing a single text node has a .string.z<b>f<i>e</i>o</b>r� N)r( r� r& r r* r r r �test_lack_of_stringb s
z&TestElementObjects.test_lack_of_stringc C s` | � d�}| �|jjd� | �|jjdd�d� | �|j�d�d� | �|jjddd�d� d S )
zBTag.text and Tag.get_text(sep=u"") -> all child text, concatenatedz <a>a<b>r</b> <r> t </r></a>zar t T)�stripZart�,z a,r, , t za,r,tN)r( r r7 r/ �get_textr* r r r �
test_all_textj s
z TestElementObjects.test_all_textc C sJ | � d�}| �|�� d� | �|jttfd�d� | �|jd d�d� d S )N�foo<!--IGNORE-->barZfoobar)�typesZ fooIGNOREbar)r( r r� r r r* r r r � test_get_text_ignores_commentsr s
�
�z1TestElementObjects.test_get_text_ignores_commentsc C s$ | � d�}| �ddgt|j�� d S )Nr� rM r; )r( r r9 Zstringsr* r r r �!test_all_strings_ignores_comments{ s
z4TestElementObjects.test_all_strings_ignores_commentsN)r" r# r$ r9 r} r r� r� r� r� r� r� r� r� r� r� r� r r r r r| s r| c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )�TestCDAtaListAttributesz0Testing cdata-list attributes like 'class'.
c C s" | � d�}| �dg|jd � d S )Nz<a class='foo'>rM r� �r( r r7 r* r r r � test_single_value_becomes_list� s
z6TestCDAtaListAttributes.test_single_value_becomes_listc C s$ | � d�}| �ddg|jd � d S )Nz<a class='foo bar'>rM r; r� r� r* r r r �!test_multiple_values_becomes_list� s
z9TestCDAtaListAttributes.test_multiple_values_becomes_listc C s& | � d�}| �dddg|jd � d S )Nz<a class='foo bar
baz'>rM r; r� r� r� r* r r r �2test_multiple_values_separated_by_weird_whitespace� s
zJTestCDAtaListAttributes.test_multiple_values_separated_by_weird_whitespacec C s | � d�}| �d|j�� � d S )Nz<a class='foo bar'>s <a class="foo bar"></a>)r( r r7 r� r* r r r �,test_attributes_joined_into_string_on_output� s
zDTestCDAtaListAttributes.test_attributes_joined_into_string_on_outputc C s$ | � d�}| �dg|j�d�� d S )Nz<a id='abc def'>zabc defr )r( r r7 Zget_attribute_listr* r r r �test_get_attribute_list� s
z/TestCDAtaListAttributes.test_get_attribute_listc C s$ | � d�}| �ddg|jd � d S )Nz(<form accept-charset="ISO-8859-1 UTF-8">z
ISO-8859-1zUTF-8�accept-charset)r( r Zformr* r r r �test_accept_charset� s
z+TestCDAtaListAttributes.test_accept_charsetc C s$ d}| � |�}| �d|jd � d S )Nz)<a accept-charset="ISO-8859-1 UTF-8"></a>zISO-8859-1 UTF-8r� r� rG r r r �-test_cdata_attribute_applying_only_to_one_tag� s
zETestCDAtaListAttributes.test_cdata_attribute_applying_only_to_one_tagc s6 | � d�j� | �d � j� � fdd�}| �t|� d S )Nr c s
d� _ d S )NrM r� r r r r �t� s zJTestCDAtaListAttributes.test_string_has_immutable_name_property.<locals>.t)r( r r r\ r� �AttributeError)r r� r r r �'test_string_has_immutable_name_property� s z?TestCDAtaListAttributes.test_string_has_immutable_name_propertyN) r" r# r$ r9 r� r� r� r� r� r� r� r� r r r r r� s r� c sh e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
� ZS )�TestPersistencez*Testing features like pickle and deepcopy.c s& t t| ��� d| _| �| j�| _d S )Nay <!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.0 Transitional//EN"
"http://www.w3.org/TR/REC-html40/transitional.dtd">
<html>
<head>
<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
<title>Beautiful Soup: We called him Tortoise because he taught us.</title>
<link rev="made" href="mailto:leonardr@segfault.org">
<meta name="Description" content="Beautiful Soup: an HTML parser optimized for screen-scraping.">
<meta name="generator" content="Markov Approximation 1.4 (module: leonardr)">
<meta name="author" content="Leonard Richardson">
</head>
<body>
<a href="foo">foo</a>
<a href="foo"><b>bar</b></a>
</body>
</html>)r` r� ra Zpager( rb rc rd r r ra � s zTestPersistence.setUpc C s@ t �| jd�}t �|�}| �|jt� | �|�� | j�� � d S )Nr2 )�pickle�dumpsrb �loadsr re r r� )r �dumped�loadedr r r �!test_pickle_and_unpickle_identity� s
z1TestPersistence.test_pickle_and_unpickle_identityc C s& t �| j�}| �|�� | j�� � d S r� )�copy�deepcopyrb r r� )r �copiedr r r �test_deepcopy_identity� s z&TestPersistence.test_deepcopy_identityc C s: t dd�}|j}|�� }| �dt|�� | �||j� d S )Ns
<p> </p>r u <p> </p>)r Zoriginal_encoding�__copy__r r. )r r( �encodingr� r r r � test_copy_preserves_encoding� s
z,TestPersistence.test_copy_preserves_encodingc C st | � d�j}d|_d|_|�� }| �|j|j� | �|j|j� | �|j|j� | �|j|j� | �|j|j� d S )Nz<p></p>rH �! ) r( r Z
sourcelineZ sourceposr� r Zcan_be_empty_elementZcdata_list_attributesr
)r r r� r r r �'test_copy_preserves_builder_information� s$ � � �z7TestPersistence.test_copy_preserves_builder_informationc C s> d}| � |�}t�|tj�}t�|�}| �|�� |�� � d S )N�
<b>☃</b>)r( r� r� ZHIGHEST_PROTOCOLr� r r� )r r� r( r� r� r r r �test_unicode_pickle� s
z#TestPersistence.test_unicode_picklec C sz d}| � |�}|jdd�}t�|�}| �||� | �d |j� | �d |j� | �d |j� | �d |j� | �d |j� d S )Nz<b>Foo<a></a></b><b>Bar</b>r= r ) r( r) r� r r� r� �assertNotEqualr� r� �r r� r( �s1�s2r r r �1test_copy_navigablestring_is_not_attached_to_tree� s
zATestPersistence.test_copy_navigablestring_is_not_attached_to_treec C s> d}| � |�}|j}t�|�}| �||� | �t|t�� d S )Nz<b><!--Foo--></b>)r( r r� r rU r r r� r r r �0test_copy_navigablestring_subclass_has_same_type s
z@TestPersistence.test_copy_navigablestring_subclass_has_same_typec C s( d}| � |�}t�|�}| �||� d S )N�)<div><b>Foo<a></a></b><b>Bar</b></div>end)r( r� r )r r� r( Z soup_copyr r r �test_copy_entire_soup
s
z%TestPersistence.test_copy_entire_soupc C s� d}| � |�}|j}t�|�}| �t|�t|�� | �||� | �||k� | �d |j� | �d |j� | �d |jdd�j � | �
d |jdd�j � d S )Nr� ZBarr ) r( ry r� r r. r� r� r� r) r� r� )r r� r( ry Zdiv_copyr r r � test_copy_tag_copies_contents s
z-TestPersistence.test_copy_tag_copies_contents)r" r# r$ r9 ra r� r� r� r� r� r� r� r� r� r{ r r rd r r� � s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!S )"�TestSubstitutionsc C s0 d}| � |�}|jdd�}| �|| �d�� d S �N�# <b><<Sacré bleu!>></b>�minimal�� formatter�r( r� r r �r r� r( �decodedr r r �!test_default_formatter_is_minimal& s
��z3TestSubstitutions.test_default_formatter_is_minimalc C s0 d}| � |�}|jdd�}| �|| �d�� d S )N�' <br><b><<Sacré bleu!>></b>r� r� z.<br/><b><<Sacré bleu!>></b>r� r� r r r �test_formatter_html0 s
�z%TestSubstitutions.test_formatter_htmlc C s0 d}| � |�}|jdd�}| �|| �d�� d S )Nr� Zhtml5r� z-<br><b><<Sacré bleu!>></b>r� r� r r r �test_formatter_html58 s
�z&TestSubstitutions.test_formatter_html5c C s0 d}| � |�}|jdd�}| �|| �d�� d S r� r� r� r r r �test_formatter_minimal@ s
��z(TestSubstitutions.test_formatter_minimalc C s0 d}| � |�}|jd d�}| �|| �d�� d S )Nr� r� u <b><<Sacré bleu!>></b>r� r� r r r �test_formatter_nullJ s
�z%TestSubstitutions.test_formatter_nullc C s4 d}| � |�}|jdd� d�}| �|| �d�� d S )Nz!<b><foo></b><b>bar</b><br/>c S s | � � S r� ��upper��xr r r �<lambda>V � z9TestSubstitutions.test_formatter_custom.<locals>.<lambda>r� z<b><FOO></b><b>BAR</b><br/>r� r� r r r �test_formatter_customS s
�z'TestSubstitutions.test_formatter_customc C s� d}| � |�}|j}d}| �||�� � | �||jdd�� d}| �||jdd�� | �||jd d�� d}| �||jdd � d�� d S )
Nu% <a href="http://a.com?a=b&c=é">e</a>u) <a href="http://a.com?a=b&c=é">e</a>r� r� z/<a href="http://a.com?a=b&c=é">e</a>r� u% <a href="HTTP://A.COM?A=B&C=É">E</a>c S s | � � S r� r� r� r r r r� l r� zMTestSubstitutions.test_formatter_is_run_on_attribute_values.<locals>.<lambda>)r( r7 r r� )r r� r( r7 Zexpect_minimalZ expect_htmlZ expect_upperr r r �)test_formatter_is_run_on_attribute_values] s
z;TestSubstitutions.test_formatter_is_run_on_attribute_valuesc C s$ d}t |d��� }| �d|k� d S )NzO
<script type="text/javascript">
console.log("< < hey > > ");
</script>
r � < < hey > >�r r� rU �r r& �encodedr r r �2test_formatter_skips_script_tag_for_html_documentsn s zDTestSubstitutions.test_formatter_skips_script_tag_for_html_documentsc C s$ d}t |d��� }| �d|k� d S )NzF
<style type="text/css">
console.log("< < hey > > ");
</style>
r r� r� r� r r r �1test_formatter_skips_style_tag_for_html_documentsw s zCTestSubstitutions.test_formatter_skips_style_tag_for_html_documentsc C s | � d�}| �d|j�� � d S )NzN<div> foo <pre> bar
</pre> baz <textarea> eee
fff </textarea></div>zO<div>
foo
<pre> bar
</pre>
baz
<textarea> eee
fff </textarea>
</div>)r( r ry �prettifyr* r r r �,test_prettify_leaves_preformatted_text_alone� s
�z>TestSubstitutions.test_prettify_leaves_preformatted_text_alonec C s, t dd�}|jdd� d�}| �d|k� d S )Nz <html><body>foo</body></html>r c S s | � � S r� r� r� r r r r� � r� zLTestSubstitutions.test_prettify_accepts_formatter_function.<locals>.<lambda>r� rX )r r� rU )r r( Zprettyr r r �(test_prettify_accepts_formatter_function� s
z:TestSubstitutions.test_prettify_accepts_formatter_functionc C s" | � d�}| �tt|�� �� d S rO )r( r r. �typer� r* r r r �(test_prettify_outputs_unicode_by_default� s
z:TestSubstitutions.test_prettify_outputs_unicode_by_defaultc C s$ | � d�}| �tt|�d��� d S )NrP �utf-8)r( r �bytesr� r� r* r r r � test_prettify_can_encode_data� s
z/TestSubstitutions.test_prettify_can_encode_datac C s0 d}| � |�}|j�d�}| �||�d�� d S )Nu <b>Sacré bleu!</b>r� )r( r& r� r )r r� r( r� r r r �,test_html_entity_substitution_off_by_default� s
z>TestSubstitutions.test_html_entity_substitution_off_by_defaultc C s� d}| � |�}| �|jd d� |�d�}| �d|k� |�d�}| �d|k� |�d�}| �d |k� |�d
��d
�}| �d |k� d S ) NzE<meta content="text/html; charset=x-sjis" http-equiv="Content-type"/>Zcontentztext/html; charset=x-sjisr� s
charset=utf-8�euc_jps charset=euc_jpz shift-jiss charset=shift-jiszutf-16zcharset=utf-16)r( r �metar� rU r� )r Zmeta_tagr( �utf_8r� � shift_jisZutf_16_ur r r �test_encoding_substitution� s
z,TestSubstitutions.test_encoding_substitutionc C s2 d}t d�}| j||d�}| �|jd jd� d S )Nz`<head><meta content="text/html; charset=x-sjis" http-equiv="Content-type"/></head><pre>foo</pre>Zpre)Z
parse_onlyr )r
r( r r� r\ )r r� r� r( r r r �;test_encoding_substitution_doesnt_happen_if_tag_is_strained� s zMTestSubstitutions.test_encoding_substitution_doesnt_happen_if_tag_is_strainedN)r" r# r$ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� $ s
r� c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )� TestEncodingz0Test the ability to encode objects into strings.c C s. d}| � |�}| �|jj�d�d�d�� d S )Nr� r� � ☃)r( r r& r r� �r r� r( r r r �"test_unicode_string_can_be_encoded� s
�z/TestEncoding.test_unicode_string_can_be_encodedc C s, d}| � |�}| �|j�d�|�d�� d S )Nr� r� �r( r r& r� r� r r r �1test_tag_containing_unicode_string_can_be_encoded� s
�z>TestEncoding.test_tag_containing_unicode_string_can_be_encodedc C s& d}| � |�}| �|j�d�d� d S )Nr� �asciis <b>☃</b>r� r� r r r �<test_encoding_substitutes_unrecognized_characters_by_default� s
zITestEncoding.test_encoding_substitutes_unrecognized_characters_by_defaultc C s& d}| � |�}| jt|jddd� d S )Nr� r� �strict)�errors)r( r� �UnicodeEncodeErrorr� r� r r r � test_encoding_can_be_made_strict� s
�z-TestEncoding.test_encoding_can_be_made_strictc C s$ d}| � |�}| �d|j�� � d S )Nr� r� )r( r r& Zdecode_contentsr� r r r �test_decode_contents� s
z!TestEncoding.test_decode_contentsc C s. d}| � |�}| �d�d�|jjdd�� d S )Nr� r� r� )r� )r( r r� r& Zencode_contentsr� r r r �test_encode_contents� s
��z!TestEncoding.test_encode_contentsc C s* d}| � |�}| �d�d�|j�� � d S )Nr� r� r� )r( r r� r& ZrenderContentsr� r r r � test_deprecated_renderContents� s
�z+TestEncoding.test_deprecated_renderContentsc C s8 d}| � |�}tr$| �|t|�� n| �dt|�� d S )Nr� s
<b>\u2603</b>)r( r r �reprr� r r r � test_repr� s
zTestEncoding.test_reprN) r" r# r$ r9 r� r� r� r� r� r� r� r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�
TestFormatterc C sL G dd� dt �}| �d�}|� }|j|d�}| �|j|j� | �d|� d S )Nc @ s e Zd Zdd� ZdS )z=TestFormatter.test_sort_attributes.<locals>.UnsortedFormatterc s s6 || _ t|j�� �D ] \}}|dkr&q||fV qd S )N�ignore)� called_with�sortedr[ �items)r r �k�vr r r �
attributes� s
zHTestFormatter.test_sort_attributes.<locals>.UnsortedFormatter.attributesN)r" r# r$ r r r r r �UnsortedFormatter� s r z*<p cval="1" aval="2" ignore="ignored"></p>r� z<p aval="2" cval="1"></p>)r r( r� r r r )r r r( r� r� r r r �test_sort_attributes� s
z"TestFormatter.test_sort_attributesN)r" r# r$ r r r r r r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� TestNavigableStringSubclassesc C sX | � d�}td�}|�d|� | �t|�d� | �|jdd�d� | �|jd d� d S )Nr� rM rG z<![CDATA[foo]]>r< r )r( r r$ r r. r) r� ry r r r �
test_cdata s
z(TestNavigableStringSubclasses.test_cdatac sV d� _ � fdd�}� �d�}td�}|�d|� � �d|j|d�� � �d� j � d S )
zkText inside a CData object is passed into the formatter.
But the return value is ignored.
r c s � j d7 _ dS )NrG zBITTER FAILURE)�count)�argsrc r r � increment s zNTestNavigableStringSubclasses.test_cdata_is_never_formatted.<locals>.incrementr� z<><><>rG s <![CDATA[<><><>]]>r� N)r r( r r$ r r� )r r
r( rz r rc r � test_cdata_is_never_formatted s
�z;TestNavigableStringSubclasses.test_cdata_is_never_formattedc C s2 t d�}| �d�}|�d|� | �|�� d� d S )NrM r� rG s <!DOCTYPE foo>
)r
r( r$ r r� )r Zdoctyper( r r r � test_doctype_ends_in_newline' s
z:TestNavigableStringSubclasses.test_doctype_ends_in_newlinec C s t d�}| �d|�� � d S )NrM z<?foo?>)r r Z output_ready)r r� r r r �test_declaration/ s z.TestNavigableStringSubclasses.test_declarationN)r" r# r$ r
r r r r r r r r s
r c @ s� e Zd ZdZdd� Zdd� ZeZdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z!d<d=� Z"d>d?� Z#d@dA� Z$dBdC� Z%dDdE� Z&dFdG� Z'dHdI� Z(dJdK� Z)dLdM� Z*dNdO� Z+dPdQ� Z,dRdS� Z-dTdU� Z.dVdW� Z/dXdY� Z0dZd[� Z1d\d]� Z2d^d_� Z3d`da� Z4dbdc� Z5ddde� Z6dfdg� Z7dhdi� Z8djdk� Z9dldm� Z:dndo� Z;dpdq� Z<drds� Z=dtS )u�TestSoupSelectora�
<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01//EN"
"http://www.w3.org/TR/html4/strict.dtd">
<html>
<head>
<title>The title</title>
<link rel="stylesheet" href="blah.css" type="text/css" id="l1">
</head>
<body>
<custom-dashed-tag class="dashed" id="dash1">Hello there.</custom-dashed-tag>
<div id="main" class="fancy">
<div id="inner">
<h1 id="header1">An H1</h1>
<p>Some text</p>
<p class="onep" id="p1">Some more text</p>
<h2 id="header2">An H2</h2>
<p class="class1 class2 class3" id="pmulti">Another</p>
<a href="http://bob.example.org/" rel="friend met" id="bob">Bob</a>
<h2 id="header3">Another H2</h2>
<a id="me" href="http://simonwillison.net/" rel="me">me</a>
<span class="s1">
<a href="#" id="s1a1">span1a1</a>
<a href="#" id="s1a2">span1a2 <span id="s1a2s1">test</span></a>
<span class="span2">
<a href="#" id="s2a1">span2a1</a>
</span>
<span class="span3"></span>
<custom-dashed-tag class="dashed" id="dash2"/>
<div data-tag="dashedvalue" id="data1"/>
</span>
</div>
<x id="xid">
<z id="zida"/>
<z id="zidab"/>
<z id="zidac"/>
</x>
<y id="yid">
<z id="zidb"/>
</y>
<p lang="en" id="lang-en">English</p>
<p lang="en-gb" id="lang-en-gb">English UK</p>
<p lang="en-us" id="lang-en-us">English US</p>
<p lang="fr" id="lang-fr">French</p>
</div>
<div id="footer">
</div>
c C s t | jd�| _d S )Nr )r �HTMLr( rc r r r ra f s zTestSoupSelector.setUpc
K sR dd� | j j|f|�D �}|�� |�� | �||d|d�|�d�|�f � d S )Nc S s g | ] }|d �qS r r )r �elr r r r j s z2TestSoupSelector.assertSelects.<locals>.<listcomp>z$Selector %s, expected [%s], got [%s]z, )r( �select�sortr �join)r �selector� expected_ids�kwargsZel_idsr r r r i s ��z TestSoupSelector.assertSelectsc G s |D ]\}}| � ||� qd S r� )� assertSelect)r Ztestsr r r r r �assertSelectMultipleu s z%TestSoupSelector.assertSelectMultiplec C sF | j �d�}| �t|�d� | �|d jd� | �|d jdg� d S )Nr� rG r z The title)r( r r r4 r\ r� �r �elsr r r �test_one_tag_oney s z!TestSoupSelector.test_one_tag_onec C sT | j �d�}| �t|�d� |D ]}| �|jd� q | j �d�}| �d|d � d S )Nry � �mainr )r( r r r4 r\ �
select_one)r r ry r r r r �test_one_tag_many s z"TestSoupSelector.test_one_tag_manyc C s | j �d�}| �d |� d S )NZnonexistenttag)r( r! r )r �matchr r r �(test_select_one_returns_none_if_no_match� s z9TestSoupSelector.test_select_one_returns_none_if_no_matchc C s | j �d�}| �dddg� d S )Nzdiv div�inner�data1)r( r r r r r r �test_tag_in_tag_one� s z$TestSoupSelector.test_tag_in_tag_onec C s" dD ]}| � |ddddg� qd S )N)�html div�
html body div�body divr& r r% �footer�r �r r r r r �test_tag_in_tag_many� s z%TestSoupSelector.test_tag_in_tag_manyc C sB | j ddgdd� | j dddgdd� | j dd ddd
gd d� d S ) Nr( r rG rC r) r% r2 r* r& r+ rH r, rc r r r �
test_limit� s
�zTestSoupSelector.test_limitc C s | � t| j�d��d� d S )N�delr �r r4 r( r rc r r r �test_tag_no_match� s z"TestSoupSelector.test_tag_no_matchc C s | � t| jjd� d S )Nztag%t�r� � SyntaxErrorr( r rc r r r �test_invalid_tag� s z!TestSoupSelector.test_invalid_tagc C s | � dddg� d S )N�custom-dashed-tag�dash1�dash2r, rc r r r �test_select_dashed_tag_ids� s z+TestSoupSelector.test_select_dashed_tag_idsc C s6 | j �d�}| �|d jd� | �|d d d� d S )Nz custom-dashed-tag[id="dash2"]r r6 r r8 )r( r r r\ )r Zdashedr r r �test_select_dashed_by_id� s z)TestSoupSelector.test_select_dashed_by_idc C s | � | j�d�d jd� d S )Nzbody > custom-dashed-tagr z Hello there.)r r( r r/ rc r r r �test_dashed_tag_text� s z%TestSoupSelector.test_dashed_tag_textc C s | � | j�d�| j�d�� d S )Nr6 )r r( r r5 rc r r r �#test_select_dashed_matches_find_all� s z4TestSoupSelector.test_select_dashed_matches_find_allc C s | � ddgfdddgf� d S )NZh1�header1Zh2�header2�header3�r rc r r r �test_header_tags� s
�z!TestSoupSelector.test_header_tagsc C sR dD ]H}| j �|�}| �t|�d� | �|d jd� | �|d d dg� qd S )N)z.onepzp.onepz html p.oneprG r r r� �onep)r( r r r4 r\ )r r r r r r �test_class_one� s
z TestSoupSelector.test_class_onec C s | j �d�}| �t|�d� d S )Nzdiv.onepr �r( r r r4 r r r r �test_class_mismatched_tag� s z*TestSoupSelector.test_class_mismatched_tagc C s dD ]}| � |dg� qd S )N)z div#innerz#innerz
div div#innerr% r, r- r r r � test_one_id� s z TestSoupSelector.test_one_idc C s | j �d�}| �t|�d� d S )Nz
#doesnotexistr rD r r r r � test_bad_id� s z TestSoupSelector.test_bad_idc C sb | j �d�}| �t|�d� |D ]}| �|jd� q | �|d d dg� | �|d �d�� d S )Nz div#inner prB r rG r� rB r )r( r r r4 r\ r� r� )r r r r r r �test_items_in_id� s z!TestSoupSelector.test_items_in_idc C s& dD ] }| � t| j�|��d� qd S )N)z div#main delzdiv#main div.oopsz div div#mainr r1 r- r r r �test_a_bunch_of_emptys� s z'TestSoupSelector.test_a_bunch_of_emptysc C s dD ]}| � |dg� qd S )N)z.class1zp.class1z.class2zp.class2z.class3zp.class3z
html p.class2zdiv#inner .class2�pmultir, r- r r r �test_multi_class_support� s z)TestSoupSelector.test_multi_class_supportc C s dD ]}| � |dg� qd S )N)z.class1.class3z.class3.class2z.class1.class2.class3rJ r, r- r r r �test_multi_class_selection� s z+TestSoupSelector.test_multi_class_selectionc C s" | � dddg� | � ddg� d S )Nz.s1 > a�s1a1�s1a2z .s1 > a span�s1a2s1r, rc r r r �test_child_selector� s z$TestSoupSelector.test_child_selectorc C s | � ddg� d S )Nz.s1 > a#s1a2 spanrO r, rc r r r �test_child_selector_id� s z'TestSoupSelector.test_child_selector_idc C st | � ddgfddgfddgfddgfddgfddgfd dgfd
g fd dgfd dgfd
dgfdg fdg fdg f� d S )Nzp[class="onep"]r: z
p[id="p1"]z[class="onep"]z [id="p1"]zlink[rel="stylesheet"]�l1zlink[type="text/css"]zlink[href="blah.css"]zlink[href="no-blah.css"]z[rel="stylesheet"]z[type="text/css"]z[href="blah.css"]z[href="no-blah.css"]zp[href="no-blah.css"]r@ rc r r r �test_attribute_equals� s �z&TestSoupSelector.test_attribute_equalsc
C s\ | � ddgfddgfddgfddgfddgfddgfdd gfd
d gfd d gfd d gf�
d S )
Nzp[class~="class1"]rJ zp[class~="class2"]zp[class~="class3"]z[class~="class1"]z[class~="class2"]z[class~="class3"]za[rel~="friend"]�bobz
a[rel~="met"]z[rel~="friend"]z [rel~="met"]r@ rc r r r �test_attribute_tilde� s �z%TestSoupSelector.test_attribute_tildec C sv | � ddgfddgfdg fdg fdg fddgfdd d
gfd d d
gfd d
dgfdd
dgfddgfdd
gfddgf�
d S )Nz[rel^="style"]rR zlink[rel^="style"]znotlink[rel^="notstyle"]z[rel^="notstyle"]zlink[rel^="notstyle"]zlink[href^="bla"]za[href^="http://"]rT �mez[href^="http://"]z [id^="p"]rJ r: z [id^="m"]r z div[id^="m"]z
a[id^="m"]zdiv[data-tag^="dashed"]r& r@ rc r r r �test_attribute_startswith s
�z*TestSoupSelector.test_attribute_startswithc C sH | � ddgfddgfddgfdddddd d
d d gfd
dgfdg f� d S )Nz[href$=".css"]rR zlink[href$=".css"]z
link[id$="1"]z [id$="1"]r& r: r= rM �s2a1rO r7 z div[id$="1"]z[id$="noending"]r@ rc r r r �test_attribute_endswith s �z(TestSoupSelector.test_attribute_endswithc C s� | � ddgfddgfdg fdg fdg fddgfdd d
gfd d d
gfddgfdd
gfddgfddgfddgfdddd
ddddddg fd dgfd g fd d d
dgfd d d
gfd dgfd!dd"gfd#d"gfd$dgf� d S )%Nz[rel*="style"]rR zlink[rel*="style"]znotlink[rel*="notstyle"]z[rel*="notstyle"]zlink[rel*="notstyle"]zlink[href*="bla"]z[href*="http://"]rT rV z [id*="p"]rJ r: z div[id*="m"]r z
a[id*="m"]z[href*=".css"]zlink[href*=".css"]z
link[id*="1"]z [id*="1"]r& r= rM rN rX rO r7 z div[id*="1"]z[id*="noending"]z [href*="."]z a[href*="."]zlink[href*="."]z div[id*="n"]r% z
div[id*="nn"]zdiv[data-tag*="edval"]r@ rc r r r �test_attribute_contains! s0
�z(TestSoupSelector.test_attribute_containsc C s2 | � ddddgfddddgfddgfdg f� d S ) Nz
p[lang|="en"]�lang-en�
lang-en-gb�
lang-en-usz [lang|="en"]z
p[lang|="fr"]�lang-frz
p[lang|="gb"]r@ rc r r r � test_attribute_exact_or_hypen> s �z.TestSoupSelector.test_attribute_exact_or_hypenc C sV | � ddddgfddgfdddgfddd d
d gfd d
dgfdg fdg fddgf� d S )Nz[rel]rR rT rV z link[rel]za[rel]z[lang]r[ r\ r] r^ zp[class]r: rJ z[blah]zp[blah]z
div[data-tag]r& r@ rc r r r �test_attribute_existsF s
�z&TestSoupSelector.test_attribute_existsc C s, d}t |d�}|�d�\}| �d|j� d S )Nz]<div style="display: wrong">nope</div>
<div style="display: right">yes</div>
r zdiv[style="display: right"]Zyes)r r r r )r r� r( Zchosenr r r �"test_quoted_space_in_selector_nameR s
z3TestSoupSelector.test_quoted_space_in_selector_namec C s( | � t| jjd� | � t| jjd� d S )Nza:no-such-pseudoclassza:nth-of-type(a))r� r^ r( r r4 rc r r r � test_unsupported_pseudoclassZ s � �z-TestSoupSelector.test_unsupported_pseudoclassc C s� | j �d�}| �t|�d� | �|d jd� | j �d�}| �t|�d� | �|d jd� | j �d�}| �t|�d� | j �d�}| �t|�d� d S ) Nzdiv#inner p:nth-of-type(1)rG r � Some textzdiv#inner p:nth-of-type(3)ZAnotherzdiv#inner p:nth-of-type(4)zdiv p:nth-of-type(0)�r( r r r4 r r r r r �test_nth_of_typea s z!TestSoupSelector.test_nth_of_typec C s2 | j �d�}| �t|�d� | �|d jd� d S )Nz div#inner > p:nth-of-type(1)rG r rc rd r r r r �"test_nth_of_type_direct_descendantt s z3TestSoupSelector.test_nth_of_type_direct_descendantc C s | � ddg� d S )Nz#inner > p:nth-of-type(2)r: r, rc r r r �"test_id_child_selector_nth_of_typey s z3TestSoupSelector.test_id_child_selector_nth_of_typec C s. | j jddd�}|�d�}| �|ddg� d S )Nry r r r% r& )r( r) r r! )r r% �selectedr r r �test_select_on_element| s
z'TestSoupSelector.test_select_on_elementc C s | � ddg� | � dg � d S )Nz
.fancy #innerr% z.normal #innerr, rc r r r �test_overspecified_child_id� s z,TestSoupSelector.test_overspecified_child_idc C sB | � ddg� | � ddg� | � ddg� | �g | j�d�� d S )Nz#p1 + h2r> z #p1 + h2 + prJ z#p1 + #header2 + .class1z#p1 + p�r r r( r rc r r r � test_adjacent_sibling_selector� s z/TestSoupSelector.test_adjacent_sibling_selectorc C sR | � dddg� | � ddg� | � ddg� | � ddg� | �g | j�d�� d S ) Nz#p1 ~ h2r> r? z#p1 ~ #header2z #p1 ~ h2 + arV z#p1 ~ h2 + [rel="me"]z #inner ~ h2rk rc r r r � test_general_sibling_selector� s
z.TestSoupSelector.test_general_sibling_selectorc C s | � t| jjd� d S )Nzh1 >r3 rc r r r �test_dangling_combinator� s z)TestSoupSelector.test_dangling_combinatorc C s | � ddddg� d S )Nz p[lang] ~ pr\ r] r^ r, rc r r r �2test_sibling_combinator_wont_select_same_tag_twice� s zCTestSoupSelector.test_sibling_combinator_wont_select_same_tag_twicec C s | � dddg� d S )Nzx, y�xid�yidr, rc r r r �test_multiple_select� s z%TestSoupSelector.test_multiple_selectc C s | � dddg� d S )Nzx,yrp rq r, rc r r r �"test_multiple_select_with_no_space� s z3TestSoupSelector.test_multiple_select_with_no_spacec C s | � dddg� d S )Nzx, yrp rq r, rc r r r �$test_multiple_select_with_more_space� s z5TestSoupSelector.test_multiple_select_with_more_spacec C s | � ddg� d S )Nzx, xrp r, rc r r r � test_multiple_select_duplicated� s z0TestSoupSelector.test_multiple_select_duplicatedc C s | � dddg� d S )Nzx, y ~ p[lang=fr]rp r^ r, rc r r r � test_multiple_select_sibling� s z-TestSoupSelector.test_multiple_select_siblingc C s | � dddg� d S )Nzx, y > zrp �zidbr, rc r r r �.test_multiple_select_tag_and_direct_descendant� s z?TestSoupSelector.test_multiple_select_tag_and_direct_descendantc C s | � dddddddg� d S )Nz
div > x, y, zrp rq �zidarw �zidab�zidacr, rc r r r �/test_multiple_select_direct_descendant_and_tags� s z@TestSoupSelector.test_multiple_select_direct_descendant_and_tagsc C s | � dddddddg� d S )Nz div x,y, zrp rq ry rw rz r{ r, rc r r r �(test_multiple_select_indirect_descendant� s z9TestSoupSelector.test_multiple_select_indirect_descendantc C s( | � t| jjd� | � t| jjd� d S )Nz,x, yzx,,yr3 rc r r r � test_invalid_multiple_select� s z-TestSoupSelector.test_invalid_multiple_selectc C s | � dddg� d S )Nzp[lang=en], p[lang=en-gb]r[ r\ r, rc r r r �test_multiple_select_attrs� s z+TestSoupSelector.test_multiple_select_attrsc C s | � ddddg� d S )Nz*x, y > z[id=zida], z[id=zidab], z[id=zidb]rp rw rz r, rc r r r �test_multiple_select_ids� s z)TestSoupSelector.test_multiple_select_idsc C s | � dddg� d S )Nzbody > div > x, y > zrp rw r, rc r r r �test_multiple_select_nested� s z,TestSoupSelector.test_multiple_select_nestedc C sN d}t |d�}|�d�}| �dt|�� |jddgd�D ]}||ks8t�q8d S )Nz3<div class="c1"/><div class="c2"/><div class="c1"/>r z.c1, .c2rB Zc1Zc2r� )r r r r4 r5 r8 )r r� r( rh r� r r r � test_select_duplicate_elements� s
z/TestSoupSelector.test_select_duplicate_elementsN)>r" r# r$ r ra r r r r r" r$ r' r. r/ r2 r5 r9 r: r; r< rA rC rE rF rG rH rI rK rL rP rQ rS rU rW rY rZ r_ r` ra rb re rf rg ri rj rl rm rn ro rr rs rt ru rv rx r| r} r~ r r� r� r� r r r r r 3 sv 1
r )6r9 Zpdbr r� r� r? r� Zbs4r Z bs4.builderr r Z bs4.elementr r r r r
r r r
r Z bs4.testingr r r# r Z LXML_PRESENTr r% r: rY r^ r| r� r� r� r� r� r� r� r� r r r r r| r� r� r� r� r� r r r r r r �<module> sT
, ;O 6,3(3--* jn+z 4*