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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
from nose.tools import istest, assert_equal
from mammoth.docx.xmlparser import element as xml_element
from mammoth.docx.styles_xml import read_styles_xml_element
@istest
def paragraph_style_is_null_if_no_style_with_that_id_exists():
element = xml_element("w:styles")
styles = read_styles_xml_element(element)
assert_equal(None, styles.find_paragraph_style_by_id("Heading1"))
@istest
def paragraph_style_can_be_found_by_id():
element = xml_element("w:styles", {}, [
_paragraph_style_element("Heading1", "Heading 1"),
])
styles = read_styles_xml_element(element)
assert_equal(
"Heading1",
styles.find_paragraph_style_by_id("Heading1").style_id
)
@istest
def character_style_can_be_found_by_id():
element = xml_element("w:styles", {}, [
_character_style_element("Heading1Char", "Heading 1 Char"),
])
styles = read_styles_xml_element(element)
assert_equal(
"Heading1Char",
styles.find_character_style_by_id("Heading1Char").style_id
)
@istest
def table_style_can_be_found_by_id():
element = xml_element("w:styles", {}, [
_table_style_element("TableNormal", "Normal Table"),
])
styles = read_styles_xml_element(element)
assert_equal(
"TableNormal",
styles.find_table_style_by_id("TableNormal").style_id
)
@istest
def paragraph_and_character_styles_are_distinct():
element = xml_element("w:styles", {}, [
_paragraph_style_element("Heading1", "Heading 1"),
_character_style_element("Heading1Char", "Heading 1 Char"),
])
styles = read_styles_xml_element(element)
assert_equal(None, styles.find_character_style_by_id("Heading1"))
assert_equal(None, styles.find_paragraph_style_by_id("Heading1Char"))
@istest
def styles_include_names():
element = xml_element("w:styles", {}, [
_paragraph_style_element("Heading1", "Heading 1"),
])
styles = read_styles_xml_element(element)
assert_equal(
"Heading 1",
styles.find_paragraph_style_by_id("Heading1").name
)
@istest
def style_name_is_none_if_name_element_does_not_exist():
element = xml_element("w:styles", {}, [
_style_without_name_element("paragraph", "Heading1"),
_style_without_name_element("character", "Heading1Char")
])
styles = read_styles_xml_element(element)
assert_equal(None, styles.find_paragraph_style_by_id("Heading1").name)
assert_equal(None, styles.find_character_style_by_id("Heading1Char").name)
@istest
def numbering_style_is_none_if_no_style_with_that_id_exists():
element = xml_element("w:styles", {}, [])
styles = read_styles_xml_element(element)
assert_equal(None, styles.find_numbering_style_by_id("List1"))
@istest
def numbering_style_has_none_num_id_if_style_has_no_paragraph_properties():
element = xml_element("w:styles", {}, [
xml_element("w:style", {"w:type": "numbering", "w:styleId": "List1"}),
])
styles = read_styles_xml_element(element)
assert_equal(None, styles.find_numbering_style_by_id("List1").num_id)
@istest
def numbering_style_has_num_id_read_from_paragraph_properties():
element = xml_element("w:styles", {}, [
xml_element("w:style", {"w:type": "numbering", "w:styleId": "List1"}, [
xml_element("w:pPr", {}, [
xml_element("w:numPr", {}, [
xml_element("w:numId", {"w:val": "42"})
]),
]),
]),
])
styles = read_styles_xml_element(element)
assert_equal("42", styles.find_numbering_style_by_id("List1").num_id)
def _paragraph_style_element(style_id, name):
return _style_element("paragraph", style_id, name)
def _character_style_element(style_id, name):
return _style_element("character", style_id, name)
def _table_style_element(style_id, name):
return _style_element("table", style_id, name)
def _style_element(element_type, style_id, name):
children = [xml_element("w:name", {"w:val": name}, [])]
return _style_element_with_children(element_type, style_id, children)
def _style_without_name_element(element_type, style_id):
return _style_element_with_children(element_type, style_id, [])
def _style_element_with_children(element_type, style_id, children):
attributes = {"w:type": element_type, "w:styleId": style_id}
return xml_element("w:style", attributes, children)
|
