diff options
Diffstat (limited to 'tests/docx')
| -rw-r--r-- | tests/docx/styles_xml_tests.py | 31 |
1 files changed, 30 insertions, 1 deletions
diff --git a/tests/docx/styles_xml_tests.py b/tests/docx/styles_xml_tests.py index 7e58333..c4f659d 100644 --- a/tests/docx/styles_xml_tests.py +++ b/tests/docx/styles_xml_tests.py @@ -17,12 +17,41 @@ def paragraph_style_can_be_found_by_id(): _paragraph_style_element("Heading1", "Heading 1"), ]) styles = read_styles_xml_element(element) - assert_equal("Heading1", styles.find_paragraph_style_by_id("Heading1").style_id) + 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 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")) 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 _style_element(element_type, style_id, name): return xml_element("w:style", {"w:type": element_type, "w:styleId": style_id}, [ xml_element("w:name", {"w:val": name}, []) |