From f59bfb6b8cebf98c7b25e95b7b498949f5462222 Mon Sep 17 00:00:00 2001 From: Sebastian Huber Date: Fri, 24 Jul 2020 16:05:49 +0200 Subject: content: Add Content.ensure_blank_line() --- rtemsspec/content.py | 5 +++++ rtemsspec/tests/test_content.py | 8 ++++++++ 2 files changed, 13 insertions(+) diff --git a/rtemsspec/content.py b/rtemsspec/content.py index 5e8da8b6..cfef89c6 100644 --- a/rtemsspec/content.py +++ b/rtemsspec/content.py @@ -386,6 +386,11 @@ class Content: """ Adds a blank line. """ self._lines.append("") + def ensure_blank_line(self): + """ Ensures that the last line is blank. """ + if not self._lines or self._lines[-1]: + self._lines.append("") + def register_license(self, the_license: str) -> None: """ Registers a licence for the content. """ licenses = re.split(r"\s+OR\s+", the_license) diff --git a/rtemsspec/tests/test_content.py b/rtemsspec/tests/test_content.py index de30276f..8383ca8e 100644 --- a/rtemsspec/tests/test_content.py +++ b/rtemsspec/tests/test_content.py @@ -178,6 +178,14 @@ def test_add_blank_line(): """ +def test_ensure_blank_line(): + content = Content("BSD-2-Clause", True) + content.ensure_blank_line() + content.ensure_blank_line() + assert str(content) == """ +""" + + def test_indent(): content = Content("BSD-2-Clause", True) content.add_blank_line() -- cgit v1.2.3