3
dp @ 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 s< dddg}x,|D ]$}t jt | j|}W d Q R X qW d S )Ns
)pytestZraisesr soup)selfZ
bad_markupmarkupr r C/tmp/pip-build-8z3xcdsh/beautifulsoup4/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 )Nr )r
r r r test_namespaced_public_doctype. s z8TestHTMLParserTreeBuilder.test_namespaced_public_doctypec C s: | j d}tj|d}tj|}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 | j dd | j dd d S )Nz
z
z ) assert_soup)r
r r r )test_redundant_empty_element_closing_tags; s zCTestHTMLParserTreeBuilder.test_redundant_empty_element_closing_tagsc C s | j dd d S )Nz
foo &# barzfoo &# bar)r )r
r r r test_empty_element? s z,TestHTMLParserTreeBuilder.test_empty_elementc C sz d}| j |}d|jjks td|jjks.td|jjdjksDt| 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 j }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| |< | | j| d S )N)r listappend)attrskeyvaluer r r
accumulatep s zITestHTMLParserTreeBuilder.test_on_duplicate_attribute..accumulateurl2)r r- r r ZREPLACEZIGNORE)r
r r/ r6 r )r r
r test_on_duplicate_attributeR s
z5TestHTMLParserTreeBuilder.test_on_duplicate_attributec C sn xhd>D ]`\}}}d*| }| j |j}|j }d+|jd, }||ksFt|jd-d.}d+| }||kstqW d S )?N⇄ ⇄ ⇄⊧ ⊧ ⊧𝔑 𝔑 𝔑≧̸ ≧̸ ≧̸¬ ¬ ¬⫬ ⫬ ⫬"" "∴ ∴ ∴ ∴ ∴fjfj fj⊔ ⊔ ⊔⊔︀ ⊔︀ ⊔︀'' '|| |z
%s
s
%s
utf8html) formatterr9 r: r; r< r= r> r? r@ rA rB rC rD rE rF rG rH rI rJ rK rL rM rN rO rP rQ rO rP rR rO rP rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra )re rf rg rh ri rj rk rl rm rn ro rp rq rr rs )r divencoder ) r
Z
input_elementZoutput_unicodeZoutput_elementr rt Zwithout_elementexpectZ with_elementr r r test_html5_attributesv s0 z/TestHTMLParserTreeBuilder.test_html5_attributesN)__name__
__module__ __qualname__r Zdefault_builderr r r r r r r' r8 rw 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