tabs: add 'position in tab_stops' functionality

DKWoods · DKWoods · commit eaf4cf607d06 · 2016-05-11T12:10:09.000-05:00
diff --git a/docx/oxml/text/parfmt.py b/docx/oxml/text/parfmt.py
@@ -333,3 +333,13 @@ class CT_TabStops(BaseOxmlElement):
     ``<w:tabs>`` element, container for a sorted sequence of tab stops.
     """
     tab = OneOrMore('w:tab', successors=())
+
+    def __contains__(self, position):
+        """
+        Allows 'in' operator to test for Position duplication rather than
+        looking at the entire tab_stop object
+        """
+        for tab in self.tab_lst:
+            if tab.pos == position:
+                return True
+        return False
diff --git a/docx/text/tabstops.py b/docx/text/tabstops.py
@@ -21,6 +21,12 @@ class TabStops(ElementProxy):
 
     __slots__ = ()
 
+    def __contains__(self, position):
+        """
+        Enables 'in' operator to test for Position duplication
+        """
+        return position in self._element.tabs
+
     def __getitem__(self, idx):
         """
         Enables list-style access by index.
diff --git a/features/tab-access-tabs.feature b/features/tab-access-tabs.feature
@@ -20,7 +20,6 @@ Feature: Adjust tab stops
       And I can access a tab stop by index
 
 
-  @wip
   Scenario Outline: Determine if tab stops contain a specific tab
     Given a tab_stops having 3 tab stops
      Then <position> in tab_stops is <result>
diff --git a/tests/text/test_tabstops.py b/tests/text/test_tabstops.py
@@ -143,8 +143,23 @@ def it_raises_on_indexed_access_when_empty(self):
         with pytest.raises(IndexError):
             tab_stops[0]
 
+    def it_knows_if_it_contains_a_tab_stop_at_a_position(self, in_fixture):
+        tab_stops, value, expected_result = in_fixture
+        assert ((int(value) in tab_stops) == expected_result)
+
     # fixture --------------------------------------------------------
 
+    @pytest.fixture(params=[
+        ('w:pPr/w:tabs/w:tab{w:pos=1}', Twips(1), 'True'),
+        ('w:pPr/w:tabs/w:tab{w:pos=1}', Twips(2), 'False'),
+    ])
+    def in_fixture(self, request):
+        pPr_cxml, value, expected_result = request.param
+        pPr = element(pPr_cxml)
+        tab_stops = TabStops(pPr)
+        expected_result = (expected_result == 'True')
+        return tab_stops, value, expected_result
+
     @pytest.fixture(params=[
         ('w:pPr/w:tabs/w:tab{w:pos=0}',                                 0),
         ('w:pPr/w:tabs/(w:tab{w:pos=1},w:tab{w:pos=2},w:tab{w:pos=3})', 1),
