tbl: add _Columns.table

apteryks · Steve Canny · commit 7e22dc8ab544 · 2014-11-28T21:45:37.000-05:00
diff --git a/docx/table.py b/docx/table.py
@@ -344,6 +344,13 @@ def __iter__(self):
     def __len__(self):
         return len(self._gridCol_lst)
 
+    @property
+    def table(self):
+        """
+        Reference to the |Table| object this column collection belongs to.
+        """
+        return self._parent.table
+
     @property
     def _gridCol_lst(self):
         """
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -523,6 +523,10 @@ def it_raises_on_indexed_access_out_of_range(self, columns_fixture):
         with pytest.raises(IndexError):
             columns[too_high]
 
+    def it_provides_access_to_the_table_it_belongs_to(self, table_fixture):
+        columns, table_ = table_fixture
+        assert columns.table is table_
+
     # fixtures -------------------------------------------------------
 
     @pytest.fixture
@@ -532,6 +536,18 @@ def columns_fixture(self):
         columns = _Columns(tbl, None)
         return columns, column_count
 
+    @pytest.fixture
+    def table_fixture(self, table_):
+        columns = _Columns(None, table_)
+        table_.table = table_
+        return columns, table_
+
+    # fixture components ---------------------------------------------
+
+    @pytest.fixture
+    def table_(self, request):
+        return instance_mock(request, Table)
+
 
 class Describe_Row(object):
 
