adds tests for IN

Author: dmalan

tests/sql.py

@@ -7,6 +7,7 @@
 
 from cs50.sql import SQL
 
+
 class SQLTests(unittest.TestCase):
 
     def test_multiple_statements(self):
@@ -142,6 +143,7 @@ def tearDownClass(self):
             if not str(e).startswith("(1051"):
                 raise e
 
+
 class MySQLTests(SQLTests):
     @classmethod
     def setUpClass(self):
@@ -150,6 +152,7 @@ def setUpClass(self):
     def setUp(self):
         self.db.execute("CREATE TABLE cs50 (id INTEGER NOT NULL AUTO_INCREMENT, val VARCHAR(16), bin BLOB, PRIMARY KEY (id))")
 
+
 class PostgresTests(SQLTests):
     @classmethod
     def setUpClass(self):
@@ -161,7 +164,9 @@ def setUp(self):
     def test_cte(self):
         self.assertEqual(self.db.execute("WITH foo AS ( SELECT 1 AS bar ) SELECT bar FROM foo"), [{"bar": 1}])
 
+
 class SQLiteTests(SQLTests):
+
     @classmethod
     def setUpClass(self):
         open("test.db", "w").close()
@@ -207,27 +212,44 @@ def test_qmark(self):
         self.assertEqual(self.db.execute("SELECT * FROM foo WHERE firstname = ? AND lastname = ?", ["qux", "quux"]), [{"firstname": "qux", "lastname": "quux"}])
         self.db.execute("DELETE FROM foo")
 
+        self.db.execute("INSERT INTO foo VALUES (?)", ("bar", "baz"))
+        self.assertEqual(self.db.execute("SELECT * FROM foo"), [{"firstname": "bar", "lastname": "baz"}])
+        self.db.execute("DELETE FROM foo")
+
         self.db.execute("INSERT INTO foo VALUES (?, ?)", ("bar", "baz"))
         self.assertEqual(self.db.execute("SELECT * FROM foo"), [{"firstname": "bar", "lastname": "baz"}])
         self.db.execute("DELETE FROM foo")
 
-        self.db.execute("INSERT INTO foo VALUES (?, ?)", ["bar", "baz"])
+        self.db.execute("INSERT INTO foo VALUES (?)", ["bar", "baz"])
         self.assertEqual(self.db.execute("SELECT * FROM foo"), [{"firstname": "bar", "lastname": "baz"}])
         self.db.execute("DELETE FROM foo")
 
+        self.db.execute("INSERT INTO foo VALUES (?, ?)", ["bar", "baz"])
+        self.assertEqual(self.db.execute("SELECT * FROM foo"), [{"firstname": "bar", "lastname": "baz"}])
+        self.db.execute("DELETE FROM foo")
 
         self.db.execute("INSERT INTO foo VALUES (?,?)", "bar", "baz")
         self.assertEqual(self.db.execute("SELECT * FROM foo"), [{"firstname": "bar", "lastname": "baz"}])
         self.db.execute("DELETE FROM foo")
 
         self.db.execute("CREATE TABLE bar (firstname STRING)")
+
         self.db.execute("INSERT INTO bar VALUES (?)", "baz")
         self.assertEqual(self.db.execute("SELECT * FROM bar"), [{"firstname": "baz"}])
+        self.db.execute("DELETE FROM bar")
+
+        self.db.execute("INSERT INTO bar VALUES (?)", "baz")
+        self.db.execute("INSERT INTO bar VALUES (?)", "qux")
+        self.assertEqual(self.db.execute("SELECT * FROM bar WHERE firstname IN (?)", ("baz", "qux")), [{"firstname": "baz"}, {"firstname": "qux"}])
+        self.db.execute("DELETE FROM bar")
+
+        self.db.execute("INSERT INTO bar VALUES (?)", "baz")
+        self.db.execute("INSERT INTO bar VALUES (?)", "qux")
+        self.assertEqual(self.db.execute("SELECT * FROM bar WHERE firstname IN (?)", ["baz", "qux"]), [{"firstname": "baz"}, {"firstname": "qux"}])
+        self.db.execute("DELETE FROM bar")
 
         self.assertRaises(RuntimeError, self.db.execute, "INSERT INTO foo VALUES (?)")
         self.assertRaises(RuntimeError, self.db.execute, "INSERT INTO foo VALUES (?, ?)")
-        # self.assertRaises(RuntimeError, self.db.execute, "INSERT INTO foo VALUES (?)", ('bar', 'baz'))
-        # self.assertRaises(RuntimeError, self.db.execute, "INSERT INTO foo VALUES (?)", ['bar', 'baz'])
         self.assertRaises(RuntimeError, self.db.execute, "INSERT INTO foo VALUES (?, ?)", 'bar', 'baz', 'qux')
         self.assertRaises(RuntimeError, self.db.execute, "INSERT INTO foo VALUES (?, ?)", ('bar', 'baz', 'qux'))
         self.assertRaises(RuntimeError, self.db.execute, "INSERT INTO foo VALUES (?, ?)", ['bar', 'baz', 'qux'])