|
|
# Standard library imports
|
|
|
import unittest
|
|
|
|
|
|
# System library imports
|
|
|
from pygments.lexers import CLexer, CppLexer, PythonLexer
|
|
|
|
|
|
# Local imports
|
|
|
from IPython.frontend.qt.console.completion_lexer import CompletionLexer
|
|
|
|
|
|
|
|
|
class TestCompletionLexer(unittest.TestCase):
|
|
|
|
|
|
def testPython(self):
|
|
|
""" Does the CompletionLexer work for Python?
|
|
|
"""
|
|
|
lexer = CompletionLexer(PythonLexer())
|
|
|
|
|
|
# Test simplest case.
|
|
|
self.assertEqual(lexer.get_context("foo.bar.baz"),
|
|
|
[ "foo", "bar", "baz" ])
|
|
|
|
|
|
# Test trailing period.
|
|
|
self.assertEqual(lexer.get_context("foo.bar."), [ "foo", "bar", "" ])
|
|
|
|
|
|
# Test with prompt present.
|
|
|
self.assertEqual(lexer.get_context(">>> foo.bar.baz"),
|
|
|
[ "foo", "bar", "baz" ])
|
|
|
|
|
|
# Test spacing in name.
|
|
|
self.assertEqual(lexer.get_context("foo.bar. baz"), [ "baz" ])
|
|
|
|
|
|
# Test parenthesis.
|
|
|
self.assertEqual(lexer.get_context("foo("), [])
|
|
|
|
|
|
def testC(self):
|
|
|
""" Does the CompletionLexer work for C/C++?
|
|
|
"""
|
|
|
lexer = CompletionLexer(CLexer())
|
|
|
self.assertEqual(lexer.get_context("foo.bar"), [ "foo", "bar" ])
|
|
|
self.assertEqual(lexer.get_context("foo->bar"), [ "foo", "bar" ])
|
|
|
|
|
|
lexer = CompletionLexer(CppLexer())
|
|
|
self.assertEqual(lexer.get_context("Foo::Bar"), [ "Foo", "Bar" ])
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
unittest.main()
|
|
|
|
