tbl: add _Column.table

apteryks · Steve Canny · commit dc3e8e708866 · 2014-11-28T21:45:07.000-05:00
diff --git a/docx/table.py b/docx/table.py
@@ -255,7 +255,7 @@ def table(self):
         """
         Reference to the |Table| object this column belongs to.
         """
-        raise NotImplementedError
+        return self._parent.table
 
     @property
     def width(self):
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -380,6 +380,10 @@ def it_provides_access_to_its_cells(self, cells_fixture):
         column.table.column_cells.assert_called_once_with(column_idx)
         assert cells == expected_cells
 
+    def it_provides_access_to_the_table_it_belongs_to(self, table_fixture):
+        column, table_ = table_fixture
+        assert column.table is table_
+
     def it_knows_its_width_in_EMU(self, width_get_fixture):
         column, expected_width = width_get_fixture
         assert column.width == expected_width
@@ -400,6 +404,12 @@ def cells_fixture(self, _index_, table_prop_, table_):
         table_.column_cells.return_value = list(expected_cells)
         return column, column_idx, expected_cells
 
+    @pytest.fixture
+    def table_fixture(self, parent_, table_):
+        column = _Column(None, None, parent_)
+        parent_.table = table_
+        return column, table_
+
     @pytest.fixture(params=[
         ('w:gridCol{w:w=4242}',   2693670),
         ('w:gridCol{w:w=1440}',    914400),
@@ -431,6 +441,10 @@ def width_set_fixture(self, request):
     def _index_(self, request):
         return property_mock(request, _Column, '_index')
 
+    @pytest.fixture
+    def parent_(self, request):
+        return instance_mock(request, Table)
+
     @pytest.fixture
     def table_(self, request):
         return instance_mock(request, Table)
