# encoding: utf-8
"""
Test suite for the docx.oxml.styles module.
"""
from __future__ import (
absolute_import, division, print_function, unicode_literals
)
import pytest
from docx.enum.style import WD_STYLE_TYPE
from ..unitutil.cxml import element, xml
class DescribeCT_Styles(object):
def it_can_add_a_style_of_type(self, add_fixture):
styles, name, style_type, builtin, expected_xml = add_fixture
style = styles.add_style_of_type(name, style_type, builtin)
assert styles.xml == expected_xml
assert style is styles[-1]
# fixtures -------------------------------------------------------
@pytest.fixture(params=[
('w:styles', 'Foo Bar', WD_STYLE_TYPE.LIST, False,
'w:styles/w:style{w:type=numbering,w:customStyle=1,w:styleId=FooBar'
'}/w:name{w:val=Foo Bar}'),
('w:styles', 'heading 1', WD_STYLE_TYPE.PARAGRAPH, True,
'w:styles/w:style{w:type=paragraph,w:styleId=Heading1}/w:name{w:val'
'=heading 1}'),
])
def add_fixture(self, request):
styles_cxml, name, style_type, builtin, expected_cxml = request.param
styles = element(styles_cxml)
expected_xml = xml(expected_cxml)
return styles, name, style_type, builtin, expected_xml