diff options
Diffstat (limited to 'tests/test_objectiveclexer.py')
-rw-r--r-- | tests/test_objectiveclexer.py | 13 |
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))) |