tbl: add _Cell.width setter

Steve Canny · Steve Canny · commit 6113e40453cf · 2014-07-13T17:12:59.000-07:00
diff --git a/docx/oxml/table.py b/docx/oxml/table.py
@@ -212,6 +212,11 @@ def width(self):
             return None
         return tcPr.width
 
+    @width.setter
+    def width(self, value):
+        tcPr = self.get_or_add_tcPr()
+        tcPr.width = value
+
 
 class CT_TcPr(BaseOxmlElement):
     """
diff --git a/docx/table.py b/docx/table.py
@@ -128,6 +128,10 @@ def width(self):
         """
         return self._tc.width
 
+    @width.setter
+    def width(self, value):
+        self._tc.width = value
+
 
 class _Column(Parented):
     """
diff --git a/features/tbl-cell-props.feature b/features/tbl-cell-props.feature
@@ -14,7 +14,6 @@ Feature: Get and set table cell properties
       | 1 inch              | 1 inch         |
 
 
-  @wip
   Scenario Outline: Set cell width
     Given a table cell having a width of <width-setting>
      When I set the cell width to <new-setting>
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -8,6 +8,7 @@
 
 import pytest
 
+from docx.shared import Inches
 from docx.table import (
     _Cell, _Column, _ColumnCells, _Columns, _Row, _RowCells, _Rows, Table
 )
@@ -172,6 +173,12 @@ def it_knows_its_width_in_EMU(self, width_get_fixture):
         cell, expected_width = width_get_fixture
         assert cell.width == expected_width
 
+    def it_can_change_its_width(self, width_set_fixture):
+        cell, value, expected_xml = width_set_fixture
+        cell.width = value
+        assert cell.width == value
+        assert cell._tc.xml == expected_xml
+
     # fixtures -------------------------------------------------------
 
     @pytest.fixture
@@ -203,6 +210,18 @@ def width_get_fixture(self, request):
         cell = _Cell(element(tc_cxml), None)
         return cell, expected_width
 
+    @pytest.fixture(params=[
+        ('w:tc', Inches(1),
+         'w:tc/w:tcPr/w:tcW{w:w=1440,w:type=dxa}'),
+        ('w:tc/w:tcPr/w:tcW{w:w=25%,w:type=pct}', Inches(2),
+         'w:tc/w:tcPr/w:tcW{w:w=2880,w:type=dxa}'),
+    ])
+    def width_set_fixture(self, request):
+        tc_cxml, new_value, expected_cxml = request.param
+        cell = _Cell(element(tc_cxml), None)
+        expected_xml = xml(expected_cxml)
+        return cell, new_value, expected_xml
+
 
 class Describe_Column(object):
 
