diff options
Diffstat (limited to 'tests/docx')
| -rw-r--r-- | tests/docx/body_xml_tests.py | 20 | ||||
| -rw-r--r-- | tests/docx/document_matchers.py | 2 |
2 files changed, 22 insertions, 0 deletions
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=[]) |