From 764885cdd67087cbb93cf73641a2357ecc562606 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sat, 20 May 2023 19:54:13 +0100 Subject: Support reading multiple elements in tests --- tests/docx/body_xml_tests.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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 -- cgit v1.2.3