U
lu
fp @ sn d Z ddlmZ ddlZddlZddlZddlmZmZm Z ddl
m Z ddl m
Z
mZ G dd d e
eZdS )
zGTests to ensure that the html.parser tree builder generates good
trees. ) set_traceN)HTMLParserTreeBuilderParserRejectedMarkupXMLParsedAsHTMLWarning)BeautifulSoupHTMLParser )SoupTestHTMLTreeBuilderSmokeTestc @ sX e Zd ZeZdd Zdd Zdd Zdd Zd d
Z d d Z
d
d Z dd Z dd Z
dS )TestHTMLParserTreeBuilderc
C s8 dddg}|D ]$}t t | |}W 5 Q R X qd S )Ns
)pytestZraisesr soup)selfZ
bad_markupmarkupr r =/tmp/pip-unpacked-wheel-pg4vfqbd/bs4/tests/test_htmlparser.pytest_rejected_input s
z-TestHTMLParserTreeBuilder.test_rejected_inputc C s d S Nr r
r r r test_namespaced_system_doctype* s z8TestHTMLParserTreeBuilder.test_namespaced_system_doctypec C s d S r r r r r r test_namespaced_public_doctype. s z8TestHTMLParserTreeBuilder.test_namespaced_public_doctypec C s: | d}t|d}t|}t|jt|js6tdS )zfUnlike most tree builders, HTMLParserTreeBuilder and will
be restored after pickling.
z
foo N)r pickledumpsloads
isinstanceZbuildertypeAssertionError)r
treeZdumpedZloadedr r r test_builder_is_pickled2 s
z1TestHTMLParserTreeBuilder.test_builder_is_pickledc C s | dd | dd d S )Nz
z
z Z assert_soupr r r r )test_redundant_empty_element_closing_tags; s zCTestHTMLParserTreeBuilder.test_redundant_empty_element_closing_tagsc C s | dd d S )Nz
foo &# barzfoo &# barr r r r r test_empty_element? s z,TestHTMLParserTreeBuilder.test_empty_elementc C sz d}| |}d|jjks td|jjks.td|jdjksDt| j |dd}d|jjjksdtd|jjjksvtd S )Nz=
text
r
sourcelineF)Zstore_line_numbers sourcepos)r pr$ r r% findname)r
r r r r r test_tracking_line_numbersD s
z4TestHTMLParserTreeBuilder.test_tracking_line_numbersc s d }d|jd ks tdg|jd ks4td|jd ksFt fdd}|d d |tjd |tjd d
d }||d d dg d S )
Nz;Zurl3hrefclsclassidc sJ j | d}||jd ks tdg|jd ks4td|jd ksFtd S )N)on_duplicate_attributer* r+ r, r- )r ar )r. expectedr r r
r r assert_attribute` s zOTestHTMLParserTreeBuilder.test_on_duplicate_attribute..assert_attributeurl1c S s. t | | ts | | g| |< | | | d S r )r listappend)attrskeyvaluer r r
accumulatep s zITestHTMLParserTreeBuilder.test_on_duplicate_attribute..accumulateurl2)r r/ r r ZREPLACEZIGNORE)r
r r2 r9 r r1 r test_on_duplicate_attributeR s
z5TestHTMLParserTreeBuilder.test_on_duplicate_attributec C sj dD ]`\}}}d| }| |j}| }d|d }||ksDt|jdd}d| }||kstqd S )N))z⇄u ⇄s ⇄)z⊧u ⊧s ⊧)z𝔑u 𝔑s 𝔑)z≧̸u ≧̸s ≧̸)z¬ ¬s ¬)z⫬u ⫬s ⫬)z"" ")z∴ ∴ ∴)z ∴r? r@ )z ∴r? r@ )zfjfjs fj)z⊔u ⊔s ⊔)z⊔︀u ⊔︀s ⊔︀)z'' ')z|| |z
%s
s
%s
utf8html) formatter)r divencoder ) r
Z
input_elementZoutput_unicodeZoutput_elementr rI Zwithout_elementexpectZ with_elementr r r test_html5_attributesv s z/TestHTMLParserTreeBuilder.test_html5_attributesN)__name__
__module__ __qualname__r Zdefault_builderr r r r r! r" r) r; rL r r r r r
s $r
)__doc__pdbr r r warningsZ bs4.builderr r r Zbs4.builder._htmlparserr r r r r
r r r r s