summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--tests/docx/body_xml_tests.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/docx/body_xml_tests.py b/tests/docx/body_xml_tests.py
index 2459f4a..1b02b97 100644
--- a/tests/docx/body_xml_tests.py
+++ b/tests/docx/body_xml_tests.py
@@ -1183,8 +1183,11 @@ def _read_and_get_document_xml_element(element, **kwargs):
def _read_and_get_document_xml_elements(element, **kwargs):
+ if not isinstance(element, list):
+ element = [element]
+
reader = _create_body_reader(**kwargs)
- result = reader.read_all([element])
+ result = reader.read_all(element)
assert_equal([], result.messages)
return result.value