Add more tests

1 files changed, 46 insertions, 0 deletions

diff --git a/yarnlib/mdparser_tests.py b/yarnlib/mdparser_tests.py
index ad59f50..af0c7af 100644
--- a/yarnlib/mdparser_tests.py
+++ b/yarnlib/mdparser_tests.py
@@ -36,3 +36,49 @@ More text.
 ''')
         self.assertEqual(self.parser.blocks, ['this is a code block\n'])
 
+    def test_finds_consecutive_code_blocks_as_one(self):
+        self.parser.parse_string('''
+This is blah blah text.
+
+    this is a code block
+
+    this is a second code block
+
+More text.
+''')
+        self.assertEqual(
+            self.parser.blocks,
+            ['this is a code block\n\nthis is a second code block\n'])
+
+    def test_finds_code_blocks_with_text_in_between_as_two_blocks(self):
+        self.parser.parse_string('''
+This is blah blah text.
+
+    this is a code block
+
+Blah.
+
+    this is a second code block
+
+More text.
+''')
+        self.assertEqual(
+            self.parser.blocks,
+            ['this is a code block\n', 'this is a second code block\n'])
+
+    def test_only_finds_top_level_code_blocks(self):
+        self.parser.parse_string('''
+This is blah blah text.
+
+    this is a code block
+
+And now a list:
+
+*   list item
+
+        this is a second level code block
+
+More text.
+''')
+        self.assertEqual(self.parser.blocks, ['this is a code block\n'])
+