1- import unittest
2-
31from fluent .pygments .lexer import FluentLexer
42from pygments .token import Token
53
64
7- class LexerTest (unittest .TestCase ):
8- def setUp (self ):
9- self .lexer = FluentLexer ()
10-
5+ class TestLexer :
116 def test_comment (self ):
127 fragment = "# comment\n "
138 tokens = [
149 (Token .Comment .Multiline , "# comment" ),
1510 (Token .Punctuation , "\n " ),
1611 ]
17- self . assertEqual ( tokens , list (self . lexer . get_tokens (fragment )))
12+ assert list (FluentLexer (). get_tokens (fragment )) == tokens
1813
1914 def test_message (self ):
2015 fragment = "msg = some value\n "
@@ -24,7 +19,7 @@ def test_message(self):
2419 (Token .Literal , "some value" ),
2520 (Token .Punctuation , "\n " ),
2621 ]
27- self . assertEqual ( tokens , list (self . lexer . get_tokens (fragment )))
22+ assert list (FluentLexer (). get_tokens (fragment )) == tokens
2823
2924 def test_message_with_comment (self ):
3025 fragment = "# good comment\n msg = some value\n "
@@ -36,4 +31,4 @@ def test_message_with_comment(self):
3631 (Token .Literal , "some value" ),
3732 (Token .Punctuation , "\n " ),
3833 ]
39- self . assertEqual ( tokens , list (self . lexer . get_tokens (fragment )))
34+ assert list (FluentLexer (). get_tokens (fragment )) == tokens
0 commit comments