sect: add Sections.__getitem__()

Steve Canny · Steve Canny · commit e0537246c0d9 · 2014-06-19T00:24:29.000-07:00
diff --git a/docx/parts/document.py b/docx/parts/document.py
@@ -236,7 +236,11 @@ def __init__(self, document_elm):
         self._document_elm = document_elm
 
     def __getitem__(self, key):
-        pass
+        if isinstance(key, slice):
+            sectPr_lst = self._document_elm.sectPr_lst[key]
+            return [Section(sectPr) for sectPr in sectPr_lst]
+        sectPr = self._document_elm.sectPr_lst[key]
+        return Section(sectPr)
 
     def __iter__(self):
         for sectPr in self._document_elm.sectPr_lst:
diff --git a/features/doc-access-sections.feature b/features/doc-access-sections.feature
@@ -10,7 +10,6 @@ Feature: Access document sections
        And the length of the section collection is 3
 
 
-  @wip
   Scenario: Access section in section collection
      Given a section collection
       Then I can iterate over the sections
diff --git a/tests/parts/test_document.py b/tests/parts/test_document.py
@@ -635,8 +635,19 @@ def it_can_iterate_over_its_Section_instances(self, iter_fixture):
             assert isinstance(section, Section)
         assert section_count == expected_count
 
+    def it_can_access_its_Section_instances_by_index(self, index_fixture):
+        sections, indicies = index_fixture
+        assert len(sections[0:2]) == 2
+        for index in indicies:
+            assert isinstance(sections[index], Section)
+
     # fixtures -------------------------------------------------------
 
+    @pytest.fixture
+    def index_fixture(self, document_elm):
+        sections = Sections(document_elm)
+        return sections, [0, 1]
+
     @pytest.fixture
     def iter_fixture(self, document_elm):
         sections = Sections(document_elm)
