summaryrefslogtreecommitdiff
path: root/tests/test_mark.py
blob: f30a1217049b5c3c1a3b469cc8d9cd6058acb73d (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

import yaml

def test_marks(marks_filename, verbose=False):
    inputs = open(marks_filename, 'rb').read().split('---\n')[1:]
    for input in inputs:
        index = 0
        line = 0
        column = 0
        while input[index] != '*':
            if input[index] == '\n':
                line += 1
                column = 0
            else:
                column += 1
            index += 1
        mark = yaml.Mark(marks_filename, index, line, column, unicode(input), index)
        snippet = mark.get_snippet(indent=2, max_length=79)
        if verbose:
            print snippet
        assert isinstance(snippet, str), type(snippet)
        assert snippet.count('\n') == 1, snippet.count('\n')
        data, pointer = snippet.split('\n')
        assert len(data) < 82, len(data)
        assert data[len(pointer)-1] == '*', data[len(pointer)-1]

test_marks.unittest = ['.marks']

if __name__ == '__main__':
    import test_appliance
    test_appliance.run(globals())