1 files changed, 118 insertions, 0 deletions
diff --git a/_test/test_copy.py b/_test/test_copy.py
new file mode 100644
index 0000000..37f978d
--- /dev/null
+++ b/_test/test_copy.py
@@ -0,0 +1,118 @@
+# coding: utf-8
+
+"""
+Testing copy and deepcopy, instigated by Issue 84 (Peter Amstutz)
+"""
+
+import copy
+
+import pytest            # NOQA
+import ruamel.yaml       # NOQA
+
+from roundtrip import dedent, round_trip_load, round_trip_dump
+
+
+class TestDeepCopy:
+    def test_preserve_flow_style_simple(self):
+        x = dedent("""\
+        {foo: bar, baz: quux}
+        """)
+        data = round_trip_load(x)
+        data_copy = copy.deepcopy(data)
+        y = round_trip_dump(data_copy)
+        print('x [{}]'.format(x))
+        print('y [{}]'.format(y))
+        assert y == x
+        assert data.fa.flow_style() == data_copy.fa.flow_style()
+
+    def test_deepcopy_flow_style_nested_dict(self):
+        x = dedent("""\
+        a: {foo: bar, baz: quux}
+        """)
+        data = round_trip_load(x)
+        assert data['a'].fa.flow_style() is True
+        data_copy = copy.deepcopy(data)
+        assert data_copy['a'].fa.flow_style() is True
+        data_copy['a'].fa.set_block_style()
+        assert data['a'].fa.flow_style() != data_copy['a'].fa.flow_style()
+        assert data['a'].fa._flow_style is True
+        assert data_copy['a'].fa._flow_style is False
+        y = round_trip_dump(data_copy)
+
+        print('x [{}]'.format(x))
+        print('y [{}]'.format(y))
+        assert y == dedent("""\
+        a:
+          foo: bar
+          baz: quux
+        """)
+
+    def test_deepcopy_flow_style_nested_list(self):
+        x = dedent("""\
+        a: [1, 2, 3]
+        """)
+        data = round_trip_load(x)
+        assert data['a'].fa.flow_style() is True
+        data_copy = copy.deepcopy(data)
+        assert data_copy['a'].fa.flow_style() is True
+        data_copy['a'].fa.set_block_style()
+        assert data['a'].fa.flow_style() != data_copy['a'].fa.flow_style()
+        assert data['a'].fa._flow_style is True
+        assert data_copy['a'].fa._flow_style is False
+        y = round_trip_dump(data_copy)
+
+        print('x [{}]'.format(x))
+        print('y [{}]'.format(y))
+        assert y == dedent("""\
+        a:
+        - 1
+        - 2
+        - 3
+        """)
+
+
+class TestCopy:
+    def test_copy_flow_style_nested_dict(self):
+        x = dedent("""\
+        a: {foo: bar, baz: quux}
+        """)
+        data = round_trip_load(x)
+        assert data['a'].fa.flow_style() is True
+        data_copy = copy.copy(data)
+        assert data_copy['a'].fa.flow_style() is True
+        data_copy['a'].fa.set_block_style()
+        assert data['a'].fa.flow_style() == data_copy['a'].fa.flow_style()
+        assert data['a'].fa._flow_style is False
+        assert data_copy['a'].fa._flow_style is False
+        y = round_trip_dump(data_copy)
+        z = round_trip_dump(data)
+        assert y == z
+
+        assert y == dedent("""\
+        a:
+          foo: bar
+          baz: quux
+        """)
+
+    def test_copy_flow_style_nested_list(self):
+        x = dedent("""\
+        a: [1, 2, 3]
+        """)
+        data = round_trip_load(x)
+        assert data['a'].fa.flow_style() is True
+        data_copy = copy.copy(data)
+        assert data_copy['a'].fa.flow_style() is True
+        data_copy['a'].fa.set_block_style()
+        assert data['a'].fa.flow_style() == data_copy['a'].fa.flow_style()
+        assert data['a'].fa._flow_style is False
+        assert data_copy['a'].fa._flow_style is False
+        y = round_trip_dump(data_copy)
+
+        print('x [{}]'.format(x))
+        print('y [{}]'.format(y))
+        assert y == dedent("""\
+        a:
+        - 1
+        - 2
+        - 3
+        """)