tbl: add _Row.cells

Steve Canny · Steve Canny · commit 8e9f36ad76e5 · 2013-12-22T00:00:15.000-08:00
diff --git a/docx/table.py b/docx/table.py
@@ -52,6 +52,9 @@ class _CellCollection(object):
     """
     Sequence of |_Cell| instances corresponding to the cells in a table row.
     """
+    def __init__(self, tr):
+        super(_CellCollection, self).__init__()
+        self._tr = tr
 
 
 class _Column(object):
@@ -71,6 +74,13 @@ def __init__(self, tr):
         super(_Row, self).__init__()
         self._tr = tr
 
+    @property
+    def cells(self):
+        """
+        Sequence of |_Cell| instances corresponding to the cells in this row.
+        """
+        return _CellCollection(self._tr)
+
 
 class _RowCollection(object):
     """
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -8,7 +8,7 @@
 
 import pytest
 
-from docx.table import _Column, _Row, _RowCollection, Table
+from docx.table import _CellCollection, _Column, _Row, _RowCollection, Table
 
 from .oxml.unitdata.table import a_gridCol, a_tbl, a_tblGrid, a_tc, a_tr
 from .oxml.unitdata.text import a_p
@@ -56,6 +56,22 @@ def row_access_fixture(self):
         return table
 
 
+class Describe_Row(object):
+
+    def it_provides_access_to_the_row_cells(self, cells_access_fixture):
+        row = cells_access_fixture
+        cells = row.cells
+        assert isinstance(cells, _CellCollection)
+
+    # fixtures -------------------------------------------------------
+
+    @pytest.fixture
+    def cells_access_fixture(self):
+        tr = a_tr().with_nsdecls().element
+        row = _Row(tr)
+        return row
+
+
 class Describe_RowCollection(object):
 
     def it_contains__Row_instances(self, row_count_fixture):
