Sha256: e47c0cdf82bcede2616ba133ee40215eb3b3b355513d8567eed7908ae472fd12

Contents?: true

Size: 1.06 KB

Versions: 5

Compression:

Stored size: 1.06 KB

Contents

# -*- coding: utf-8 -*-
"""
    Pygments tests for using()
    ~~~~~~~~~~~~~~~~~~~~~~~~~~

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

import unittest

from pygments.lexer import using, bygroups, this, RegexLexer
from pygments.token import String, Text, Keyword

class TestLexer(RegexLexer):
    tokens = {
        'root': [
            (r'#.*',
             using(this, state='invalid')),
            (r'(")(.+?)(")',
             bygroups(String, using(this, state='string'), String)),
            (r'[^"]+', Text),
        ],
        'string': [
            (r'.+', Keyword),
        ],
    }


class UsingStateTest(unittest.TestCase):
    def test_basic(self):
        expected = [(Text, 'a'), (String, '"'), (Keyword, 'bcd'),
                    (String, '"'), (Text, 'e\n')]
        t = list(TestLexer().get_tokens('a"bcd"e'))
        self.assertEquals(t, expected)

    def test_error(self):
        def gen():
            return list(TestLexer().get_tokens('#a'))
        self.assertRaises(KeyError, gen)

Version data entries

5 entries across 5 versions & 1 rubygems

Version Path
pygments.rb-0.2.4 vendor/pygments-main/tests/test_using_api.py
pygments.rb-0.2.3 vendor/pygments-main/tests/test_using_api.py
pygments.rb-0.2.2 vendor/pygments-main/tests/test_using_api.py
pygments.rb-0.2.1 vendor/pygments-main/tests/test_using_api.py
pygments.rb-0.2.0 vendor/pygments-main/tests/test_using_api.py