acpt: rewrite scenario for Document.add_table()

Steve Canny · Steve Canny · commit 74931e2c9ba9 · 2015-02-21T14:11:35.000-08:00
Add steps to check column and cell width are properly applied to a new
table.
diff --git a/features/doc-add-table.feature b/features/doc-add-table.feature
@@ -4,11 +4,14 @@ Feature: Add a table
   I need a way to add a table
 
 
+  @wip
   Scenario: Add a table specifying only row and column count
     Given a blank document
      When I add a 2 x 2 table specifying only row and column count
      Then the document contains a 2 x 2 table
       And table.style is styles['Normal Table']
+      And the width of each column is 3.0 inches
+      And the width of each cell is 3.0 inches
 
 
   Scenario: Add a table specifying style
diff --git a/features/steps/table.py b/features/steps/table.py
@@ -358,3 +358,19 @@ def _cell(table, idx):
     idx, inches = int(n_str) - 1, float(inches_str)
     cell = _cell(context.table_, idx)
     assert cell.width == Inches(inches), 'got %s' % cell.width.inches
+
+
+@then('the width of each cell is {inches} inches')
+def then_the_width_of_each_cell_is_inches(context, inches):
+    table = context.table_
+    expected_width = Inches(float(inches))
+    for cell in table._cells:
+        assert cell.width == expected_width, 'got %s' % cell.width.inches
+
+
+@then('the width of each column is {inches} inches')
+def then_the_width_of_each_column_is_inches(context, inches):
+    table = context.table_
+    expected_width = Inches(float(inches))
+    for column in table.columns:
+        assert column.width == expected_width, 'got %s' % column.width.inches
