Mercurial > jouvence

annotate tests/conftest.py @ 2:59fe8cb6190d

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add lots of tests, fix lots of bugs.
author Ludovic Chabant <ludovic@chabant.com>
date Tue, 03 Jan 2017 09:05:28 -0800
parents 74b83e3d921e
children ee741bbe96a8
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
2
59fe8cb6190d Add lots of tests, fix lots of bugs.
Ludovic Chabant <ludovic@chabant.com>
parents: 1
diff changeset
1 import re
0
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
2 import sys
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
3 import logging
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
4 import yaml
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
5 import pytest
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
6 from fontaine.document import (
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
7 FontaineSceneElement,
1
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
8 TYPE_ACTION, TYPE_CENTEREDACTION, TYPE_CHARACTER, TYPE_DIALOG,
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
9 TYPE_PARENTHETICAL, TYPE_TRANSITION, TYPE_LYRICS, TYPE_PAGEBREAK,
2
59fe8cb6190d Add lots of tests, fix lots of bugs.
Ludovic Chabant <ludovic@chabant.com>
parents: 1
diff changeset
10 TYPE_EMPTYLINES,
1
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
11 _scene_element_type_str)
0
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
12 from fontaine.parser import FontaineParser, FontaineParserError
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
13
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
14
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
15 def pytest_addoption(parser):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
16 parser.addoption(
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
17 '--log-debug',
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
18 action='store_true',
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
19 help="Sets the Fontaine logger to output debug info to stdout.")
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
20
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
21
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
22 def pytest_configure(config):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
23 if config.getoption('--log-debug'):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
24 hdl = logging.StreamHandler(stream=sys.stdout)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
25 logging.getLogger('fontaine').addHandler(hdl)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
26 logging.getLogger('fontaine').setLevel(logging.DEBUG)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
27
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
28
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
29 def pytest_collect_file(parent, path):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
30 if path.ext == '.yaml' and path.basename.startswith("test"):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
31 return FontaineScriptTestFile(path, parent)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
32 return None
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
33
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
34
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
35 def assert_scenes(actual, scenes):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
36 assert len(actual) == len(scenes)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
37 for a, e in zip(actual, scenes):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
38 assert_scene(a, e[0], e[1:])
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
39
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
40
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
41 def assert_scene(actual, header, paragraphs):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
42 if header is not None:
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
43 assert actual.header == header
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
44 assert len(actual.paragraphs) == len(paragraphs)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
45 for a, e in zip(actual.paragraphs, paragraphs):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
46 assert_paragraph(a, e)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
47
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
48
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
49 def assert_paragraph(actual, expected):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
50 if isinstance(expected, str):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
51 assert isinstance(actual, FontaineSceneElement)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
52 assert actual.type == TYPE_ACTION
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
53 assert actual.text == expected
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
54 elif isinstance(expected, FontaineSceneElement):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
55 assert isinstance(actual, FontaineSceneElement)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
56 assert actual.type == expected.type
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
57 assert actual.text == expected.text
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
58 else:
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
59 raise NotImplementedError("Don't know what this is: %s" % expected)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
60
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
61
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
62 def _c(name):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
63 return FontaineSceneElement(TYPE_CHARACTER, name)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
64
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
65
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
66 def _p(text):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
67 return FontaineSceneElement(TYPE_PARENTHETICAL, text)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
68
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
69
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
70 def _d(text):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
71 return FontaineSceneElement(TYPE_DIALOG, text)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
72
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
73
1
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
74 def _t(text):
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
75 return FontaineSceneElement(TYPE_TRANSITION, text)
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
76
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
77
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
78 def _l(text):
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
79 return FontaineSceneElement(TYPE_LYRICS, text)
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
80
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
81
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
82 class UnexpectedScriptOutput(Exception):
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
83 def __init__(self, actual, expected):
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
84 self.actual = actual
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
85 self.expected = expected
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
86
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
87
0
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
88 class FontaineScriptTestFile(pytest.File):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
89 def collect(self):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
90 spec = yaml.load_all(self.fspath.open(encoding='utf8'))
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
91 for i, item in enumerate(spec):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
92 name = '%s_%d' % (self.fspath.basename, i)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
93 if 'test_name' in item:
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
94 name += '_%s' % item['test_name']
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
95 yield FontaineScriptTestItem(name, self, item)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
96
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
97
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
98 class FontaineScriptTestItem(pytest.Item):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
99 def __init__(self, name, parent, spec):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
100 super().__init__(name, parent)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
101 self.spec = spec
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
102
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
103 def reportinfo(self):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
104 return self.fspath, 0, "fontaine script test: %s" % self.name
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
105
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
106 def runtest(self):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
107 intext = self.spec.get('in')
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
108 expected = self.spec.get('out')
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
109 title = self.spec.get('title')
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
110 if intext is None or expected is None:
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
111 raise Exception("No 'in' or 'out' specified.")
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
112
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
113 parser = FontaineParser()
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
114 doc = parser.parseString(intext)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
115 if title is not None:
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
116 assert title == doc.title_values
1
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
117
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
118 exp_scenes = make_scenes(expected)
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
119 try:
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
120 assert_scenes(doc.scenes, exp_scenes)
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
121 except AssertionError:
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
122 raise UnexpectedScriptOutput(doc.scenes, exp_scenes)
0
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
123
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
124 def repr_failure(self, excinfo):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
125 if isinstance(excinfo.value, FontaineParserError):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
126 return ('\n'.join(
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
127 ['Parser error:', str(excinfo.value)]))
1
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
128 if isinstance(excinfo.value, UnexpectedScriptOutput):
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
129 return ('\n'.join(
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
130 ['Unexpected output:'] +
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
131 ['', 'Actual:'] +
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
132 list(_repr_doc_scenes(excinfo.value.actual)) +
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
133 ['', 'Expected:'] +
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
134 list(_repr_expected_scenes(excinfo.value.expected))))
0
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
135 return super().repr_failure(excinfo)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
136
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
137
1
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
138 def _repr_doc_scenes(scenes):
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
139 for s in scenes:
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
140 yield 'Scene: "%s"' % s.header
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
141 for p in s.paragraphs:
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
142 yield ' %s: "%s"' % (_scene_element_type_str(p.type),
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
143 p.text)
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
144
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
145
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
146 def _repr_expected_scenes(scenes):
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
147 for s in scenes:
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
148 yield 'Scene: "%s"' % s[0]
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
149 for p in s[1:]:
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
150 if isinstance(p, str):
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
151 yield ' ACTION: "%s"' % p
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
152 else:
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
153 yield ' %s: "%s"' % (_scene_element_type_str(p.type),
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
154 p.text)
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
155
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
156
2
59fe8cb6190d Add lots of tests, fix lots of bugs.
Ludovic Chabant <ludovic@chabant.com>
parents: 1
diff changeset
157 RE_BLANK_LINE = re.compile(r"^\s+$")
59fe8cb6190d Add lots of tests, fix lots of bugs.
Ludovic Chabant <ludovic@chabant.com>
parents: 1
diff changeset
158
59fe8cb6190d Add lots of tests, fix lots of bugs.
Ludovic Chabant <ludovic@chabant.com>
parents: 1
diff changeset
159
0
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
160 def make_scenes(spec):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
161 if not isinstance(spec, list):
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
162 raise Exception("Script specs must be lists.")
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
163
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
164 out = []
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
165 cur_header = None
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
166 cur_paras = []
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
167
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
168 for item in spec:
1
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
169 if item == '<pagebreak>':
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
170 cur_paras.append(FontaineSceneElement(TYPE_PAGEBREAK, None))
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
171 continue
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
172
2
59fe8cb6190d Add lots of tests, fix lots of bugs.
Ludovic Chabant <ludovic@chabant.com>
parents: 1
diff changeset
173 if RE_BLANK_LINE.match(item):
59fe8cb6190d Add lots of tests, fix lots of bugs.
Ludovic Chabant <ludovic@chabant.com>
parents: 1
diff changeset
174 text = len(item) * '\n'
59fe8cb6190d Add lots of tests, fix lots of bugs.
Ludovic Chabant <ludovic@chabant.com>
parents: 1
diff changeset
175 cur_paras.append(FontaineSceneElement(TYPE_EMPTYLINES, text))
59fe8cb6190d Add lots of tests, fix lots of bugs.
Ludovic Chabant <ludovic@chabant.com>
parents: 1
diff changeset
176 continue
59fe8cb6190d Add lots of tests, fix lots of bugs.
Ludovic Chabant <ludovic@chabant.com>
parents: 1
diff changeset
177
0
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
178 token = item[:1]
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
179 if token == '.':
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
180 if cur_header or cur_paras:
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
181 out.append([cur_header] + cur_paras)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
182 cur_header = item[1:]
1
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
183 cur_paras = []
0
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
184 elif token == '!':
1
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
185 if item[1:3] == '><':
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
186 cur_paras.append(
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
187 FontaineSceneElement(TYPE_CENTEREDACTION, item[3:]))
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
188 else:
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
189 cur_paras.append(item[1:])
0
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
190 elif token == '@':
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
191 cur_paras.append(_c(item[1:]))
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
192 elif token == '=':
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
193 cur_paras.append(_d(item[1:]))
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
194 elif token == '_':
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
195 cur_paras.append(_p(item[1:]))
1
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
196 elif token == '>':
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
197 cur_paras.append(_t(item[1:]))
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
198 elif token == '~':
74b83e3d921e Add more states, add more tests.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
199 cur_paras.append(_l(item[1:]))
0
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
200 else:
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
201 raise Exception("Unknown token: %s" % token)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
202 if cur_header or cur_paras:
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
203 out.append([cur_header] + cur_paras)
243401c49520 Initial commit.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
204 return out