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
|
import pickle
from tatsu.semantics import ModelBuilderSemantics
from tatsu.tool import compile
from tatsu.util import asjson
def test_synth_model():
grammar = """
start::ASeq
=
aseq
$
;
aseq
=
{'a'}+
;
"""
m = compile(grammar, 'ASeq')
model = m.parse('a a a', semantics=ModelBuilderSemantics())
assert type(model).__name__ == 'ASeq'
p = pickle.dumps(model)
new_model = pickle.loads(p)
assert type(new_model).__name__ == 'ASeq'
assert model.ast == new_model.ast
def test_nested_class_synth_model():
grammar = """
start::ASeq
=
seqs:aseq
$
;
aseq::Seq
=
values:{'a'}+
;
"""
m = compile(grammar, 'ASeq')
model = m.parse('a a a', semantics=ModelBuilderSemantics())
assert type(model).__name__ == 'ASeq'
p = pickle.dumps(model)
new_model = pickle.loads(p)
assert type(new_model).__name__ == 'ASeq'
# NOTE: Since we are unpickling an object which contains nested objects, we can't do
# self.assertEqual(model.ast, new_model.ast) as the memory locations will be different.
# So either (1) we recursively walk the objects and compare fields or (2) we convert it into a
# str()/repr()/JSON and compare that. The latter as it is easier.
assert asjson(model.ast) == asjson(new_model.ast)
|
