Skip to content

Commit 42f85a3

Browse files
refactor(test): Move tokenizer tests to test/test_tokenizer/
1 parent f5e0042 commit 42f85a3

File tree

3 files changed

+19
-27
lines changed

3 files changed

+19
-27
lines changed

test/test_integration/test_ws_at_end_issue_001.py

Lines changed: 0 additions & 25 deletions
This file was deleted.

test/test_tokenizer/__init__.py

Whitespace-only changes.
Lines changed: 19 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,14 @@
11
import unittest
22

33
from parser.lexer import Tokenizer
4-
from parser.lexer.tokens import WhitespaceToken, StringToken, EofToken, NumberToken
4+
from parser.lexer.tokens import (
5+
WhitespaceToken, StringToken, EofToken, NumberToken, SemicolonToken)
56
from parser.common import StrRegion
67
from parser.tokens import IdentNameToken, DotToken, AttrNameToken, OpToken
78
from test.common import CommonTestCase, TokenStreamFlag
89

910

10-
class MyTestCase(CommonTestCase):
11+
class TestInternalFuncs(CommonTestCase):
1112
def test__t_ident_name__at_end(self):
1213
t = Tokenizer('abc')
1314
end = t._t_ident_name(0)
@@ -63,6 +64,8 @@ def test__t_number(self):
6364
self.assertTokensEqual(t, [NumberToken(StrRegion(0, 2))])
6465
self.assertEqual(end, 2)
6566

67+
68+
class TestFullTokenizer(CommonTestCase):
6669
def test_mod_supported(self):
6770
t = Tokenizer('a+b%2')
6871
t.tokenize()
@@ -93,6 +96,20 @@ def test_tokenize_concat_works(self):
9396
EofToken(StrRegion(9, 9)),
9497
], TokenStreamFlag.CONTENT)
9598

99+
def test_ws_at_end(self):
100+
t = Tokenizer('let a =1; \n').tokenize()
101+
self.assertTokensEqual(t, [
102+
IdentNameToken(),
103+
WhitespaceToken(),
104+
IdentNameToken(),
105+
WhitespaceToken(),
106+
OpToken(op_str='='),
107+
NumberToken(),
108+
SemicolonToken(),
109+
WhitespaceToken(),
110+
EofToken()
111+
], TokenStreamFlag.FULL, check_regions=False)
112+
96113

97114
if __name__ == '__main__':
98115
unittest.main()

0 commit comments

Comments
 (0)