diff options
Diffstat (limited to 'tests/docx')
| -rw-r--r-- | tests/docx/body_xml_tests.py | 20 | ||||
| -rw-r--r-- | tests/docx/styles_xml_tests.py | 15 |
2 files changed, 27 insertions, 8 deletions
diff --git a/tests/docx/body_xml_tests.py b/tests/docx/body_xml_tests.py index c81fade..ad509c4 100644 --- a/tests/docx/body_xml_tests.py +++ b/tests/docx/body_xml_tests.py @@ -62,7 +62,11 @@ class ParagraphTests(object): properties_xml = xml_element("w:pPr", {}, [style_xml]) paragraph_xml = xml_element("w:p", {}, [properties_xml]) - styles = Styles({"Heading1": Style(style_id="Heading1", name="Heading 1")}, {}) + styles = Styles( + paragraph_styles={"Heading1": Style(style_id="Heading1", name="Heading 1")}, + character_styles={}, + table_styles={}, + ) paragraph = _read_and_get_document_xml_element(paragraph_xml, styles=styles) assert_equal("Heading1", paragraph.style_id) @@ -74,9 +78,7 @@ class ParagraphTests(object): properties_xml = xml_element("w:pPr", {}, [style_xml]) paragraph_xml = xml_element("w:p", {}, [properties_xml]) - styles = Styles({}, {}) - - result = _read_document_xml_element(paragraph_xml, styles=styles) + result = _read_document_xml_element(paragraph_xml, styles=Styles.EMPTY) paragraph = result.value assert_equal("Heading1", paragraph.style_id) assert_equal(None, paragraph.style_name) @@ -153,7 +155,11 @@ class RunTests(object): def run_has_style_id_and_name_read_from_run_properties_if_present(self): style_xml = xml_element("w:rStyle", {"w:val": "Heading1Char"}) - styles = Styles({}, {"Heading1Char": Style(style_id="Heading1Char", name="Heading 1 Char")}) + styles = Styles( + paragraph_styles={}, + character_styles={"Heading1Char": Style(style_id="Heading1Char", name="Heading 1 Char")}, + table_styles={}, + ) run = self._read_run_with_properties([style_xml], styles=styles) assert_equal("Heading1Char", run.style_id) @@ -165,9 +171,7 @@ class RunTests(object): properties_xml = xml_element("w:rPr", {}, [style_xml]) run_xml = xml_element("w:r", {}, [properties_xml]) - styles = Styles({}, {}) - - result = _read_document_xml_element(run_xml, styles=styles) + result = _read_document_xml_element(run_xml, styles=Styles.EMPTY) run = result.value assert_equal("Heading1Char", run.style_id) assert_equal(None, run.style_name) diff --git a/tests/docx/styles_xml_tests.py b/tests/docx/styles_xml_tests.py index 94d91ce..d70e068 100644 --- a/tests/docx/styles_xml_tests.py +++ b/tests/docx/styles_xml_tests.py @@ -36,6 +36,18 @@ def character_style_can_be_found_by_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"), @@ -75,6 +87,9 @@ def _paragraph_style_element(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) |