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
|
# Skipped because we have no idea where the "recursive_filename"
# fixture is supposed to come from
import pytest
import ruyaml
pytestmark = pytest.mark.skip
class AnInstance:
def __init__(self, foo, bar):
self.foo = foo
self.bar = bar
def __repr__(self):
try:
return '%s(foo=%r, bar=%r)' % (self.__class__.__name__, self.foo, self.bar)
except RuntimeError:
return '%s(foo=..., bar=...)' % self.__class__.__name__
class AnInstanceWithState(AnInstance):
def __getstate__(self):
return {'attributes': [self.foo, self.bar]}
def __setstate__(self, state):
self.foo, self.bar = state['attributes']
def test_recursive(recursive_filename, verbose=False):
yaml = ruyaml.YAML(typ='safe', pure=True)
context = globals().copy()
with open(recursive_filename, 'rb') as fp0:
exec(fp0.read(), context)
value1 = context['value']
output1 = None
value2 = None
output2 = None
try:
buf = ruyaml.compat.StringIO()
output1 = yaml.dump(value1, buf)
yaml.load(output1)
value2 = buf.getvalue()
buf = ruyaml.compat.StringIO()
yaml.dump(value2, buf)
output2 = buf.getvalue()
assert output1 == output2, (output1, output2)
finally:
if verbose:
print('VALUE1:', value1)
print('VALUE2:', value2)
print('OUTPUT1:')
print(output1)
print('OUTPUT2:')
print(output2)
test_recursive.unittest = ['.recursive']
if __name__ == '__main__':
import test_appliance
test_appliance.run(globals())
|
