Added Python cls builtin word token recognition

2 files changed, 48 insertions, 2 deletions

diff --git a/pygments/lexers/python.py b/pygments/lexers/python.py
index 7601afa8..35635ed1 100644
--- a/pygments/lexers/python.py
+++ b/pygments/lexers/python.py
@@ -116,7 +116,7 @@ class PythonLexer(RegexLexer):
                 'unichr', 'unicode', 'vars', 'xrange', 'zip'),
                 prefix=r'(?<!\.)', suffix=r'\b'),
              Name.Builtin),
-            (r'(?<!\.)(self|None|Ellipsis|NotImplemented|False|True'
+            (r'(?<!\.)(self|None|Ellipsis|NotImplemented|False|True|cls'
              r')\b', Name.Builtin.Pseudo),
             (words((
                 'ArithmeticError', 'AssertionError', 'AttributeError',
@@ -303,7 +303,7 @@ class Python3Lexer(RegexLexer):
             'sum', 'super', 'tuple', 'type', 'vars', 'zip'), prefix=r'(?<!\.)',
             suffix=r'\b'),
          Name.Builtin),
-        (r'(?<!\.)(self|Ellipsis|NotImplemented)\b', Name.Builtin.Pseudo),
+        (r'(?<!\.)(self|Ellipsis|NotImplemented|cls)\b', Name.Builtin.Pseudo),
         (words((
             'ArithmeticError', 'AssertionError', 'AttributeError',
             'BaseException', 'BufferError', 'BytesWarning', 'DeprecationWarning',
diff --git a/tests/test_python.py b/tests/test_python.py
new file mode 100644
index 00000000..21c69605
--- /dev/null
+++ b/tests/test_python.py
@@ -0,0 +1,46 @@
+# -*- coding: utf-8 -*-
+"""
+    Python Tests
+    ~~~~~~~~~
+
+    :copyright: Copyright 2006-2015 by the Pygments team, see AUTHORS.
+    :license: BSD, see LICENSE for details.
+"""
+
+import unittest
+
+from pygments.lexers import PythonLexer
+from pygments.token import Token
+
+
+class PythonTest(unittest.TestCase):
+    def setUp(self):
+        self.lexer = PythonLexer()
+
+    def testNeedsName(self):
+        fragment = 'class TestClass():\n    @classmethod\n    def hello(cls):\n        pass\n'
+        tokens = [
+            (Token.Keyword, 'class'),
+            (Token.Text, ' '),
+            (Token.Name.Class, 'TestClass'),
+            (Token.Punctuation, '('),
+            (Token.Punctuation, ')'),
+            (Token.Punctuation, ':'),
+            (Token.Text, '\n'),
+            (Token.Text, '    '),
+            (Token.Name.Decorator, '@classmethod'),
+            (Token.Text, '\n'),
+            (Token.Text, '    '),
+            (Token.Keyword, 'def'),
+            (Token.Text, ' '),
+            (Token.Name.Function, 'hello'),
+            (Token.Punctuation, '('),
+            (Token.Name.Builtin.Pseudo, 'cls'),
+            (Token.Punctuation, ')'),
+            (Token.Punctuation, ':'),
+            (Token.Text, '\n'),
+            (Token.Text, '        '),
+            (Token.Keyword, 'pass'),
+            (Token.Text, '\n'),
+        ]
+        self.assertEqual(tokens, list(self.lexer.get_tokens(fragment)))