def test_get_next_paragraph(self):
# given
doc_render = DocRender([
'This is a sample docstring.',
'and the paragraph continues here',
'',
'',
'This is another sample'])
# when/then
self.assertEqual(
doc_render.get_next_paragraph(),
['This is a sample docstring.',
'and the paragraph continues here'])
self.assertEqual(
doc_render.docstring, ['', '', 'This is another sample'])
# when/then
self.assertEqual(doc_render.get_next_paragraph(), [])
self.assertEqual(
doc_render.docstring, ['', '', 'This is another sample'])
# when/then
doc_render.index = 2
self.assertEqual(
doc_render.get_next_paragraph(),
['This is another sample'])
self.assertEqual(doc_render.docstring, ['', ''])
# when/then
self.assertEqual(
doc_render.get_next_paragraph(), [])
def test_get_next_block(self):
doc_render = DocRender([
'term1',
' Definition1',
'term2 : classifier',
' Definition2',
' ',
'term3',
'',
'',
'term4 : classifier',
' Definition3',
'',
' MoreDefinition3',
''])
self.assertEqual(
doc_render.get_next_block(),
['term1', ' Definition1'])
self.assertEqual(
doc_render.get_next_block(),
['term2 : classifier', ' Definition2'])
self.assertEqual(
doc_render.get_next_block(), ['term3'])
doc_render.index = 2
self.assertEqual(
doc_render.get_next_block(),
['term4 : classifier',
' Definition3', '', ' MoreDefinition3'])
def test_seek_to_next_empty_line(self):
# given
doc_render = DocRender([
'This is a sample docstring.',
'and the paragraph continues here',
'',
'',
'This is another sample'])
# when
doc_render.seek_to_next_non_empty_line()
# then
self.assertEqual(doc_render.index, 0)
# when
doc_render.index = 2
doc_render.seek_to_next_non_empty_line()
# then
self.assertEqual(doc_render.index, 4)
# when
doc_render.seek_to_next_non_empty_line()
# then
self.assertEqual(doc_render.index, 4)
def test_extract_item_blocks(self):
# given
doc_render = DocRender([
'Section',
'=======',
'',
'Section',
'=======',
'term1',
' Definition1',
'term2 : classifier',
' Definition2',
' ',
'Section',
'=======',
'term4 : classifier',
' Definition3',
'',
' MoreDefinition3',
''])
# when/then
self.assertEqual(doc_render.extract_items(), [])
# when/then
doc_render.index = 5
self.assertEqual(
doc_render.extract_items(),
[('term1', [], ['Definition1']),
('term2', ['classifier'], ['Definition2'])])
# when/then
doc_render.index = 7
self.assertEqual(
doc_render.extract_items(),
[('term4', ['classifier'],
['Definition3', '', 'MoreDefinition3'])])
def test_is_section(self):
# given
doc_render = DocRender([
"My Header ",
"--------- ",
"MyHeader ",
"--------- ",
"No header",
"Input\Output header",
"===================",
"My Header ",
" -------- "])
# when/then
doc_render.index = 0
self.assertTrue(doc_render.is_section())
# when/then
doc_render.index = 1
self.assertFalse(doc_render.is_section())
# when/then
doc_render.index = 2
self.assertFalse(doc_render.is_section())
# when/then
doc_render.index = 4
self.assertFalse(doc_render.is_section())
# when/then
doc_render.index = 5
self.assertTrue(doc_render.is_section())
# when/then
doc_render.index = 7
self.assertFalse(doc_render.is_section())
# when/then
doc_render.index = 8
self.assertFalse(doc_render.is_section())
