1 files changed, 18 insertions, 9 deletions
diff --git a/tests/test_using_api.py b/tests/test_using_api.py
index 0af12441..ad8f641d 100644
--- a/tests/test_using_api.py
+++ b/tests/test_using_api.py
@@ -1,12 +1,24 @@
+# -*- coding: utf-8 -*-
+"""
+    Pygments tests for using()
+    ~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+    :copyright: Copyright 2006-2009 by the Pygments team, see AUTHORS.
+    :license: BSD, see LICENSE for details.
+"""
+
 import unittest
+
 from pygments.lexer import using, bygroups, this, RegexLexer
 from pygments.token import String, Text, Keyword
 
 class TestLexer(RegexLexer):
     tokens = {
         'root': [
-            (r'#.*', using(this, state='invalid')),
-            (r'(")(.+?)(")', bygroups(String, using(this, state='string'), String)),
+            (r'#.*',
+             using(this, state='invalid')),
+            (r'(")(.+?)(")',
+             bygroups(String, using(this, state='string'), String)),
             (r'[^"]+', Text),
         ],
         'string': [
@@ -14,18 +26,15 @@ class TestLexer(RegexLexer):
         ],
     }
 
+
 class UsingStateTest(unittest.TestCase):
     def test_basic(self):
         expected = [(Text, 'a'), (String, '"'), (Keyword, 'bcd'),
                     (String, '"'), (Text, 'e\n')]
         t = list(TestLexer().get_tokens('a"bcd"e'))
         self.assertEquals(t, expected)
+
     def test_error(self):
         def gen():
-            x = list(TestLexer().get_tokens('#a'))
-        #XXX: should probably raise a more specific exception if the state
-        #     doesn't exist.
-        self.assertRaises(Exception, gen)
-
-if __name__ == "__main__":
-    unittest.main()
+            return list(TestLexer().get_tokens('#a'))
+        self.assertRaises(KeyError, gen)