fix issue 54: space on empty lines break roundtrip0.12.7

1 files changed, 22 insertions, 0 deletions

diff --git a/_test/test_comments.py b/_test/test_comments.py
index 4bdaed1..63895cc 100644
--- a/_test/test_comments.py
+++ b/_test/test_comments.py
@@ -569,3 +569,25 @@ class TestEmptyLines:
 
           - 3
         """)
+
+    def test_line_with_only_spaces(self):
+        # issue 54
+        yaml_str = "---\n\na: 'x'\n \nb: y\n"
+        d = round_trip_load(yaml_str, preserve_quotes=True)
+        y = round_trip_dump(d, explicit_start=True)
+        stripped = ""
+        for line in yaml_str.splitlines():
+            stripped += line.rstrip() + '\n'
+            print(line + '$')
+        assert stripped == y
+
+    def test_some_eol_spaces(self):
+        # spaces after tokens and on empty lines
+        yaml_str = '---  \n  \na: "x"  \n   \nb: y  \n'
+        d = round_trip_load(yaml_str, preserve_quotes=True)
+        y = round_trip_dump(d, explicit_start=True)
+        stripped = ""
+        for line in yaml_str.splitlines():
+            stripped += line.rstrip() + '\n'
+            print(line + '$')
+        assert stripped == y