diff options
Diffstat (limited to 'tests/test_using_api.py')
-rw-r--r-- | tests/test_using_api.py | 21 |
1 files changed, 10 insertions, 11 deletions
diff --git a/tests/test_using_api.py b/tests/test_using_api.py index 35546508..616bff6f 100644 --- a/tests/test_using_api.py +++ b/tests/test_using_api.py @@ -7,7 +7,7 @@ :license: BSD, see LICENSE for details. """ -import unittest +from pytest import raises from pygments.lexer import using, bygroups, this, RegexLexer from pygments.token import String, Text, Keyword @@ -28,14 +28,13 @@ class MyLexer(RegexLexer): } -class UsingStateTest(unittest.TestCase): - def test_basic(self): - expected = [(Text, 'a'), (String, '"'), (Keyword, 'bcd'), - (String, '"'), (Text, 'e\n')] - t = list(MyLexer().get_tokens('a"bcd"e')) - self.assertEqual(t, expected) +def test_basic(): + expected = [(Text, 'a'), (String, '"'), (Keyword, 'bcd'), + (String, '"'), (Text, 'e\n')] + assert list(MyLexer().get_tokens('a"bcd"e')) == expected - def test_error(self): - def gen(): - return list(MyLexer().get_tokens('#a')) - self.assertRaises(KeyError, gen) + +def test_error(): + def gen(): + return list(MyLexer().get_tokens('#a')) + assert raises(KeyError, gen) |