added test, tolerating non-DDL, non-DML keywords like BEGIN

Author: dmalan

setup.py

@@ -16,5 +16,5 @@
     package_dir={"": "src"},
     packages=["cs50"],
     url="https://github.com/cs50/python-cs50",
-    version="5.0.2"
+    version="5.0.3"
 )

src/cs50/sql.py

@@ -126,7 +126,7 @@ def execute(self, sql, *args, **kwargs):
                 command = token.value.upper()
                 break
         else:
-            raise RuntimeError("unrecognized command")
+            command = None
 
         # Flatten statement
         tokens = list(statements[0].flatten())

tests/sql.py

@@ -128,6 +128,9 @@ def test_rollback(self):
         self.db.execute("ROLLBACK")
         self.assertEqual(self.db.execute("SELECT val FROM cs50"), [])
 
+    def test_cte(self):
+        self.assertEqual(self.db.execute("WITH foo AS ( SELECT 1 ) SELECT * FROM foo"), [{"1": 1}])
+
     def tearDown(self):
         self.db.execute("DROP TABLE cs50")
         self.db.execute("DROP TABLE IF EXISTS foo")