style: add LatentStyles.load_count setter

Steve Canny · Steve Canny · commit 6a449d5f7d15 · 2015-02-08T00:01:47.000-08:00
diff --git a/docx/styles/latent.py b/docx/styles/latent.py
@@ -95,6 +95,10 @@ def load_count(self):
         """
         return self._element.count
 
+    @load_count.setter
+    def load_count(self, value):
+        self._element.count = value
+
 
 class _LatentStyle(ElementProxy):
     """
diff --git a/tests/styles/test_latent.py b/tests/styles/test_latent.py
@@ -52,6 +52,11 @@ def it_knows_its_load_count(self, count_get_fixture):
         latent_styles, expected_value = count_get_fixture
         assert latent_styles.load_count == expected_value
 
+    def it_can_change_its_load_count(self, count_set_fixture):
+        latent_styles, value, expected_xml = count_set_fixture
+        latent_styles.load_count = value
+        assert latent_styles._element.xml == expected_xml
+
     def it_knows_its_boolean_properties(self, bool_prop_get_fixture):
         latent_styles, prop_name, expected_value = bool_prop_get_fixture
         actual_value = getattr(latent_styles, prop_name)
@@ -87,6 +92,17 @@ def count_get_fixture(self, request):
         latent_styles = LatentStyles(element(latentStyles_cxml))
         return latent_styles, expected_value
 
+    @pytest.fixture(params=[
+        ('w:latentStyles',             42,   'w:latentStyles{w:count=42}'),
+        ('w:latentStyles{w:count=24}', 42,   'w:latentStyles{w:count=42}'),
+        ('w:latentStyles{w:count=24}', None, 'w:latentStyles'),
+    ])
+    def count_set_fixture(self, request):
+        latentStyles_cxml, value, expected_cxml = request.param
+        latent_styles = LatentStyles(element(latentStyles_cxml))
+        expected_xml = xml(expected_cxml)
+        return latent_styles, value, expected_xml
+
     @pytest.fixture(params=[
         ('w:lsdException{w:name=Ab},w:lsdException,w:lsdException', 'Ab', 0),
         ('w:lsdException,w:lsdException{w:name=Cd},w:lsdException', 'Cd', 1),
