acpt: add InlineShape.type scenario

Steve Canny · Steve Canny · commit 3420abdb07cc · 2013-12-25T22:48:41.000-08:00
diff --git a/docx/enum/__init__.py b/docx/enum/__init__.py
@@ -0,0 +1,18 @@
+# encoding: utf-8
+
+"""
+Enumerations used in python-docx
+"""
+
+from __future__ import absolute_import, print_function, unicode_literals
+
+
+class Enumeration(object):
+
+    @classmethod
+    def from_xml(cls, xml_val):
+        return cls._xml_to_idx[xml_val]
+
+    @classmethod
+    def to_xml(cls, enum_val):
+        return cls._idx_to_xml[enum_val]
diff --git a/docx/enum/shape.py b/docx/enum/shape.py
@@ -0,0 +1,19 @@
+# encoding: utf-8
+
+"""
+Enumerations related to DrawingML shapes in WordprocessingML files
+"""
+
+from __future__ import absolute_import, print_function, unicode_literals
+
+
+class WD_INLINE_SHAPE_TYPE(object):
+    """
+    Corresponds to WdInlineShapeType enumeration
+    http://msdn.microsoft.com/en-us/library/office/ff192587.aspx
+    """
+    PICTURE = 3
+    LINKED_PICTURE = 4
+    NOT_IMPLEMENTED = -6
+
+WD_INLINE_SHAPE = WD_INLINE_SHAPE_TYPE
diff --git a/features/shp-inline-shape-access.feature b/features/shp-inline-shape-access.feature
@@ -12,3 +12,13 @@ Feature: Access inline shapes in document
      Given an inline shape collection containing two shapes
       Then I can iterate over the inline shape collection
        And I can access an inline shape by index
+
+  @wip
+  Scenario Outline: Identify type of inline shape
+     Given an inline shape known to be <shape of type>
+      Then its inline shape type is <shape type>
+
+   Examples: Inline shapes of recognized types
+     | shape of type       | shape type                     |
+     | an embedded picture | WD_INLINE_SHAPE.PICTURE        |
+     | a linked picture    | WD_INLINE_SHAPE.LINKED_PICTURE |
diff --git a/features/steps/shape.py b/features/steps/shape.py
@@ -9,6 +9,7 @@
 from behave import given, then
 
 from docx import Document
+from docx.enum.shape import WD_INLINE_SHAPE
 from docx.parts import InlineShape, InlineShapes
 
 from .helpers import test_docx
@@ -29,6 +30,17 @@ def given_inline_shape_collection_containing_two_shapes(context):
     context.inline_shapes = document.inline_shapes
 
 
+@given('an inline shape known to be {shp_of_type}')
+def given_inline_shape_known_to_be_shape_of_type(context, shp_of_type):
+    inline_shape_idx = {
+        'an embedded picture': 0,
+        'a linked picture':    1,
+    }[shp_of_type]
+    docx_path = test_docx('shp-inline-shape-access')
+    document = Document(docx_path)
+    context.inline_shape = document.inline_shapes[inline_shape_idx]
+
+
 # then =====================================================
 
 @then('I can access an inline shape by index')
@@ -56,6 +68,16 @@ def then_can_iterate_over_inline_shape_collection(context):
     assert actual_count == 2
 
 
+@then('its inline shape type is {shape_type}')
+def then_inline_shape_type_is_shape_type(context, shape_type):
+    expected_value = {
+        'WD_INLINE_SHAPE.PICTURE':        WD_INLINE_SHAPE.PICTURE,
+        'WD_INLINE_SHAPE.LINKED_PICTURE': WD_INLINE_SHAPE.LINKED_PICTURE,
+    }[shape_type]
+    inline_shape = context.inline_shape
+    assert inline_shape.type == expected_value
+
+
 @then('the length of the inline shape collection is 2')
 def then_len_of_inline_shape_collection_is_2(context):
     inline_shapes = context.document.inline_shapes
diff --git a/features/steps/test_files/monty-truth.png b/features/steps/test_files/monty-truth.png
diff --git a/features/steps/test_files/shp-inline-shape-access.docx b/features/steps/test_files/shp-inline-shape-access.docx
