style: add LatentStyles.__iter__()

Steve Canny · Steve Canny · commit faf2dee5b495 · 2015-02-08T00:01:47.000-08:00
diff --git a/docx/styles/latent.py b/docx/styles/latent.py
@@ -20,6 +20,9 @@ class LatentStyles(ElementProxy):
 
     __slots__ = ()
 
+    def __iter__(self):
+        return (_LatentStyle(ls) for ls in self._element.lsdException_lst)
+
     def __len__(self):
         return len(self._element.lsdException_lst)
 
diff --git a/tests/styles/test_latent.py b/tests/styles/test_latent.py
@@ -10,7 +10,7 @@
 
 import pytest
 
-from docx.styles.latent import LatentStyles
+from docx.styles.latent import _LatentStyle, LatentStyles
 
 from ..unitutil.cxml import element
 
@@ -21,8 +21,25 @@ def it_knows_how_many_latent_styles_it_contains(self, len_fixture):
         latent_styles, expected_value = len_fixture
         assert len(latent_styles) == expected_value
 
+    def it_can_iterate_over_its_latent_styles(self, iter_fixture):
+        latent_styles, expected_count = iter_fixture
+        lst = [ls for ls in latent_styles]
+        assert len(lst) == expected_count
+        for latent_style in lst:
+            assert isinstance(latent_style, _LatentStyle)
+
     # fixture --------------------------------------------------------
 
+    @pytest.fixture(params=[
+        ('w:latentStyles',                                  0),
+        ('w:latentStyles/w:lsdException',                   1),
+        ('w:latentStyles/(w:lsdException,w:lsdException)',  2),
+    ])
+    def iter_fixture(self, request):
+        latentStyles_cxml, count = request.param
+        latent_styles = LatentStyles(element(latentStyles_cxml))
+        return latent_styles, count
+
     @pytest.fixture(params=[
         ('w:latentStyles',                                  0),
         ('w:latentStyles/w:lsdException',                   1),
