font: add Font.subscript setter

Steve Canny · Steve Canny · commit 05065bf1d1cf · 2015-02-08T00:01:42.000-08:00
diff --git a/docx/oxml/text/run.py b/docx/oxml/text/run.py
@@ -242,6 +242,17 @@ def subscript(self):
             return True
         return False
 
+    @subscript.setter
+    def subscript(self, value):
+        if value is None:
+            self._remove_vertAlign()
+        elif bool(value) is True:
+            self.get_or_add_vertAlign().val = ST_VerticalAlignRun.SUBSCRIPT
+        elif self.vertAlign is None:
+            return
+        elif self.vertAlign.val == ST_VerticalAlignRun.SUBSCRIPT:
+            self._remove_vertAlign()
+
     @property
     def superscript(self):
         """
diff --git a/docx/text/run.py b/docx/text/run.py
@@ -492,6 +492,11 @@ def subscript(self):
             return None
         return rPr.subscript
 
+    @subscript.setter
+    def subscript(self, value):
+        rPr = self._element.get_or_add_rPr()
+        rPr.subscript = value
+
     @property
     def superscript(self):
         """
diff --git a/tests/text/test_run.py b/tests/text/test_run.py
@@ -373,6 +373,11 @@ def it_knows_whether_it_is_subscript(self, subscript_get_fixture):
         font, expected_value = subscript_get_fixture
         assert font.subscript == expected_value
 
+    def it_can_change_whether_it_is_subscript(self, subscript_set_fixture):
+        font, value, expected_xml = subscript_set_fixture
+        font.subscript = value
+        assert font._element.xml == expected_xml
+
     def it_knows_whether_it_is_superscript(self, superscript_get_fixture):
         font, expected_value = superscript_get_fixture
         assert font.superscript == expected_value
@@ -535,6 +540,34 @@ def subscript_get_fixture(self, request):
         font = Font(element(r_cxml))
         return font, expected_value
 
+    @pytest.fixture(params=[
+        ('w:r',                                      True,
+         'w:r/w:rPr/w:vertAlign{w:val=subscript}'),
+        ('w:r',                                      False,
+         'w:r/w:rPr'),
+        ('w:r',                                      None,
+         'w:r/w:rPr'),
+        ('w:r/w:rPr/w:vertAlign{w:val=subscript}',   True,
+         'w:r/w:rPr/w:vertAlign{w:val=subscript}'),
+        ('w:r/w:rPr/w:vertAlign{w:val=subscript}',   False,
+         'w:r/w:rPr'),
+        ('w:r/w:rPr/w:vertAlign{w:val=subscript}',   None,
+         'w:r/w:rPr'),
+        ('w:r/w:rPr/w:vertAlign{w:val=superscript}', True,
+         'w:r/w:rPr/w:vertAlign{w:val=subscript}'),
+        ('w:r/w:rPr/w:vertAlign{w:val=superscript}', False,
+         'w:r/w:rPr/w:vertAlign{w:val=superscript}'),
+        ('w:r/w:rPr/w:vertAlign{w:val=superscript}', None,
+         'w:r/w:rPr'),
+        ('w:r/w:rPr/w:vertAlign{w:val=baseline}',    True,
+         'w:r/w:rPr/w:vertAlign{w:val=subscript}'),
+    ])
+    def subscript_set_fixture(self, request):
+        r_cxml, value, expected_r_cxml = request.param
+        font = Font(element(r_cxml))
+        expected_xml = xml(expected_r_cxml)
+        return font, value, expected_xml
+
     @pytest.fixture(params=[
         ('w:r',                                      None),
         ('w:r/w:rPr',                                None),
