diff options
Diffstat (limited to 'yarnlib/mdparser_tests.py')
-rw-r--r-- | yarnlib/mdparser_tests.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/yarnlib/mdparser_tests.py b/yarnlib/mdparser_tests.py index a9b9cb7..099ff2b 100644 --- a/yarnlib/mdparser_tests.py +++ b/yarnlib/mdparser_tests.py @@ -27,7 +27,7 @@ class MarkdownParserTests(unittest.TestCase): self.parser = yarnlib.MarkdownParser() def test_finds_code_block(self): - self.parser.parse_string(''' + result = self.parser.parse_string(''' This is blah blah text. this is a code block @@ -35,6 +35,7 @@ This is blah blah text. More text. ''') self.assertEqual(self.parser.blocks, ['this is a code block\n']) + self.assertEqual(result, ['this is a code block\n']) def test_finds_consecutive_code_blocks_as_one(self): self.parser.parse_string(''' @@ -83,11 +84,13 @@ More text. self.assertEqual(self.parser.blocks, ['this is a code block\n']) def test_parses_multiple_files(self): - self.parser.parse_string(''' + result1 = self.parser.parse_string(''' block 1 ''') - self.parser.parse_string(''' + result2 = self.parser.parse_string(''' block 2 ''') + self.assertEqual(result1, ['block 1\n']) + self.assertEqual(result2, ['block 2\n']) self.assertEqual(self.parser.blocks, ['block 1\n', 'block 2\n']) |