acpt: add Row.cells access scenario

Steve Canny · Steve Canny · commit cdb413c0255c · 2013-12-21T23:51:33.000-08:00
diff --git a/docx/table.py b/docx/table.py
@@ -42,6 +42,18 @@ def rows(self):
         return _RowCollection(self._tbl)
 
 
+class _Cell(object):
+    """
+    Table cell
+    """
+
+
+class _CellCollection(object):
+    """
+    Sequence of |_Cell| instances corresponding to the cells in a table row.
+    """
+
+
 class _Column(object):
     """
     Table column
diff --git a/features/steps/table.py b/features/steps/table.py
@@ -9,7 +9,7 @@
 from behave import given, then
 
 from docx import Document
-from docx.table import _Row
+from docx.table import _Cell, _CellCollection, _Row
 
 from .helpers import test_docx
 
@@ -23,8 +23,30 @@ def given_a_table_having_two_rows(context):
     context.table_ = document.body.tables[0]
 
 
+@given('a table row having two cells')
+def given_a_table_row_having_two_cells(context):
+    docx_path = test_docx('blk-containing-table')
+    document = Document(docx_path)
+    context.row = document.body.tables[0].rows[0]
+
+
 # then =====================================================
 
+@then('I can access a collection cell by index')
+def then_can_access_collection_cell_by_index(context):
+    cells = context.row.cells
+    for idx in range(2):
+        cell = cells[idx]
+        assert isinstance(cell, _Cell)
+
+
+@then('I can access the cell collection of the row')
+def then_can_access_cell_collection_of_row(context):
+    row = context.row
+    cells = row.cells
+    assert isinstance(cells, _CellCollection)
+
+
 @then('I can access the rows by index')
 def then_can_access_rows_by_index(context):
     rows = context.table_.rows
@@ -34,6 +56,23 @@ def then_can_access_rows_by_index(context):
         assert isinstance(row, _Row)
 
 
+@then('I can get the length of the cell collection')
+def then_can_get_length_of_cell_collection(context):
+    row = context.row
+    cells = row.cells
+    assert len(cells) == 2
+
+
+@then('I can iterate over the cell collection')
+def then_can_iterate_over_cell_collection(context):
+    row = context.row
+    actual_count = 0
+    for cell in row.cells:
+        actual_count += 1
+        assert isinstance(cell, _Cell)
+    assert actual_count == 2
+
+
 @then('the length of its row collection is 2')
 def then_len_of_row_collection_is_2(context):
     rows = context.table_.rows
diff --git a/features/tbl-item-access.feature b/features/tbl-item-access.feature
@@ -11,3 +11,11 @@ Feature: Access table rows, columns, and cells
   Scenario: Access table row by collection index
      Given a table having two rows
       Then I can access the rows by index
+
+  @wip
+  Scenario: Access cell collection of table row
+     Given a table row having two cells
+      Then I can access the cell collection of the row
+       And I can get the length of the cell collection
+       And I can iterate over the cell collection
+       And I can access a collection cell by index
