DocumentationComment.py: Improve assemble()

assemble() accounts for `marker[1]==marker[2]` Fixes coala#2645
damngamerz · Jun 8, 2017 · c79ec5a · c79ec5a
1 parent e585c07
commit c79ec5a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/coalib/bearlib/languages/documentation/DocumentationComment.py b/coalib/bearlib/languages/documentation/DocumentationComment.py
@@ -221,6 +221,7 @@ def assemble(self):
         assembled += ''.join('\n' if line == '\n' and not self.marker[1]
                              else self.indent + self.marker[1] + line
                              for line in lines[1:])
-        return (assembled +
-                (self.indent if lines[-1][-1] == '\n' else '') +
-                self.marker[2])
+        return (assembled if self.marker[1] == self.marker[2] else
+                (assembled +
+                 (self.indent if lines[-1][-1] == '\n' else '') +
+                 self.marker[2]))
diff --git a/tests/bearlib/languages/documentation/DocumentationCommentTest.py b/tests/bearlib/languages/documentation/DocumentationCommentTest.py
@@ -222,6 +222,13 @@ def test_python_assembly(self):
         for doc in extract_documentation(data, 'python', 'default'):
             self.assertIn(doc.assemble(), docs)
 
+    def test_doxygen_assembly(self):
+        data = load_testdata('doxygen.py')
+        docs = ''.join(data)
+
+        for doc in extract_documentation(data, 'python', 'doxygen'):
+            self.assertIn(doc.assemble(), docs)
+
     def test_c_assembly(self):
         data = load_testdata('default.c')
         docs = ''.join(data)