tbl: add Table.alignment.setter

Steve Canny · Steve Canny · commit f9e033e2dc41 · 2014-12-01T19:17:36.000-05:00
diff --git a/docx/oxml/table.py b/docx/oxml/table.py
@@ -158,6 +158,14 @@ def alignment(self):
             return None
         return jc.val
 
+    @alignment.setter
+    def alignment(self, value):
+        self._remove_jc()
+        if value is None:
+            return
+        jc = self.get_or_add_jc()
+        jc.val = value
+
     @property
     def autofit(self):
         """
diff --git a/docx/table.py b/docx/table.py
@@ -49,6 +49,10 @@ def alignment(self):
         """
         return self._tblPr.alignment
 
+    @alignment.setter
+    def alignment(self, value):
+        self._tblPr.alignment = value
+
     @property
     def autofit(self):
         """
diff --git a/features/tbl-props.feature b/features/tbl-props.feature
@@ -16,7 +16,6 @@ Feature: Get and set table properties
       | center    | WD_TABLE_ALIGNMENT.CENTER |
 
 
-  @wip
   Scenario Outline: Set table alignment
     Given a table having <alignment> alignment
      When I assign <value> to table.alignment
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -28,6 +28,11 @@ def it_knows_its_alignment_setting(self, alignment_get_fixture):
         table, expected_value = alignment_get_fixture
         assert table.alignment == expected_value
 
+    def it_can_change_its_alignment_setting(self, alignment_set_fixture):
+        table, new_value, expected_xml = alignment_set_fixture
+        table.alignment = new_value
+        assert table._tbl.xml == expected_xml
+
     def it_knows_whether_it_should_autofit(self, autofit_get_fixture):
         table, expected_value = autofit_get_fixture
         assert table.autofit is expected_value
@@ -133,6 +138,20 @@ def alignment_get_fixture(self, request):
         table = Table(element(tbl_cxml), None)
         return table, expected_value
 
+    @pytest.fixture(params=[
+        ('w:tbl/w:tblPr', WD_TABLE_ALIGNMENT.LEFT,
+         'w:tbl/w:tblPr/w:jc{w:val=left}'),
+        ('w:tbl/w:tblPr/w:jc{w:val=left}', WD_TABLE_ALIGNMENT.RIGHT,
+         'w:tbl/w:tblPr/w:jc{w:val=right}'),
+        ('w:tbl/w:tblPr/w:jc{w:val=right}', None,
+         'w:tbl/w:tblPr'),
+    ])
+    def alignment_set_fixture(self, request):
+        tbl_cxml, new_value, expected_tbl_cxml = request.param
+        table = Table(element(tbl_cxml), None)
+        expected_xml = xml(expected_tbl_cxml)
+        return table, new_value, expected_xml
+
     @pytest.fixture(params=[
         ('w:tbl/w:tblPr',                             True),
         ('w:tbl/w:tblPr/w:tblLayout',                 True),
