def test_parse_flat(self): d = Document(TEST_DOC_FLAT) target = d.get_section("test_flat.py") self.assertEqual(target, ['print("Hello, world!")'])
def test_parse_nested(self): d = Document(TEST_DOC_NESTED) target = d.get_section("test_nested.py") self.assertEqual(target, ['print("Hello, world!")']) target = d.get_section("test code inner") self.assertEqual(target, ['print("Hello, world!")'])