def test_doesnt_give_leadin_subsections(self):
wikitext = ("Lead in text\n"
"=== Not a subsection ===\n"
"Some text\n")
wcode = mwpm.parse(wikitext)
top_level = section.generate_sections_from_wikicode(wcode)
self.assertEqual(len(top_level), 2, top_level)
self.assertEqual(len(top_level[0].subsections), 0, top_level[0].subsections)
def test_increasing_section_level(self):
wikitext = ("= Heading 1 =\n"
"Some text\n"
"== Heading 2 ==\n"
"Some text\n"
"=== Heading 3 ===\n"
"Some text\n"
"==== Heading 4 ====\n"
"Some text\n")
wcode = mwpm.parse(wikitext)
top_level = section.generate_sections_from_wikicode(wcode)
self.assertEqual(len(top_level), 1, top_level)
def test_parses_generic_sections(self):
wikitext = ("== Heading 1 ==\n"
"Some text\n"
"=== Heading 1a ===\n"
"Some text\n"
"== Heading 2 ==\n"
"Some text\n"
"==== Heading 2a ====\n"
"Some text\n"
"=== Heading 2b ===\n"
"Some text")
wcode = mwpm.parse(wikitext)
top_level = section.generate_sections_from_wikicode(wcode)
self.assertEqual(len(top_level), 2, str(top_level))
self.assertEqual(len(top_level[0].subsections), 1, top_level[0].subsections)
self.assertEqual(len(top_level[1].subsections), 2, top_level[1].subsections)