acpt: add scenario for Document.tables

Steve Canny · Steve Canny · commit a9608c8edc20 · 2015-02-15T12:28:30.000-08:00
* Improve analogous step for Document.paragraphs
diff --git a/features/doc-access-collections.feature b/features/doc-access-collections.feature
@@ -22,3 +22,8 @@ Feature: Access document collections
   Scenario: Access the styles collection of a document
     Given a document having styles
      Then document.styles is a Styles object
+
+
+  Scenario: Access the tables collection of a document
+    Given a document having three tables
+     Then document.tables is a list containing three tables
diff --git a/features/steps/document.py b/features/steps/document.py
@@ -15,6 +15,7 @@
 from docx.section import Sections
 from docx.styles.styles import Styles
 from docx.table import Table
+from docx.text.paragraph import Paragraph
 
 from helpers import test_docx, test_file
 
@@ -41,6 +42,11 @@ def given_a_document_having_styles(context):
     context.document = Document(test_docx('sty-having-styles-part'))
 
 
+@given('a document having three tables')
+def given_a_document_having_three_tables(context):
+    context.document = Document(test_docx('tbl-having-tables'))
+
+
 @given('a single-section document having portrait layout')
 def given_a_single_section_document_having_portrait_layout(context):
     context.document = Document(test_docx('doc-add-section'))
@@ -164,6 +170,8 @@ def then_document_paragraphs_is_a_list_containing_three_paragraphs(context):
     paragraphs = document.paragraphs
     assert isinstance(paragraphs, list)
     assert len(paragraphs) == 3
+    for paragraph in paragraphs:
+        assert isinstance(paragraph, Paragraph)
 
 
 @then('document.sections is a Sections object')
@@ -179,6 +187,16 @@ def then_document_styles_is_a_Styles_object(context):
     assert isinstance(styles, Styles)
 
 
+@then('document.tables is a list containing three tables')
+def then_document_tables_is_a_list_containing_three_tables(context):
+    document = context.document
+    tables = document.tables
+    assert isinstance(tables, list)
+    assert len(tables) == 3
+    for table in tables:
+        assert isinstance(table, Table)
+
+
 @then('the document contains a 2 x 2 table')
 def then_document_contains_2x2_table(context):
     document = context.document
diff --git a/features/steps/test_files/tbl-having-tables.docx b/features/steps/test_files/tbl-having-tables.docx
