summaryrefslogtreecommitdiff
path: root/tests/test_objectiveclexer.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_objectiveclexer.py')
-rw-r--r--tests/test_objectiveclexer.py13
1 files changed, 12 insertions, 1 deletions
diff --git a/tests/test_objectiveclexer.py b/tests/test_objectiveclexer.py
index aee7db66..faadb08f 100644
--- a/tests/test_objectiveclexer.py
+++ b/tests/test_objectiveclexer.py
@@ -69,7 +69,7 @@ class ObjectiveCLexerTest(unittest.TestCase):
]
self.assertEqual(expected, list(self.lexer.get_tokens(fragment)))
- def testLieralNumberBoolExpression(self):
+ def testLiteralNumberBoolExpression(self):
fragment = u'@(YES);\n'
expected = [
(Token.Literal, u'@('),
@@ -79,3 +79,14 @@ class ObjectiveCLexerTest(unittest.TestCase):
(Token.Text, u'\n'),
]
self.assertEqual(expected, list(self.lexer.get_tokens(fragment)))
+
+ def testModuleImport(self):
+ fragment = u'@import ModuleA;\n'
+ expected = [
+ (Token.Keyword, u'@import'),
+ (Token.Text, u' '),
+ (Token.Name, u'ModuleA'),
+ (Token.Punctuation, u';'),
+ (Token.Text, u'\n'),
+ ]
+ self.assertEqual(expected, list(self.lexer.get_tokens(fragment)))