tbl: base _ColumnCells on Parented

Steve Canny · Steve Canny · commit ca211caa9e9a · 2014-07-11T17:15:18.000-07:00
diff --git a/docx/table.py b/docx/table.py
@@ -124,7 +124,7 @@ def cells(self):
         Sequence of |_Cell| instances corresponding to cells in this column.
         Supports ``len()``, iteration and indexed access.
         """
-        return _ColumnCells(self._tbl, self._gridCol)
+        return _ColumnCells(self._tbl, self._gridCol, self)
 
     @property
     def width(self):
@@ -139,13 +139,13 @@ def width(self, value):
         self._gridCol.w = value
 
 
-class _ColumnCells(object):
+class _ColumnCells(Parented):
     """
     Sequence of |_Cell| instances corresponding to the cells in a table
     column.
     """
-    def __init__(self, tbl, gridCol):
-        super(_ColumnCells, self).__init__()
+    def __init__(self, tbl, gridCol, parent):
+        super(_ColumnCells, self).__init__(parent)
         self._tbl = tbl
         self._gridCol = gridCol
 
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -232,7 +232,7 @@ def cells_fixture(self):
         cell_count = 2
         tbl = _tbl_bldr(rows=cell_count, cols=1).element
         gridCol = tbl.tblGrid.gridCol_lst[0]
-        cells = _ColumnCells(tbl, gridCol)
+        cells = _ColumnCells(tbl, gridCol, None)
         return cells, cell_count
 
 
