def test_get_header_none_if_no_header(self):
code_node = get_code_node(
"<html>",
" <body>",
" <code>This is code</code>",
" </body>",
"</html>",
)
text = extract_header_above(code_node)
self.assertIsNone(text)
def test_get_header_type_h2_above(self):
code_node = get_code_node(
"<html>",
" <body>",
" <h2>Header</h2>",
" <code>This is code</code>",
" </body>",
"</html>",
)
text = extract_header_above(code_node)
self.assertEqual(text, "Header")
def test_get_header_several_siblings_above(self):
code_node = get_code_node(
"<html>",
" <body>",
" <h2>Header</h2>",
" <p>Text between</p>",
" <code>This is code</code>",
" </body>",
"</html>",
)
text = extract_header_above(code_node)
self.assertEqual(text, "Header")
