From cd95535a1bb0f486d4a102cfc622ae5228c3071d Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Fri, 3 Nov 2017 16:00:19 +0000
Subject: Read table styles

---
 tests/docx/styles_xml_tests.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'tests/docx/styles_xml_tests.py')

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
@@ -35,6 +35,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", {}, [
@@ -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)
-- 
cgit v1.2.3