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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
|
# coding: utf-8
from __future__ import print_function, unicode_literals
import pytest # NOQA
import warnings # NOQA
from ruamel.std.pathlib import Path
base_path = Path('data') # that is ruamel.yaml.data
class YAMLData(object):
yaml_tag = '!YAML'
def __init__(self, s):
self._s = s
# Conversion tables for input. E.g. "<TAB>" is replaced by "\t"
# fmt: off
special = {
'SPC': ' ',
'TAB': '\t',
'---': '---',
'...': '...',
}
# fmt: on
@property
def value(self):
if hasattr(self, '_p'):
return self._p
assert ' \n' not in self._s
assert '\t\n' not in self._s
self._p = self._s
for k, v in YAMLData.special.items():
k = '<' + k + '>'
self._p = self._p.replace(k, v)
return self._p
def test_rewrite(self, s):
assert ' \n' not in s
assert '\t\n' not in s
for k, v in YAMLData.special.items():
k = '<' + k + '>'
s = s.replace(k, v)
return s
@classmethod
def from_yaml(cls, constructor, node):
from ruamel.yaml.nodes import MappingNode
if isinstance(node, MappingNode):
return cls(constructor.construct_mapping(node))
return cls(node.value)
class Python(YAMLData):
yaml_tag = '!Python'
class Output(YAMLData):
yaml_tag = '!Output'
class Assert(YAMLData):
yaml_tag = '!Assert'
@property
def value(self):
from ruamel.yaml.compat import Mapping
if hasattr(self, '_pa'):
return self._pa
if isinstance(self._s, Mapping):
self._s['lines'] = self.test_rewrite(self._s['lines'])
self._pa = self._s
return self._pa
def pytest_generate_tests(metafunc):
test_yaml = []
paths = sorted(base_path.glob('**/*.yaml'))
idlist = []
for path in paths:
idlist.append(path.stem)
test_yaml.append([path])
metafunc.parametrize(['yaml'], test_yaml, ids=idlist, scope='class')
class TestYAMLData(object):
def yaml(self, yaml_version=None):
from ruamel.yaml import YAML
y = YAML()
y.preserve_quotes = True
if yaml_version:
y.version = yaml_version
return y
def docs(self, path):
from ruamel.yaml import YAML
tyaml = YAML(typ='safe', pure=True)
tyaml.register_class(YAMLData)
tyaml.register_class(Python)
tyaml.register_class(Output)
tyaml.register_class(Assert)
return list(tyaml.load_all(path))
def yaml_load(self, value, yaml_version=None):
yaml = self.yaml(yaml_version=yaml_version)
data = yaml.load(value)
return yaml, data
def round_trip(self, input, output=None, yaml_version=None):
from ruamel.yaml.compat import StringIO
yaml, data = self.yaml_load(input.value, yaml_version=yaml_version)
buf = StringIO()
yaml.dump(data, buf)
expected = input.value if output is None else output.value
assert buf.getvalue() == expected
def load_assert(self, input, confirm, yaml_version=None):
from ruamel.yaml.compat import Mapping
d = self.yaml_load(input.value, yaml_version=yaml_version)[1] # NOQA
print('confirm.value', confirm.value, type(confirm.value))
if isinstance(confirm.value, Mapping):
r = range(confirm.value['range'])
lines = confirm.value['lines'].splitlines()
for idx in r: # NOQA
for line in lines:
line = 'assert ' + line
print(line)
exec(line)
else:
for line in confirm.value.splitlines():
line = 'assert ' + line
print(line)
exec(line)
def run_python(self, python, data, tmpdir):
from roundtrip import save_and_run
assert save_and_run(python.value, base_dir=tmpdir, output=data.value) == 0
# this is executed by pytest the methods with names not starting with test_
# are helpers
def test_yaml_data(self, yaml, tmpdir):
from ruamel.yaml.compat import Mapping
idx = 0
typ = None
yaml_version = None
docs = self.docs(yaml)
if isinstance(docs[0], Mapping):
d = docs[0]
typ = d.get('type')
yaml_version = d.get('yaml_version')
idx += 1
data = output = confirm = python = None
for doc in docs[idx:]:
if isinstance(doc, Output):
output = doc
elif isinstance(doc, Assert):
confirm = doc
elif isinstance(doc, Python):
python = doc
if typ is None:
typ = 'pyrun'
elif isinstance(doc, YAMLData):
data = doc
else:
print('no handler for type:', type(doc), repr(doc))
assert False
if typ is None:
if data is not None and output is not None:
typ = 'rt'
elif data is not None and confirm is not None:
typ = 'load_assert'
else:
assert data is not None
typ = 'rt'
print('type:', typ)
print('data:', data.value, end='')
print('output:', output.value if output is not None else output)
if typ == 'rt':
self.round_trip(data, output, yaml_version=yaml_version)
elif typ == 'pyrun':
self.run_python(python, output if output is not None else data, tmpdir)
elif typ == 'load_assert':
self.load_assert(data, confirm, yaml_version=yaml_version)
else:
print('\nrun type unknown:', typ)
assert False
|
