class EmptyListItemsRemoverTest(unittest.TestCase):
def setUp(self):
self.remover = EmptyListItemsRemover()
def test_basic(self):
text = "<ul><li>first</li><li><!-- comment --></li><li>third</li></ul>"
expected = "<ul><li>first</li><!-- comment --><li>third</li></ul>"
self.assertEqual(expected, self.remover.run(text))
def test_items_with_content_not_removed(self):
text1 = \
"<ul>" + \
"<li>first</li>" + \
"<li><!-- comment -->content</li>" + \
"<li>content<!-- comment --></li>" + \
"<li>fourth</li>" + \
"</ul>"
self.assertEqual(text1, self.remover.run(text1))
text2 = \
"<ul>" + \
"<li>first</li>" + \
"<li>content<!-- comment --></li>" + \
"<li><!-- comment -->content</li>" + \
"<li>fourth</li>" + \
"</ul>"
self.assertEqual(text2, self.remover.run(text2))
def setUp(self): self.remover = EmptyListItemsRemover()