summaryrefslogtreecommitdiff
path: root/tests/docx/body_xml_tests.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/docx/body_xml_tests.py')
-rw-r--r--tests/docx/body_xml_tests.py8
1 files changed, 3 insertions, 5 deletions
diff --git a/tests/docx/body_xml_tests.py b/tests/docx/body_xml_tests.py
index 2a2454d..badd5b8 100644
--- a/tests/docx/body_xml_tests.py
+++ b/tests/docx/body_xml_tests.py
@@ -1199,11 +1199,9 @@ def _read_and_get_document_xml(func, *args, **kwargs):
return result.value
-def _read_document_xml_element(*args, **kwargs):
- return _read_document_xml(
- lambda reader, element: reader.read_all([element]).map(single),
- *args,
- **kwargs)
+def _read_document_xml_element(element, **kwargs):
+ reader = _create_body_reader(**kwargs)
+ return reader.read_all([element]).map(single)
def _read_document_xml(func, element, **kwargs):