fix #32

pycontribs · Jan 2, 2021 · 33e86bb · 33e86bb
1 parent f41f182
commit 33e86bb
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/_test/test_comment_manipulation.py b/_test/test_comment_manipulation.py
@@ -638,7 +638,6 @@ def test_map_set_comment_before_and_after_non_first_key_02(self):
         compare(data, exp)
 
     # issue 32
-    @pytest.mark.xfail(reason="open issue", raises=TypeError)
     def test_yaml_add_eol_comment_issue_32(self):
         data = load(
             """

diff --git a/lib/ruyaml/comments.py b/lib/ruyaml/comments.py
@@ -426,7 +426,7 @@ def __eq__(self, other):
     def _yaml_add_comment(self, comment, key=NoComment):
         # type: (Any, Optional[Any]) -> None
         if key is not NoComment:
-            self.yaml_key_comment_extend(key, comment)
+            self.yaml_key_comment_extend(key, comment, clear=True)
         else:
             self.ca.comment = comment