shr: add ElementProxy.element

Steve Canny · Steve Canny · commit 576c047fd04b · 2014-12-27T22:07:12.000-08:00
diff --git a/docx/shared.py b/docx/shared.py
@@ -194,6 +194,13 @@ def __ne__(self, other):
             return True
         return self._element is not other._element
 
+    @property
+    def element(self):
+        """
+        The lxml element proxied by this object.
+        """
+        return self._element
+
 
 class Parented(object):
     """
diff --git a/tests/test_shared.py b/tests/test_shared.py
@@ -28,8 +28,18 @@ def it_knows_when_its_equal_to_another_proxy_object(self, eq_fixture):
         assert (proxy != proxy_3) is True
         assert (proxy != not_a_proxy) is True
 
+    def it_knows_its_element(self, element_fixture):
+        proxy, element = element_fixture
+        assert proxy.element is element
+
     # fixture --------------------------------------------------------
 
+    @pytest.fixture
+    def element_fixture(self):
+        p = element('w:p')
+        proxy = ElementProxy(p)
+        return proxy, p
+
     @pytest.fixture
     def eq_fixture(self):
         p, q = element('w:p'), element('w:p')
