U
lu
f7 @ s d Z ddlZddlZddlZddlZddlmZ ddlmZm Z m
Z
ddl m Z G dd de Z
G d d
d
e ZG d d d e ZdS )
z*Tests of the bs4.element.PageElement class N)
BeautifulSoup)Comment ResultSet SoupStrainer )SoupTestc @ sX e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z dS ) TestEncodingz0Test the ability to encode objects into strings.c C s. d}| |}|jjdddks*td S )N
☃utf-8 ☃)soupbstringencodeAssertionErrorselfhtmlr r >/tmp/pip-unpacked-wheel-pg4vfqbd/bs4/tests/test_pageelement.py"test_unicode_string_can_be_encoded s
z/TestEncoding.test_unicode_string_can_be_encodedc C s, d}| |}|jd|dks(td 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}| |}|jddks"td S )Nr asciis ☃r r r r r s )sysgetrecursionlimitr r countr )r limitmarkupr encodedr r r "test_encode_deeply_nested_document5 s
z/TestEncoding.test_encode_deeply_nested_documentc C s2 d}| |}| dd|j ks.td S )Nr r r! )r ZrenderContentsr r
r r r r r test_deprecated_renderContents? s
z+TestEncoding.test_deprecated_renderContentsc C s" d}| |}|t|ks td S Nr )r reprr r r r r test_reprE s
zTestEncoding.test_reprN)
__name__
__module__ __qualname____doc__r 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"d# Zd$S )%TestFormatterszrTest the formatting feature, used by methods like decode() and
prettify(), and the formatters themselves.
c C s0 d}| |}|jdd}|| dks,td S N# <<Sacré bleu!>>minimal formatterr decodeZ document_forr r r) r decodedr r r !test_default_formatter_is_minimalP s
z0TestFormatters.test_default_formatter_is_minimalc C s0 d}| |}|jdd}|| dks,td S )N' <<Sacré bleu!>>r r8 z. <<Sacré bleu!>>r: r< r r r test_formatter_htmlY s
z"TestFormatters.test_formatter_htmlc C s0 d}| |}|jdd}|| dks,td S )Nr? html5r8 z- <<Sacré bleu!>>r: r< r r r test_formatter_html5a s
z#TestFormatters.test_formatter_html5c C s0 d}| |}|jdd}|| dks,td S r5 r: r< r r r test_formatter_minimali s
z%TestFormatters.test_formatter_minimalc C s0 d}| |}|jd d}|| dks,td S )Nr6 r8 u <>r: r< r r r test_formatter_nullr s
z"TestFormatters.test_formatter_nullc C s4 d}| |}|jdd d}|| dks0td S )Nz!<foo>bar c S s | S Nupperxr r r z6TestFormatters.test_formatter_custom..r8 zBAR r: r< r r r test_formatter_custom| s
z$TestFormatters.test_formatter_customc C s d}| |}|j}d}|| ks(t||jddkseu) er7 r8 z/er u% Ec S s | S rE rF rH r r r rJ rK zJTestFormatters.test_formatter_is_run_on_attribute_values..)r ar; r )r r) r rM Zexpect_minimalZ expect_htmlZ expect_upperr r r )test_formatter_is_run_on_attribute_values s
z8TestFormatters.test_formatter_is_run_on_attribute_valuesc C s" d}t |d }d|ks td S )NzO
html.parser < < hey > >r r r r docr* r r r 2test_formatter_skips_script_tag_for_html_documents s zATestFormatters.test_formatter_skips_script_tag_for_html_documentsc C s" d}t |d }d|ks td S )NzF
rO rP rQ rR r r r 1test_formatter_skips_style_tag_for_html_documents s z@TestFormatters.test_formatter_skips_style_tag_for_html_documentsc C s | d}d|j ks td S )NzN
foo
bar
baz
zP
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( d}d}| |}||j ks$td S )NzR
some
for you
zV
some
for you
rV )r r) expectr r r r 0test_prettify_handles_nested_string_literal_tags s
z?TestFormatters.test_prettify_handles_nested_string_literal_tagsc C s* t dd}|jdd d}d|ks&td S )Nz foorO c S s | S rE rF rH r r r rJ rK zITestFormatters.test_prettify_accepts_formatter_function..r8 ZFOO)r rX r )r r prettyr r r (test_prettify_accepts_formatter_function s
z7TestFormatters.test_prettify_accepts_formatter_functionc C s" | d}tt| ks td S )N)r strtyperX r rY r r r (test_prettify_outputs_unicode_by_default s
z7TestFormatters.test_prettify_outputs_unicode_by_defaultc C s$ | d}tt|dks td S )Nr_ r
)r bytesra rX r rY r r r test_prettify_can_encode_data s
z,TestFormatters.test_prettify_can_encode_datac C s0 d}| |}|jd}||dks,td S )Nu Sacré bleu!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}| |}|jd dks t|d}d|ks6t|d}d|ksLt|d}d |ksbt|d
d
}d |ks~td S ) NzEcontentztext/html; charset=x-sjisr
s
charset=utf-8euc_jps charset=euc_jpz shift-jiss charset=shift-jiszutf-16zcharset=utf-16)r metar r r; )r Zmeta_tagr utf_8rg 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)r0 r1 r2 r3 r> r@ rB rC rD rL rN rT rU rZ r\ r^ rb rd re rk ro r r r r r4 K s$
r4 c @ sh e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z dd Z
dd ZdS )TestPersistencez*Testing features like pickle and deepcopy.c C s d| _ | | 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 | jd}t |}|jtks&t| | j ks rO u
)r Zoriginal_encoding__copy__r` r )r r r" r| r r r test_copy_preserves_encoding+ s
z,TestPersistence.test_copy_preserves_encodingc C s | dj}d|_d|_| }|j|jks0t|j|jks@t|j|jksPt|j|jks`t|j|jkspt|j |j kstd S )Nz
! )
r pZ
sourcelineZ sourceposr r Zcan_be_empty_elementZcdata_list_attributesZpreserve_whitespace_tagsZinteresting_string_types)r tagr~ r r r 'test_copy_preserves_builder_information2 s z7TestPersistence.test_copy_preserves_builder_informationc C s> d}| |}t|tj}t|}| | ks:td S r- )r ru rv HIGHEST_PROTOCOLrw r; r )r r r ry rz r r r test_unicode_pickleE s
z#TestPersistence.test_unicode_picklec C sz d}| |}|jdd}t|}||ks0td |jks>td |jksLtd |jksZtd |jkshtd |jksvtd S )NzFooBarZFoor )r findr| r parent next_elementZ next_siblingprevious_elementr r r s1s2r r r 1test_copy_navigablestring_is_not_attached_to_treeM s
zATestPersistence.test_copy_navigablestring_is_not_attached_to_treec C s< d}| |}|j}t|}||ks*tt|ts8td S )Nz)r r r| r
isinstancer r r r r 0test_copy_navigablestring_subclass_has_same_typeY s
z@TestPersistence.test_copy_navigablestring_subclass_has_same_typec C s( d}| |}t|}||ks$td S )N)
FooBar
end)r r| r )r r r Z soup_copyr r r test_copy_entire_soupa s
z%TestPersistence.test_copy_entire_soupc C s d}| |}|j}t|}t|t|ks2t||ks>t||k sJtd |jksXtd |jksftd |jddjks|td |jddjkstd S )Nr Barr ) r rW r| r` r r r r r )r r r rW Zdiv_copyr r r test_copy_tag_copies_contentsg s
z-TestPersistence.test_copy_tag_copies_contentsN)r0 r1 r2 r3 rs r{ r r r r r r r r r r r r r rp s rp )r3 r| ru r r% Zbs4r Z bs4.elementr r r r r r4 rp r r r r s : 2