fixed unknown table suppression in sql tests

Kareem Zidane · Kareem Zidane · commit a45a3ec575b7 · 2017-05-27T11:55:19.000+02:00
diff --git a/tests/sqltests.py b/tests/sqltests.py
@@ -1,6 +1,7 @@
 import sys
 import unittest
 from cs50.sql import SQL
+import warnings
 
 class SQLTests(unittest.TestCase):
     def test_delete_returns_affected_rows(self):
@@ -86,10 +87,9 @@ def tearDown(self):
     def tearDownClass(self):
         try:
             self.db.execute("DROP TABLE IF EXISTS cs50")
-        except RuntimeError as e:
-
+        except Warning as e:
             # suppress "unknown table"
-            if not str(e).startswith("(1051L"):
+            if not str(e).startswith("(1051"):
                 raise e
 
 class PostgresTests(SQLTests):
