tbl: add _Rows.table

Steve Canny · Steve Canny · commit f0ade55524fc · 2014-11-28T21:22:23.000-05:00
diff --git a/docx/table.py b/docx/table.py
@@ -386,3 +386,10 @@ def __iter__(self):
 
     def __len__(self):
         return len(self._tbl.tr_lst)
+
+    @property
+    def table(self):
+        """
+        Reference to the |Table| object this row collection belongs to.
+        """
+        return self._parent.table
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -547,6 +547,10 @@ def it_raises_on_indexed_access_out_of_range(self, rows_fixture):
             too_high = row_count
             rows[too_high]
 
+    def it_provides_access_to_the_table_it_belongs_to(self, table_fixture):
+        rows, table_ = table_fixture
+        assert rows.table is table_
+
     # fixtures -------------------------------------------------------
 
     @pytest.fixture
@@ -556,6 +560,18 @@ def rows_fixture(self):
         rows = _Rows(tbl, None)
         return rows, row_count
 
+    @pytest.fixture
+    def table_fixture(self, table_):
+        rows = _Rows(None, table_)
+        table_.table = table_
+        return rows, table_
+
+    # fixture components ---------------------------------------------
+
+    @pytest.fixture
+    def table_(self, request):
+        return instance_mock(request, Table)
+
 
 # fixtures -----------------------------------------------------------
 
