1 # This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
2 # Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
6 from librarian.builders import HtmlBuilder
7 from librarian.document import WLDocument
8 from librarian.html import extract_annotations
11 class AnnotationsTests(unittest.TestCase):
14 def _test_annotation(self, expected, got, name):
16 got[0].startswith('anchor-'),
17 "%s: Unexpected anchor: '%s', should begin with 'anchor-'" % (name, got[0])
21 "%s: Unexpected type, expected '%s', got '%s'" % (name, expected[0], got[1])
25 "%s: Unexpected qualifier, expected '%s', got '%s'" % (name, expected[1], got[2])
29 "%s: Unexpected text representation, expected '%s', got '%s'" % (name, expected[2], got[3])
31 exp_html = '<div class="fn-%s">%s</div>' % (expected[0], expected[3])
34 "%s: Unexpected html representation, expected '%s', got '%s'" % (name, exp_html, got[4])
37 def test_annotations(self):
42 '[przypis edytorski]',
43 '<p> [przypis edytorski]</p>'
47 ('<pr>Definiendum --- definiens.</pr>', (
50 'Definiendum \u2014 definiens. [przypis redakcyjny]',
51 '<p>Definiendum \u2014 definiens. [przypis redakcyjny]</p>'
55 ('<pt><slowo_obce>Definiendum</slowo_obce> --- definiens.</pt>', (
58 'Definiendum \u2014 definiens. [przypis tłumacza]',
59 '<p><em class="foreign-word">Definiendum</em> \u2014 definiens. [przypis tłumacza]</p>'
61 'Standard footnote.'),
63 ('<pr>Definiendum (łac.) --- definiens.</pr>', (
66 'Definiendum (łac.) \u2014 definiens. [przypis redakcyjny]',
67 '<p>Definiendum (łac.) \u2014 definiens. [przypis redakcyjny]</p>'
69 'Plain footnote with qualifier'),
71 ('<pe><slowo_obce>Definiendum</slowo_obce> (łac.) --- definiens.</pe>', (
74 'Definiendum (łac.) \u2014 definiens. [przypis edytorski]',
75 '<p><em class="foreign-word">Definiendum</em> (łac.) \u2014 definiens. [przypis edytorski]</p>'
77 'Standard footnote with qualifier.'),
79 ('<pt> <slowo_obce>Definiendum</slowo_obce> (daw.) --- definiens.</pt>', (
82 'Definiendum (daw.) \u2014 definiens. [przypis tłumacza]',
83 '<p> <em class="foreign-word">Definiendum</em> (daw.) \u2014 definiens. [przypis tłumacza]</p>'
85 'Standard footnote with leading whitespace and qualifier.'),
87 ('<pr>Definiendum (łac.) --- <slowo_obce>definiens</slowo_obce>.</pr>', (
90 'Definiendum (łac.) \u2014 definiens. [przypis redakcyjny]',
91 '<p>Definiendum (łac.) \u2014 <em class="foreign-word">definiens</em>. [przypis redakcyjny]</p>'
93 'Plain footnote with qualifier and some emphasis.'),
95 ('<pe><slowo_obce>Definiendum</slowo_obce> (łac.) --- <slowo_obce>definiens</slowo_obce>.</pe>', (
98 'Definiendum (łac.) \u2014 definiens. [przypis edytorski]',
99 '<p><em class="foreign-word">Definiendum</em> (łac.) \u2014 <em class="foreign-word">definiens</em>. [przypis edytorski]</p>'
101 'Standard footnote with qualifier and some emphasis.'),
103 ('<pe>Definiendum (łac.) --- definiens (some) --- more text.</pe>', (
106 'Definiendum (łac.) \u2014 definiens (some) \u2014 more text. [przypis edytorski]',
107 '<p>Definiendum (łac.) \u2014 definiens (some) \u2014 more text. [przypis edytorski]</p>',
109 'Footnote with a second parentheses and mdash.'),
111 ('<pe><slowo_obce>gemajna</slowo_obce> (daw., z niem. <slowo_obce>gemein</slowo_obce>: zwykły) --- '
112 'częściej: gemajn, szeregowiec w wojsku polskim cudzoziemskiego autoramentu.</pe>', (
115 'gemajna (daw., z\u00A0niem. gemein: zwykły) \u2014 częściej: gemajn, '
116 'szeregowiec w\u00A0wojsku polskim cudzoziemskiego autoramentu. [przypis edytorski]',
117 '<p><em class="foreign-word">gemajna</em> (daw., z\u00A0niem. <em class="foreign-word">gemein</em>: zwykły) '
118 '\u2014 częściej: gemajn, szeregowiec w\u00A0wojsku polskim cudzoziemskiego autoramentu. [przypis edytorski]</p>'
120 'Footnote with multiple and qualifiers and emphasis.'),
123 xml_src = '''<utwor><akap> %s </akap></utwor>''' % "".join(
124 t[0] for t in annotations)
126 filename=io.BytesIO(xml_src.encode('utf-8'))
127 ).build(HtmlBuilder, base_url='/').get_file()
128 res_annotations = list(extract_annotations(html))
130 for i, (src, expected, name) in enumerate(annotations):
131 with self.subTest(i=i):
132 self._test_annotation(expected, res_annotations[i], name)