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")