diff options
-rw-r--r-- | pygments/lexers/objective.py | 2 | ||||
-rw-r--r-- | tests/test_objectiveclexer.py | 13 |
2 files changed, 13 insertions, 2 deletions
diff --git a/pygments/lexers/objective.py b/pygments/lexers/objective.py index 179928e9..1be28c55 100644 --- a/pygments/lexers/objective.py +++ b/pygments/lexers/objective.py @@ -66,7 +66,7 @@ def objective(baselexer): 'copy', 'retain', 'assign', 'unsafe_unretained', 'atomic', 'nonatomic', 'readonly', 'readwrite', 'setter', 'getter', 'typeof', 'in', 'out', 'inout', 'release', 'class', '@dynamic', '@optional', - '@required', '@autoreleasepool'), suffix=r'\b'), + '@required', '@autoreleasepool', '@import'), suffix=r'\b'), Keyword), (words(('id', 'instancetype', 'Class', 'IMP', 'SEL', 'BOOL', 'IBOutlet', 'IBAction', 'unichar'), suffix=r'\b'), 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))) |