1 files changed, 8 insertions, 0 deletions
diff --git a/tests/test_errors.py b/tests/test_errors.py
index 5645e18..cb680a2 100755
--- a/tests/test_errors.py
+++ b/tests/test_errors.py
@@ -54,6 +54,14 @@ class ErrorsTests(ConnectingTestCase):
 
         self.assertEqual(type(e), self.conn.ProgrammingError)
 
+    def test_lookup(self):
+        from psycopg2 import errors
+
+        self.assertIs(errors.lookup('42P01'), errors.UndefinedTable)
+
+        with self.assertRaises(KeyError):
+            errors.lookup('XXXXX')
+
 
 def test_suite():
     return unittest.TestLoader().loadTestsFromName(__name__)