diff --git a/IPython/lib/tests/test_lexers.py b/IPython/lib/tests/test_lexers.py index 0f79550..bb2de2e 100644 --- a/IPython/lib/tests/test_lexers.py +++ b/IPython/lib/tests/test_lexers.py @@ -5,6 +5,7 @@ from unittest import TestCase from pygments.token import Token +from pygments.lexers import BashLexer from .. import lexers @@ -13,16 +14,14 @@ class TestLexers(TestCase): """Collection of lexers tests""" def setUp(self): self.lexer = lexers.IPythonLexer() + self.bash_lexer = BashLexer() def testIPythonLexer(self): fragment = '!echo $HOME\n' tokens = [ (Token.Operator, '!'), - (Token.Name.Builtin, 'echo'), - (Token.Text, ' '), - (Token.Name.Variable, '$HOME'), - (Token.Text, '\n'), ] + tokens.extend(self.bash_lexer.get_tokens(fragment[1:])) self.assertEqual(tokens, list(self.lexer.get_tokens(fragment))) fragment_2 = '!' + fragment