diff options
Diffstat (limited to '_test/test_copy.py')
-rw-r--r-- | _test/test_copy.py | 118 |
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 + """) |