1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
# coding: utf-8
from __future__ import absolute_import
from __future__ import print_function
from __future__ import unicode_literals
from textwrap import dedent
import pytest # NOQA
import ruamel.yaml
def rt(s):
return ruamel.yaml.dump(
ruamel.yaml.load(s, Loader=ruamel.yaml.RoundTripLoader),
Dumper=ruamel.yaml.RoundTripDumper,
).strip() + '\n'
class TestIndent:
def test_roundtrip_inline_list(self):
s = 'a: [a, b, c]\n'
output = rt(s)
assert s == output
def test_roundtrip_mapping_of_inline_lists(self):
s = dedent("""\
a: [a, b, c]
j: [k, l, m]
""")
output = rt(s)
assert s == output
def test_roundtrip_mapping_of_inline_lists_comments(self):
s = dedent("""\
# comment A
a: [a, b, c]
# comment B
j: [k, l, m]
""")
output = rt(s)
assert s == output
def test_roundtrip_mapping_of_inline_sequence_eol_comments(self):
s = dedent("""\
# comment A
a: [a, b, c] # comment B
j: [k, l, m] # comment C
""")
output = rt(s)
assert s == output
# first test by explicitly setting flow style
def test_added_inline_list(self):
s1 = dedent("""
a:
- b
- c
- d
""")
s = 'a: [b, c, d]\n'
data = ruamel.yaml.load(s1, Loader=ruamel.yaml.RoundTripLoader)
val = data['a']
val.fa.set_flow_style()
# print(type(val), '_yaml_format' in dir(val))
output = ruamel.yaml.dump(data, Dumper=ruamel.yaml.RoundTripDumper)
assert s == output
# ############ flow mappings
def test_roundtrip_flow_mapping(self):
s = dedent("""\
- {a: 1, b: hallo}
- {j: fka, k: 42}
""")
data = ruamel.yaml.load(s, Loader=ruamel.yaml.RoundTripLoader)
output = ruamel.yaml.dump(data, Dumper=ruamel.yaml.RoundTripDumper)
assert s == output
def test_roundtrip_sequence_of_inline_mappings_eol_comments(self):
s = dedent("""\
# comment A
- {a: 1, b: hallo} # comment B
- {j: fka, k: 42} # comment C
""")
output = rt(s)
assert s == output
# ############ indentation
|
