summaryrefslogtreecommitdiff
path: root/_test/lib/test_recursive.py
blob: c87f879cac37d5a5eaf5f8cffce87e5a85bded08 (plain)
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
from __future__ import absolute_import
from __future__ import print_function

import ruamel.yaml as yaml


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):
    context = globals().copy()
    with open(recursive_filename, 'rb') as fp0:
        exec(fp0.read(), context)
    value1 = context['value']
    output1 = None
    value2 = None
    output2 = None
    try:
        output1 = yaml.dump(value1)
        value2 = yaml.load(output1)
        output2 = yaml.dump(value2)
        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())