Preserve blank lines after comments

2 files changed, 9 insertions, 4 deletions

diff --git a/py/scanner.py b/py/scanner.py
index 1c011e0..1286e4f 100644
--- a/py/scanner.py
+++ b/py/scanner.py
@@ -1621,6 +1621,11 @@ class RoundTripScanner(Scanner):
                         break
                     comment += ch
                     self.forward()
+                # gather any blank lines following the comment too
+                ch = self.scan_line_break()
+                while len(ch) > 0:
+                    comment += ch
+                    ch = self.scan_line_break()
                 end_mark = self.get_mark()
                 if not self.flow_level:
                     self.allow_simple_key = True
diff --git a/test/test_yaml.py b/test/test_yaml.py
index 3815b08..c5d69b4 100644
--- a/test/test_yaml.py
+++ b/test/test_yaml.py
@@ -72,13 +72,13 @@ class TestYAML:
         ? c
         """)
 
-    def test_blank_line(self):
+    def test_blank_lines(self):
         round_trip("""
-        # foo
+        # Comment with a space after it.
 
-        a: foo
+        a: 1
 
-        # bar
+        # Seq with spaces in between items.
         b:
         - bar