added top_level_colon_align/prefix_colon0.11.9

1 files changed, 13 insertions, 8 deletions

diff --git a/_test/roundtrip.py b/_test/roundtrip.py
index 4534a16..b9cc657 100644
--- a/_test/roundtrip.py
+++ b/_test/roundtrip.py
@@ -27,15 +27,16 @@ def round_trip_load(inp):
     return ruamel.yaml.load(dinp, ruamel.yaml.RoundTripLoader)
 
 
-def round_trip_dump(data, indent=None, block_seq_indent=None):
-    dumper = ruamel.yaml.RoundTripDumper
-    return ruamel.yaml.dump(data, default_flow_style=False, Dumper=dumper,
-                            allow_unicode=True,
-                            indent=indent, block_seq_indent=block_seq_indent)
+def round_trip_dump(data, indent=None, block_seq_indent=None, top_level_colon_align=None,
+                    prefix_colon=None):
+    return ruamel.yaml.round_trip_dump(data,
+                            indent=indent, block_seq_indent=block_seq_indent,
+                            top_level_colon_align=top_level_colon_align,
+                            prefix_colon=prefix_colon)
 
 
 def round_trip(inp, outp=None, extra=None, intermediate=None, indent=None,
-               block_seq_indent=None):
+               block_seq_indent=None, top_level_colon_align=None, prefix_colon=None):
     if outp is None:
         outp = inp
     doutp = dedent(outp)
@@ -48,9 +49,13 @@ def round_trip(inp, outp=None, extra=None, intermediate=None, indent=None,
                 if data[k] != v:
                     print('{0!r} <> {1!r}'.format(data[k], v))
                     raise ValueError
-    res = round_trip_dump(data, indent=indent, block_seq_indent=block_seq_indent)
+    res = round_trip_dump(data, indent=indent, block_seq_indent=block_seq_indent,
+                          top_level_colon_align=top_level_colon_align,
+                          prefix_colon=prefix_colon)
     print('roundtrip data:\n', res, sep='')
     assert res == doutp
-    res = round_trip_dump(data, indent=indent, block_seq_indent=block_seq_indent)
+    res = round_trip_dump(data, indent=indent, block_seq_indent=block_seq_indent,
+                          top_level_colon_align=top_level_colon_align,
+                          prefix_colon=prefix_colon)
     print('roundtrip second round data:\n', res, sep='')
     assert res == doutp