summaryrefslogtreecommitdiff
path: root/tests/options_tests.py
blob: f0b0beb6f5cb8c13f97da9becb04aba595abb61c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
from nose.tools import istest, assert_equal

from mammoth import style_reader
from mammoth.options import read_options, _default_style_map


@istest
def default_style_map_is_used_if_style_map_is_not_set():
    assert_equal(_default_style_map, read_options({}).value["style_map"])


@istest
def custom_style_mappings_are_prepended_to_default_style_mappings():
    style_map = read_options({
        "style_map": "p.SectionTitle => h2"
    }).value["style_map"]
    assert_equal(style_reader.read_style("p.SectionTitle => h2").value, style_map[0])
    assert_equal(_default_style_map, style_map[1:])


@istest
def default_style_mappings_are_ignored_if_include_default_style_map_is_false():
    style_map = read_options({
        "style_map": "p.SectionTitle => h2",
        "include_default_style_map": False
    }).value["style_map"]
    assert_equal([style_reader.read_style("p.SectionTitle => h2").value], style_map)


@istest
def lines_starting_with_hash_in_custom_style_map_are_ignored():
    style_map = read_options({
        "style_map": "#p.SectionTitle => h3\np.SectionTitle => h2",
        "include_default_style_map": False
    }).value["style_map"]
    assert_equal([style_reader.read_style("p.SectionTitle => h2").value], style_map)