tbl: add _Cell.tables

Steve Canny · Steve Canny · commit 8364dad0cc7b · 2014-07-18T00:29:27.000-07:00
diff --git a/docx/table.py b/docx/table.py
@@ -130,10 +130,17 @@ def paragraphs(self):
         """
         List of paragraphs in the cell. A table cell is required to contain
         at least one block-level element and end with a paragraph. By
-        default, a new cell contains a single paragraph.
+        default, a new cell contains a single paragraph. Read-only
         """
         return super(_Cell, self).paragraphs
 
+    @property
+    def tables(self):
+        """
+        List of tables in the cell, in the order they appear. Read-only.
+        """
+        return super(_Cell, self).tables
+
     @write_only_property
     def text(self, text):
         """
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -175,6 +175,18 @@ def it_provides_access_to_the_paragraphs_it_contains(
             count += 1
         assert count == 2
 
+    def it_provides_access_to_the_tables_it_contains(self, tables_fixture):
+        # test len(), iterable, and indexed access
+        cell, expected_count = tables_fixture
+        tables = cell.tables
+        assert len(tables) == expected_count
+        count = 0
+        for idx, table in enumerate(tables):
+            assert isinstance(table, Table)
+            assert tables[idx] is table
+            count += 1
+        assert count == expected_count
+
     def it_can_replace_its_content_with_a_string_of_text(
             self, text_set_fixture):
         cell, text, expected_xml = text_set_fixture
@@ -223,6 +235,18 @@ def add_table_fixture(self, request):
     def paragraphs_fixture(self):
         return _Cell(element('w:tc/(w:p, w:p)'), None)
 
+    @pytest.fixture(params=[
+        ('w:tc',                   0),
+        ('w:tc/w:tbl',             1),
+        ('w:tc/(w:tbl,w:tbl)',     2),
+        ('w:tc/(w:p,w:tbl)',       1),
+        ('w:tc/(w:tbl,w:tbl,w:p)', 2),
+    ])
+    def tables_fixture(self, request):
+        cell_cxml, expected_count = request.param
+        cell = _Cell(element(cell_cxml), None)
+        return cell, expected_count
+
     @pytest.fixture(params=[
         ('w:tc/w:p', 'foobar',
          'w:tc/w:p/w:r/w:t"foobar"'),
