style: add BaseStyle.builtin

Steve Canny · Steve Canny · commit 11efbb75c787 · 2015-02-08T00:01:39.000-08:00
diff --git a/docx/styles/style.py b/docx/styles/style.py
@@ -35,6 +35,14 @@ class BaseStyle(ElementProxy):
 
     __slots__ = ()
 
+    @property
+    def builtin(self):
+        """
+        Boolean indicating whether this style is a built-in style. False
+        indicates it is a custom (user-defined) style.
+        """
+        return not self._element.customStyle
+
     @property
     def name(self):
         """
diff --git a/features/sty-add-style.feature b/features/sty-add-style.feature
@@ -4,7 +4,6 @@ Feature: Add a style
   I need a way to add a new style
 
 
-  @wip
   Scenario Outline: Add a style
     Given a document having known styles
      When I call add_style('<name>', <type>, builtin=<builtin>)
diff --git a/tests/styles/test_style.py b/tests/styles/test_style.py
@@ -117,8 +117,22 @@ def it_can_change_its_name(self, name_set_fixture):
         style.name = new_value
         assert style._element.xml == expected_xml
 
+    def it_knows_whether_its_a_builtin_style(self, builtin_get_fixture):
+        style, expected_value = builtin_get_fixture
+        assert style.builtin is expected_value
+
     # fixture --------------------------------------------------------
 
+    @pytest.fixture(params=[
+        ('w:style',                  True),
+        ('w:style{w:customStyle=0}', True),
+        ('w:style{w:customStyle=1}', False),
+    ])
+    def builtin_get_fixture(self, request):
+        style_cxml, expected_value = request.param
+        style = BaseStyle(element(style_cxml))
+        return style, expected_value
+
     @pytest.fixture(params=[
         ('w:style',                   None),
         ('w:style{w:styleId=Foobar}', 'Foobar'),
