md_tw.py: Add TWMarkdown.output_heading.

Add prefix, if it exists, to heading.

* md_tw.py (TWMarkdown.output_heading): Override super class's method.
* tests/data/renderer-heading-w.md: Update file.
* tests/data/renderer-heading.md: Update file.

3 files changed, 16 insertions, 0 deletions

diff --git a/md_tw.py b/md_tw.py
index b61c409..866f1fa 100644
--- a/md_tw.py
+++ b/md_tw.py
@@ -356,6 +356,14 @@ class TWMarkdown(mistune.Markdown):
         if subseq:
             self.renderer.tw_set(subsequent_indent=p)
 
+    def output_heading(self):
+        rendered_heading = '{}{}'.format(
+            self.renderer.tw_get('initial_indent'),
+            super(TWMarkdown, self).output_heading()
+        )
+
+        return rendered_heading
+
     def output_block_quote(self):
         # Add prefix
         self._add_prefix('> ')
diff --git a/tests/data/renderer-heading-w.md b/tests/data/renderer-heading-w.md
index 382896d..3762596 100644
--- a/tests/data/renderer-heading-w.md
+++ b/tests/data/renderer-heading-w.md
@@ -9,3 +9,7 @@ Lost our mind.
 ### Junk Mind
 
 In the junkyard.
+
+> ### Depreston
+>
+> ...now we got the perculator...
diff --git a/tests/data/renderer-heading.md b/tests/data/renderer-heading.md
index 382896d..3762596 100644
--- a/tests/data/renderer-heading.md
+++ b/tests/data/renderer-heading.md
@@ -9,3 +9,7 @@ Lost our mind.
 ### Junk Mind
 
 In the junkyard.
+
+> ### Depreston
+>
+> ...now we got the perculator...