From 838736f4d74b43d59aa7bdc96830e0be7fd7f258 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 20 May 2017 15:28:37 +0100
Subject: Read whether row is header

---
 tests/docx/body_xml_tests.py    | 20 ++++++++++++++++++++
 tests/docx/document_matchers.py |  2 ++
 2 files changed, 22 insertions(+)

(limited to 'tests/docx')

diff --git a/tests/docx/body_xml_tests.py b/tests/docx/body_xml_tests.py
index cccf472..e768bd5 100644
--- a/tests/docx/body_xml_tests.py
+++ b/tests/docx/body_xml_tests.py
@@ -17,6 +17,8 @@ from .document_matchers import (
     is_run,
     is_hyperlink,
     is_text,
+    is_table,
+    is_row,
 )
 
 
@@ -466,6 +468,24 @@ class TableTests(object):
         ])
         assert_equal(expected_result, table)
     
+    @istest
+    def tbl_header_marks_table_row_as_header(self):
+        element = xml_element("w:tbl", {}, [
+            xml_element("w:tr", {}, [
+                xml_element("w:trPr", {}, [
+                    xml_element("w:tblHeader")
+                ]),
+            ]),
+            xml_element("w:tr"),
+        ])
+        table = _read_and_get_document_xml_element(element)
+        assert_that(table, is_table(
+            children=is_sequence(
+                is_row(is_header=True),
+                is_row(is_header=False),
+            ),
+        ))
+    
     
     @istest
     def gridspan_is_read_as_colspan_for_table_cell(self):
diff --git a/tests/docx/document_matchers.py b/tests/docx/document_matchers.py
index dd2276d..d2e5d5c 100644
--- a/tests/docx/document_matchers.py
+++ b/tests/docx/document_matchers.py
@@ -16,6 +16,8 @@ def create_element_matcher(element_type):
 is_paragraph = create_element_matcher(documents.Paragraph)
 is_run = create_element_matcher(documents.Run)
 is_hyperlink = create_element_matcher(documents.Hyperlink)
+is_table = create_element_matcher(documents.Table)
+is_row = create_element_matcher(documents.TableRow)
 
 
 is_empty_run = is_run(children=[])
-- 
cgit v1.2.3