3
d0 @ s| d Z ddlZddlZddlZddlmZ ddlmZmZm Z ddl
m Z G dd de Z G d d
d
e Z
G d d d e ZdS )
z*Tests of the bs4.element.PageElement class N)
BeautifulSoup)Comment ResultSet SoupStrainer )SoupTestc @ 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}| j |}|jjjddjdks*td S )Nu
☃zutf-8u ☃)soupbstringencodeAssertionError)selfhtmlr r D/tmp/pip-build-8z3xcdsh/beautifulsoup4/bs4/tests/test_pageelement.py"test_unicode_string_can_be_encoded s
z/TestEncoding.test_unicode_string_can_be_encodedc C s, d}| j |}|jjd|jdks(td S )Nu
☃zutf-8)r 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 |}|jjddks"td S )Nu
☃asciis ☃)r r
r r
)r r r r r r ☃r strict)errors)r pytestZraisesUnicodeEncodeErrorr )r r r r r r test_encoding_can_be_made_strict" s
z-TestEncoding.test_encoding_can_be_made_strictc C s$ d}| j |}d|jj ks td S )Nu
☃u ☃)r r
Zdecode_contentsr
)r r r r r r test_decode_contents( s
z!TestEncoding.test_decode_contentsc C s. d}| j |}djd|jjddks*td S )Nu
☃u ☃utf8)encoding)r r r
Zencode_contentsr
)r r r r r r test_encode_contents- s
z!TestEncoding.test_encode_contentsc C s* d}| j |}djd|jj ks&td S )Nu
☃u ☃r )r r r
ZrenderContentsr
)r r r r r r test_deprecated_renderContents4 s
z+TestEncoding.test_deprecated_renderContentsc C s" d}| j |}|t|ks td S )Nu
☃)r reprr
)r r r r r r test_repr9 s
zTestEncoding.test_reprN) __name__
__module__ __qualname____doc__r r r r r r r r! r r r r r s r c @ s e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z dd Z
dd Zdd Zdd Zd d Zd d Zd d! Zd"S )#TestFormatterszrTest the formatting feature, used by methods like decode() and
prettify(), and the formatters themselves.
c C s0 d}| j |}|jdd}|| jdks,td S )Nu# <<Sacré bleu!>>minimal) formatter)r decode document_forr
)r markupr decodedr r r !test_default_formatter_is_minimalD s
z0TestFormatters.test_default_formatter_is_minimalc C s0 d}| j |}|jdd}|| jdks,td S )Nu' <<Sacré bleu!>>r )r( z. <<Sacré bleu!>>)r r) r* r
)r r+ r r, r r r test_formatter_htmlM s
z"TestFormatters.test_formatter_htmlc C s0 d}| j |}|jdd}|| jdks,td S )Nu' <<Sacré bleu!>>html5)r( z- <<Sacré bleu!>>)r r) r* r
)r r+ r r, r r r test_formatter_html5U s
z#TestFormatters.test_formatter_html5c C s0 d}| j |}|jdd}|| jdks,td S )Nu# <<Sacré bleu!>>r' )r( )r r) r* r
)r r+ r r, r r r test_formatter_minimal] s
z%TestFormatters.test_formatter_minimalc C s0 d}| j |}|jd d}|| jdks,td S )Nu# <<Sacré bleu!>>)r( u <>)r r) r* r
)r r+ r r, r r r test_formatter_nullf s
z"TestFormatters.test_formatter_nullc C s4 d}| j |}|jdd d}|| jdks0td S )Nz!<foo>bar c S s | j S )N)upper)xr r r s s z6TestFormatters.test_formatter_custom..)r( zBAR )r r) r* r
)r r+ r r, r r r test_formatter_customp s
z$TestFormatters.test_formatter_customc C s d}| j |}|j}d}||j ks(t||jddkseu) er' )r( z/er u% Ec S s | j S )N)r3 )r4 r r r r5 s zJTestFormatters.test_formatter_is_run_on_attribute_values..)r ar) r
)r r+ r r7 Zexpect_minimalZ expect_htmlZ expect_upperr r r )test_formatter_is_run_on_attribute_valuesx s
z8TestFormatters.test_formatter_is_run_on_attribute_valuesc C s" d}t |dj }d|ks td S )NzO
z html.parsers < < hey > >)r r r
)r docencodedr r r 2test_formatter_skips_script_tag_for_html_documents s zATestFormatters.test_formatter_skips_script_tag_for_html_documentsc C s" d}t |dj }d|ks td S )NzF
z html.parsers < < hey > >)r r r
)r r9 r: r r r 1test_formatter_skips_style_tag_for_html_documents s z@TestFormatters.test_formatter_skips_style_tag_for_html_documentsc C s | j d}d|jj ks td S )NzN
foo
bar
baz
zO
foo
bar
baz
)r divprettifyr
)r r r r r ,test_prettify_leaves_preformatted_text_alone s
z;TestFormatters.test_prettify_leaves_preformatted_text_alonec C s* t dd}|jdd d}d|ks&td S )Nz fooz html.parserc S s | j S )N)r3 )r4 r r r r5 s zITestFormatters.test_prettify_accepts_formatter_function..)r( ZFOO)r r> r
)r r prettyr r r (test_prettify_accepts_formatter_function s
z7TestFormatters.test_prettify_accepts_formatter_functionc C s" | j d}tt|j ks td S )Nz)r strtyper> r
)r r r r r (test_prettify_outputs_unicode_by_default s
z7TestFormatters.test_prettify_outputs_unicode_by_defaultc C s$ | j d}tt|jdks td S )Nzzutf-8)r bytesrC r> r
)r r r r r test_prettify_can_encode_data s
z,TestFormatters.test_prettify_can_encode_datac C s0 d}| j |}|jjd}||jdks,td S )Nu Sacré bleu!zutf-8)r r
r r
)r r+ r r: r r r ,test_html_entity_substitution_off_by_default s
z;TestFormatters.test_html_entity_substitution_off_by_defaultc C s d}| j |}|jd dks t|jd}d|ks6t|jd}d|ksLt|jd}d |ksbt|jd
jd
}d |ks~td S ) NzEcontentztext/html; charset=x-sjiszutf-8s
charset=utf-8euc_jps charset=euc_jpz shift-jiss charset=shift-jiszutf-16zcharset=utf-16)r metar
r r) )r Zmeta_tagr utf_8rI shift_jisZutf_16_ur r r test_encoding_substitution s
z)TestFormatters.test_encoding_substitutionc C s2 d}t d}| j||d}|jd jdks.td S )Nz`
foo
pre)Z
parse_onlyr )r r contentsnamer
)r r+ Zstrainerr r r r ;test_encoding_substitution_doesnt_happen_if_tag_is_strained s zJTestFormatters.test_encoding_substitution_doesnt_happen_if_tag_is_strainedN)r" r# r$ r% r- r. r0 r1 r2 r6 r8 r; r< r? rA rD rF rG rM rQ r r r r r& ? s"
r& c @ s` e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z dd Z
dS )TestPersistencez*Testing features like pickle and deepcopy.c C s d| _ | j| j | _d S )Nay
Beautiful Soup: We called him Tortoise because he taught us.foobar
)pager tree)r r r r setup_method s z TestPersistence.setup_methodc C s@ t j| jd}t j|}|jtks&t|j | jj ks z html.parseru
)r Zoriginal_encoding__copy__rB r
)r r r r^ r r r test_copy_preserves_encoding s
z,TestPersistence.test_copy_preserves_encodingc C s | j dj}d|_d|_|j }|j|jks0t|j|jks@t|j|jksPt|j|jks`t|j|jkspt|j |j kstd S )Nz
! )
r pZ
sourcelineZ sourceposrb r
Zcan_be_empty_elementZcdata_list_attributesZpreserve_whitespace_tagsZinteresting_string_types)r tagr` r r r 'test_copy_preserves_builder_information s z7TestPersistence.test_copy_preserves_builder_informationc C s> d}| j |}tj|tj}tj|}|j |j ks:td S )Nu
☃)r rW rX HIGHEST_PROTOCOLrY r) r
)r r r r[ r\ r r r test_unicode_pickle s
z#TestPersistence.test_unicode_picklec C sz d}| j |}|jdd}tj|}||ks0td |jks>td |jksLtd |jksZtd |jkshtd |jksvtd S )NzFooBarZFoo)r )r findr^ r
parent next_elementZ next_siblingprevious_element)r r r s1s2r r r 1test_copy_navigablestring_is_not_attached_to_tree s
zATestPersistence.test_copy_navigablestring_is_not_attached_to_treec C s< d}| j |}|j}tj|}||ks*tt|ts8td S )Nz)r r r^ r
isinstancer )r r r ro rp r r r 0test_copy_navigablestring_subclass_has_same_type) s
z@TestPersistence.test_copy_navigablestring_subclass_has_same_typec C s( d}| j |}tj|}||ks$td S )Nz)
FooBar
end)r r^ r
)r r r Z soup_copyr r r test_copy_entire_soup1 s
z%TestPersistence.test_copy_entire_soupc C s d}| j |}|j}tj|}t|t|ks2t||ks>t||k sJtd |jksXtd |jksftd |jddjks|td |jddjkstd S )Nz)
FooBar
endBar)r ) r r= r^ rB r
rl rn rk rm )r r r r= Zdiv_copyr r r test_copy_tag_copies_contents7 s
z-TestPersistence.test_copy_tag_copies_contentsN)r" r# r$ r% rU r] ra rc rh rj rq rs rt rv r r r r rR s rR )r% r^ rW r Zbs4r Z bs4.elementr r r r r r& rR r r r r s /