acpt: add scenario for _Cell.add_table()

Steve Canny · Steve Canny · commit ebd5e85927e9 · 2015-02-21T14:11:35.000-08:00
diff --git a/features/cel-add-table.feature b/features/cel-add-table.feature
@@ -0,0 +1,13 @@
+Feature: Add a table into a table cell
+  In order to nest a table within a table cell
+  As a developer using python-docx
+  I need a way to add a table to a table cell
+
+
+  @wip
+  Scenario: Add a table into a table cell
+    Given a table cell
+     When I add a 2 x 2 table into the first cell
+     Then cell.tables[0] is a 2 x 2 table
+      And the width of each column is 1.5375 inches
+      And the width of each cell is 1.5375 inches
diff --git a/features/steps/cell.py b/features/steps/cell.py
@@ -10,17 +10,24 @@
 
 from docx import Document
 
+from helpers import test_docx
+
 
 # given ===================================================
 
 @given('a table cell')
 def given_a_table_cell(context):
-    table = Document().add_table(rows=2, cols=2)
+    table = Document(test_docx('tbl-2x2-table')).tables[0]
     context.cell = table.cell(0, 0)
 
 
 # when =====================================================
 
+@when('I add a 2 x 2 table into the first cell')
+def when_I_add_a_2x2_table_into_the_first_cell(context):
+    context.table_ = context.cell.add_table(2, 2)
+
+
 @when('I assign a string to the cell text attribute')
 def when_assign_string_to_cell_text_attribute(context):
     cell = context.cell
@@ -31,6 +38,14 @@ def when_assign_string_to_cell_text_attribute(context):
 
 # then =====================================================
 
+@then('cell.tables[0] is a 2 x 2 table')
+def then_cell_tables_0_is_a_2x2_table(context):
+    cell = context.cell
+    table = cell.tables[0]
+    assert len(table.rows) == 2
+    assert len(table.columns) == 2
+
+
 @then('the cell contains the string I assigned')
 def then_cell_contains_string_assigned(context):
     cell, expected_text = context.cell, context.expected_text
diff --git a/features/steps/test_files/tbl-2x2-table.docx b/features/steps/test_files/tbl-2x2-table.docx
