acpt: add scenarios for Table.alignment

Steve Canny · Steve Canny · commit e22465d048a0 · 2014-12-01T18:23:22.000-05:00
diff --git a/docs/api/enum/WdRowAlignment.rst b/docs/api/enum/WdRowAlignment.rst
@@ -0,0 +1,24 @@
+.. _WdRowAlignment:
+
+``WD_TABLE_ALIGNMENT``
+======================
+
+Specifies table justification type.
+
+Example::
+
+    from docx.enum.table import WD_TABLE_ALIGNMENT
+
+    table = document.add_table(3, 3)
+    table.alignment = WD_TABLE_ALIGNMENT.CENTER
+
+----
+
+LEFT
+    Left-aligned
+
+CENTER
+    Center-aligned.
+
+RIGHT
+    Right-aligned.
diff --git a/docs/api/enum/index.rst b/docs/api/enum/index.rst
@@ -11,4 +11,5 @@ can be found here:
    WdAlignParagraph
    WdOrientation
    WdSectionStart
+   WdRowAlignment
    WdUnderline
diff --git a/docx/enum/table.py b/docx/enum/table.py
@@ -0,0 +1,38 @@
+# encoding: utf-8
+
+"""
+Enumerations related to tables in WordprocessingML files
+"""
+
+from __future__ import absolute_import, print_function, unicode_literals
+
+from .base import XmlEnumeration, XmlMappedEnumMember
+
+
+class WD_TABLE_ALIGNMENT(XmlEnumeration):
+    """
+    Specifies table justification type.
+
+    Example::
+
+        from docx.enum.table import WD_TABLE_ALIGNMENT
+
+        table = document.add_table(3, 3)
+        table.alignment = WD_TABLE_ALIGNMENT.CENTER
+    """
+
+    __ms_name__ = 'WdRowAlignment'
+
+    __url__ = ' http://office.microsoft.com/en-us/word-help/HV080607259.aspx'
+
+    __members__ = (
+        XmlMappedEnumMember(
+            'LEFT', 0, 'left', 'Left-aligned'
+        ),
+        XmlMappedEnumMember(
+            'CENTER', 1, 'center', 'Center-aligned.'
+        ),
+        XmlMappedEnumMember(
+            'RIGHT', 2, 'right', 'Right-aligned.'
+        ),
+    )
diff --git a/features/steps/table.py b/features/steps/table.py
@@ -11,6 +11,7 @@
 from behave import given, then, when
 
 from docx import Document
+from docx.enum.table import WD_TABLE_ALIGNMENT
 from docx.shared import Inches
 from docx.table import _Column, _Columns, _Row, _Rows
 
@@ -75,6 +76,19 @@ def given_a_table_having_a_width_of_width_desc(context, width_desc):
     context.column = document.tables[0].columns[col_idx]
 
 
+@given('a table having {alignment} alignment')
+def given_a_table_having_alignment_alignment(context, alignment):
+    table_idx = {
+        'inherited': 3,
+        'left':      4,
+        'right':     5,
+        'center':    6,
+    }[alignment]
+    docx_path = test_docx('tbl-props')
+    document = Document(docx_path)
+    context.table_ = document.tables[table_idx]
+
+
 @given('a table having an applied style')
 def given_a_table_having_an_applied_style(context):
     docx_path = test_docx('tbl-having-applied-style')
@@ -129,6 +143,18 @@ def when_apply_style_to_table(context):
     table.style = 'LightShading-Accent1'
 
 
+@when('I assign {value_str} to table.alignment')
+def when_I_assign_value_to_table_alignment(context, value_str):
+    value = {
+        'None':                      None,
+        'WD_TABLE_ALIGNMENT.LEFT':   WD_TABLE_ALIGNMENT.LEFT,
+        'WD_TABLE_ALIGNMENT.RIGHT':  WD_TABLE_ALIGNMENT.RIGHT,
+        'WD_TABLE_ALIGNMENT.CENTER': WD_TABLE_ALIGNMENT.CENTER,
+    }[value_str]
+    table = context.table_
+    table.alignment = value
+
+
 @when('I merge from cell {origin} to cell {other}')
 def when_I_merge_from_cell_origin_to_cell_other(context, origin, other):
     def cell(table, idx):
@@ -218,6 +244,18 @@ def then_can_iterate_over_row_collection(context):
     assert actual_count == 2
 
 
+@then('table.alignment is {value_str}')
+def then_table_alignment_is_value(context, value_str):
+    value = {
+        'None':                      None,
+        'WD_TABLE_ALIGNMENT.LEFT':   WD_TABLE_ALIGNMENT.LEFT,
+        'WD_TABLE_ALIGNMENT.RIGHT':  WD_TABLE_ALIGNMENT.RIGHT,
+        'WD_TABLE_ALIGNMENT.CENTER': WD_TABLE_ALIGNMENT.CENTER,
+    }[value_str]
+    table = context.table_
+    assert table.alignment == value, 'got %s' % table.alignment
+
+
 @then('table.cell({row}, {col}).text is {expected_text}')
 def then_table_cell_row_col_text_is_text(context, row, col, expected_text):
     table = context.table_
diff --git a/features/steps/test_files/tbl-props.docx b/features/steps/test_files/tbl-props.docx
diff --git a/features/tbl-props.feature b/features/tbl-props.feature
@@ -4,6 +4,33 @@ Feature: Get and set table properties
   I need a way to get and set a table's properties
 
 
+  @wip
+  Scenario Outline: Determine table alignment
+    Given a table having <alignment> alignment
+     Then table.alignment is <value>
+
+    Examples: table alignment settings
+      | alignment | value                     |
+      | inherited | None                      |
+      | left      | WD_TABLE_ALIGNMENT.LEFT   |
+      | right     | WD_TABLE_ALIGNMENT.RIGHT  |
+      | center    | WD_TABLE_ALIGNMENT.CENTER |
+
+
+  @wip
+  Scenario Outline: Set table alignment
+    Given a table having <alignment> alignment
+     When I assign <value> to table.alignment
+     Then table.alignment is <value>
+
+    Examples: results of assignment to table.alignment
+      | alignment | value                     |
+      | inherited | WD_TABLE_ALIGNMENT.LEFT   |
+      | left      | WD_TABLE_ALIGNMENT.RIGHT  |
+      | right     | WD_TABLE_ALIGNMENT.CENTER |
+      | center    | None                      |
+
+
   Scenario Outline: Get autofit layout setting
     Given a table having an autofit layout of <autofit-setting>
      Then the reported autofit setting is <reported-autofit>
