num: add _NumberingDefinitions.__len__()

Steve Canny · Steve Canny · commit 4bec90c90657 · 2014-02-02T16:39:53.000-08:00
diff --git a/docx/oxml/__init__.py b/docx/oxml/__init__.py
@@ -30,6 +30,9 @@
 register_custom_element_class('w:body', CT_Body)
 register_custom_element_class('w:document', CT_Document)
 
+from docx.oxml.parts.numbering import CT_Numbering
+register_custom_element_class('w:numbering', CT_Numbering)
+
 from docx.oxml.table import CT_Row, CT_Tbl, CT_TblGrid, CT_TblPr, CT_Tc
 register_custom_element_class('w:tbl', CT_Tbl)
 register_custom_element_class('w:tblGrid', CT_TblGrid)
diff --git a/docx/oxml/parts/numbering.py b/docx/oxml/parts/numbering.py
@@ -4,11 +4,17 @@
 Custom element classes related to the numbering part
 """
 
-from docx.oxml.shared import OxmlBaseElement
+from docx.oxml.shared import OxmlBaseElement, qn
 
 
 class CT_Numbering(OxmlBaseElement):
     """
     ``<w:numbering>`` element, the root element of a numbering part, i.e.
     numbering.xml
     """
+    @property
+    def num_lst(self):
+        """
+        List of <w:num> child elements.
+        """
+        return self.findall(qn('w:num'))
diff --git a/docx/parts/numbering.py b/docx/parts/numbering.py
@@ -58,3 +58,6 @@ class _NumberingDefinitions(object):
     def __init__(self, numbering_elm):
         super(_NumberingDefinitions, self).__init__()
         self._numbering = numbering_elm
+
+    def __len__(self):
+        return len(self._numbering.num_lst)
diff --git a/features/num-get-numbering-part.feature b/features/num-get-numbering-part.feature
@@ -3,7 +3,6 @@ Feature: Get the document numbering part
   As a programmer using the advanced python-docx API
   I need access to the numbering part of the document
 
-  @wip
   Scenario: Get an existing numbering part from document
     Given a document having a numbering part
      When I get the numbering part from the document
diff --git a/tests/oxml/unitdata/numbering.py b/tests/oxml/unitdata/numbering.py
@@ -0,0 +1,27 @@
+# encoding: utf-8
+
+"""
+Test data builders for numbering part XML elements
+"""
+
+from ...unitdata import BaseBuilder
+
+
+class CT_NumBuilder(BaseBuilder):
+    __tag__ = 'w:num'
+    __nspfxs__ = ('w',)
+    __attrs__ = ('w:numId')
+
+
+class CT_NumberingBuilder(BaseBuilder):
+    __tag__ = 'w:numbering'
+    __nspfxs__ = ('w',)
+    __attrs__ = ()
+
+
+def a_num():
+    return CT_NumBuilder()
+
+
+def a_numbering():
+    return CT_NumberingBuilder()
diff --git a/tests/parts/test_numbering.py b/tests/parts/test_numbering.py
@@ -15,6 +15,7 @@
 from docx.package import Package
 from docx.parts.numbering import NumberingPart, _NumberingDefinitions
 
+from ..oxml.unitdata.numbering import a_num, a_numbering
 from ..unitutil import (
     function_mock, class_mock, initializer_mock, instance_mock, method_mock
 )
@@ -138,3 +139,23 @@ def package_(self, request):
     @pytest.fixture
     def partname_(self, request):
         return instance_mock(request, PackURI)
+
+
+class Describe_NumberingDefinitions(object):
+
+    def it_knows_how_many_numbering_definitions_it_contains(
+            self, len_fixture):
+        numbering_definitions, numbering_definition_count = len_fixture
+        assert len(numbering_definitions) == numbering_definition_count
+
+    # fixtures -------------------------------------------------------
+
+    @pytest.fixture(params=[0, 1, 2, 3])
+    def len_fixture(self, request):
+        numbering_definition_count = request.param
+        numbering_bldr = a_numbering().with_nsdecls()
+        for idx in range(numbering_definition_count):
+            numbering_bldr.with_child(a_num())
+        numbering_elm = numbering_bldr.element
+        numbering_definitions = _NumberingDefinitions(numbering_elm)
+        return numbering_definitions, numbering_definition_count
