1 files changed, 229 insertions, 0 deletions
diff --git a/_test/test_yamlfile.py b/_test/test_yamlfile.py
new file mode 100644
index 0000000..149478a
--- /dev/null
+++ b/_test/test_yamlfile.py
@@ -0,0 +1,229 @@
+# coding: utf-8
+
+"""
+various test cases for YAML files
+"""
+
+import io
+import platform
+
+import pytest  # NOQA
+
+from .roundtrip import dedent, round_trip, round_trip_dump, round_trip_load  # NOQA
+
+
+class TestYAML:
+    def test_backslash(self):
+        round_trip(
+            """
+        handlers:
+          static_files: applications/\\1/static/\\2
+        """
+        )
+
+    def test_omap_out(self):
+        # ordereddict mapped to !!omap
+        import ruyaml  # NOQA
+        from ruyaml.compat import ordereddict
+
+        x = ordereddict([('a', 1), ('b', 2)])
+        res = round_trip_dump(x, default_flow_style=False)
+        assert res == dedent(
+            """
+        !!omap
+        - a: 1
+        - b: 2
+        """
+        )
+
+    def test_omap_roundtrip(self):
+        round_trip(
+            """
+        !!omap
+        - a: 1
+        - b: 2
+        - c: 3
+        - d: 4
+        """
+        )
+
+    def test_dump_collections_ordereddict(self):
+        from collections import OrderedDict
+
+        import ruyaml  # NOQA
+
+        # OrderedDict mapped to !!omap
+        x = OrderedDict([('a', 1), ('b', 2)])
+        res = round_trip_dump(x, default_flow_style=False)
+        assert res == dedent(
+            """
+        !!omap
+        - a: 1
+        - b: 2
+        """
+        )
+
+    def test_CommentedSet(self):
+        from ruyaml.constructor import CommentedSet
+
+        s = CommentedSet(['a', 'b', 'c'])
+        s.remove('b')
+        s.add('d')
+        assert s == CommentedSet(['a', 'c', 'd'])
+        s.add('e')
+        s.add('f')
+        s.remove('e')
+        assert s == CommentedSet(['a', 'c', 'd', 'f'])
+
+    def test_set_out(self):
+        # preferable would be the shorter format without the ': null'
+        import ruyaml  # NOQA
+
+        x = set(['a', 'b', 'c'])
+        # cannot use round_trip_dump, it doesn't show null in block style
+        buf = io.StringIO()
+        yaml = ruyaml.YAML(typ='unsafe', pure=True)
+        yaml.default_flow_style = False
+        yaml.dump(x, buf)
+        assert buf.getvalue() == dedent(
+            """
+        !!set
+        a: null
+        b: null
+        c: null
+        """
+        )
+
+    # ordering is not preserved in a set
+    def test_set_compact(self):
+        # this format is read and also should be written by default
+        round_trip(
+            """
+        !!set
+        ? a
+        ? b
+        ? c
+        """
+        )
+
+    def test_blank_line_after_comment(self):
+        round_trip(
+            """
+        # Comment with spaces after it.
+
+
+        a: 1
+        """
+        )
+
+    def test_blank_line_between_seq_items(self):
+        round_trip(
+            """
+        # Seq with empty lines in between items.
+        b:
+        - bar
+
+
+        - baz
+        """
+        )
+
+    @pytest.mark.skipif(
+        platform.python_implementation() == 'Jython',
+        reason='Jython throws RepresenterError',
+    )
+    def test_blank_line_after_literal_chip(self):
+        s = """
+        c:
+        - |
+          This item
+          has a blank line
+          following it.
+
+        - |
+          To visually separate it from this item.
+
+          This item contains a blank line.
+
+
+        """
+        d = round_trip_load(dedent(s))
+        print(d)
+        round_trip(s)
+        assert d['c'][0].split('it.')[1] == '\n'
+        assert d['c'][1].split('line.')[1] == '\n'
+
+    @pytest.mark.skipif(
+        platform.python_implementation() == 'Jython',
+        reason='Jython throws RepresenterError',
+    )
+    def test_blank_line_after_literal_keep(self):
+        """have to insert an eof marker in YAML to test this"""
+        s = """
+        c:
+        - |+
+          This item
+          has a blank line
+          following it.
+
+        - |+
+          To visually separate it from this item.
+
+          This item contains a blank line.
+
+
+        ...
+        """
+        d = round_trip_load(dedent(s))
+        print(d)
+        round_trip(s)
+        assert d['c'][0].split('it.')[1] == '\n\n'
+        assert d['c'][1].split('line.')[1] == '\n\n\n'
+
+    @pytest.mark.skipif(
+        platform.python_implementation() == 'Jython',
+        reason='Jython throws RepresenterError',
+    )
+    def test_blank_line_after_literal_strip(self):
+        s = """
+        c:
+        - |-
+          This item
+          has a blank line
+          following it.
+
+        - |-
+          To visually separate it from this item.
+
+          This item contains a blank line.
+
+
+        """
+        d = round_trip_load(dedent(s))
+        print(d)
+        round_trip(s)
+        assert d['c'][0].split('it.')[1] == ""
+        assert d['c'][1].split('line.')[1] == ""
+
+    def test_load_all_perserve_quotes(self):
+        import ruyaml  # NOQA
+
+        yaml = ruyaml.YAML()
+        yaml.preserve_quotes = True
+        s = dedent(
+            """\
+        a: 'hello'
+        ---
+        b: "goodbye"
+        """
+        )
+        data = []
+        for x in yaml.load_all(s):
+            data.append(x)
+        buf = ruyaml.compat.StringIO()
+        yaml.dump_all(data, buf)
+        out = buf.getvalue()
+        print(type(data[0]['a']), data[0]['a'])
+        # out = ruyaml.round_trip_dump_all(data)
+        print(out)
+        assert out == s