Pass styles into read_numbering_xml_element

1 files changed, 13 insertions, 5 deletions

diff --git a/tests/docx/numbering_xml_tests.py b/tests/docx/numbering_xml_tests.py
index 8fb59ff..467d406 100644
--- a/tests/docx/numbering_xml_tests.py
+++ b/tests/docx/numbering_xml_tests.py
@@ -2,11 +2,12 @@ from nose.tools import istest, assert_equal
 
 from mammoth.docx.xmlparser import element as xml_element
 from mammoth.docx.numbering_xml import read_numbering_xml_element
+from mammoth.docx.styles_xml import Styles
 
 
 @istest
 def find_level_returns_none_if_num_with_id_cannot_be_found():
-    numbering = read_numbering_xml_element(xml_element("w:numbering"))
+    numbering = _read_numbering_xml_element(xml_element("w:numbering"))
     assert_equal(None, numbering.find_level("47", "0"))
 
 
@@ -27,23 +28,30 @@ _sample_numbering_xml = xml_element("w:numbering", {}, [
 
 @istest
 def level_includes_level_index():
-    numbering = read_numbering_xml_element(_sample_numbering_xml)
+    numbering = _read_numbering_xml_element(_sample_numbering_xml)
     assert_equal("0", numbering.find_level("47", "0").level_index)
     assert_equal("1", numbering.find_level("47", "1").level_index)
 
 
 @istest
 def list_is_not_ordered_if_formatted_as_bullet():
-    numbering = read_numbering_xml_element(_sample_numbering_xml)
+    numbering = _read_numbering_xml_element(_sample_numbering_xml)
     assert_equal(False, numbering.find_level("47", "0").is_ordered)
 
 @istest
 def list_is_ordered_if_formatted_as_decimal():
-    numbering = read_numbering_xml_element(_sample_numbering_xml)
+    numbering = _read_numbering_xml_element(_sample_numbering_xml)
     assert_equal(True, numbering.find_level("47", "1").is_ordered)
 
 
 @istest
 def find_level_returns_none_if_level_cannot_be_found():
-    numbering = read_numbering_xml_element(_sample_numbering_xml)
+    numbering = _read_numbering_xml_element(_sample_numbering_xml)
     assert_equal(None, numbering.find_level("47", "2"))
+
+
+def _read_numbering_xml_element(element, styles=None):
+    if styles is None:
+        styles = Styles.EMPTY
+
+    return read_numbering_xml_element(element, styles=styles)