text: add Paragraph.style setter

Steve Canny · Steve Canny · commit 80390ff23647 · 2013-12-20T20:12:40.000-08:00
diff --git a/docx/text.py b/docx/text.py
@@ -43,6 +43,10 @@ def style(self):
         style = self._p.style
         return style if style is not None else 'Normal'
 
+    @style.setter
+    def style(self, style):
+        self._p.style = None if style == 'Normal' else style
+
 
 class Run(object):
     """
diff --git a/tests/test_text.py b/tests/test_text.py
@@ -59,6 +59,17 @@ def it_knows_its_paragraph_style(self):
             p = Paragraph(p_elm)
             assert p.style == expected_style
 
+    def it_can_set_its_paragraph_style(self):
+        cases = (
+            ('foobar', 'foobar'),
+            ('Normal', None),
+        )
+        for style, expected_setting in cases:
+            p_elm = Mock(name='p_elm')
+            p = Paragraph(p_elm)
+            p.style = style
+            assert p_elm.style == expected_setting
+
 
 class DescribeRun(object):
 
