summaryrefslogtreecommitdiff
path: root/tests/test_robotframework_lexer.py
blob: 807fbc4e31ffa26807453de38738b3f4c7b1536e (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
37
38
"""
    Pygments Robot Framework lexer tests
    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

    :copyright: Copyright 2006-2022 by the Pygments team, see AUTHORS.
    :license: BSD, see LICENSE for details.
"""

import pytest

from pygments.lexers.robotframework import RobotFrameworkLexer


@pytest.fixture(scope='module')
def lexer():
    yield RobotFrameworkLexer()


def assert_same_text(lexer, text):
    """Show that lexed text does not remove any content. """
    tokens = list(lexer.get_tokens_unprocessed(text))
    output = ''.join(t[2] for t in tokens)
    assert text == output


def test_empty_brackets_after_scalar_variable(lexer):
    assert_same_text(lexer, '*** Variables ***\n'
                            '${test}[]\n')


def test_empty_brackets_after_list_variable(lexer):
    assert_same_text(lexer, '*** Variables ***\n'
                            '@{test}[]\n')


def test_empty_brackets_after_dict_variable(lexer):
    assert_same_text(lexer, '*** Variables ***\n'
                            '&{test}[]\n')