tbl: add Table.autofit setter

Steve Canny · Steve Canny · commit f19d906ef519 · 2014-07-13T15:34:23.000-07:00
diff --git a/docx/oxml/table.py b/docx/oxml/table.py
@@ -105,6 +105,11 @@ def autofit(self):
             return True
         return False if tblLayout.type == 'fixed' else True
 
+    @autofit.setter
+    def autofit(self, value):
+        tblLayout = self.get_or_add_tblLayout()
+        tblLayout.type = 'autofit' if value else 'fixed'
+
     @property
     def style(self):
         """
diff --git a/docx/table.py b/docx/table.py
@@ -48,6 +48,10 @@ def autofit(self):
         """
         return self._tblPr.autofit
 
+    @autofit.setter
+    def autofit(self, value):
+        self._tblPr.autofit = value
+
     def cell(self, row_idx, col_idx):
         """
         Return |_Cell| instance correponding to table cell at *row_idx*,
diff --git a/features/tbl-props.feature b/features/tbl-props.feature
@@ -15,7 +15,6 @@ Feature: Get and set table properties
       | fixed               | fixed            |
 
 
-  @wip
   Scenario Outline: Set autofit layout setting
     Given a table having an autofit layout of <autofit-setting>
      When I set the table autofit to <new-setting>
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -64,6 +64,11 @@ def it_knows_whether_it_should_autofit(self, autofit_get_fixture):
         table, expected_value = autofit_get_fixture
         assert table.autofit is expected_value
 
+    def it_can_change_its_autofit_setting(self, autofit_set_fixture):
+        table, new_value, expected_xml = autofit_set_fixture
+        table.autofit = new_value
+        assert table._tbl.xml == expected_xml
+
     # fixtures -------------------------------------------------------
 
     @pytest.fixture
@@ -91,6 +96,24 @@ def autofit_get_fixture(self, request):
         table = Table(element(tbl_cxml), None)
         return table, expected_autofit
 
+    @pytest.fixture(params=[
+        ('w:tbl/w:tblPr', True,
+         'w:tbl/w:tblPr/w:tblLayout{w:type=autofit}'),
+        ('w:tbl/w:tblPr', False,
+         'w:tbl/w:tblPr/w:tblLayout{w:type=fixed}'),
+        ('w:tbl/w:tblPr', None,
+         'w:tbl/w:tblPr/w:tblLayout{w:type=fixed}'),
+        ('w:tbl/w:tblPr/w:tblLayout{w:type=fixed}', True,
+         'w:tbl/w:tblPr/w:tblLayout{w:type=autofit}'),
+        ('w:tbl/w:tblPr/w:tblLayout{w:type=autofit}', False,
+         'w:tbl/w:tblPr/w:tblLayout{w:type=fixed}'),
+    ])
+    def autofit_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', None),
         ('w:tbl/w:tblPr/w:tblStyle{w:val=foobar}', 'foobar'),
