remember to close the result set

Author: Thomas Grainger

pandas/io/tests/test_sql.py

@@ -1489,20 +1489,15 @@ def test_read_procedure(self):
             SELECT * FROM test_procedure;
         END"""
 
-        connection = self.conn.connect()
-        trans = connection.begin()
-        try:
-            r1 = connection.execute(proc)
-            trans.commit()
-        except:
-            trans.rollback()
-            raise
+        with self.conn.connect() as connection:
+            with connection.begin():
+                connection.execute(proc).close()
 
-        res1 = sql.read_sql_query("CALL get_testdb();", self.conn)
+        res1 = sql.read_sql_query("CALL get_testdb()", self.conn)
         tm.assert_frame_equal(df, res1)
 
         # test delegation to read_sql_query
-        res2 = sql.read_sql("CALL get_testdb();", self.conn)
+        res2 = sql.read_sql("CALL get_testdb()", self.conn)
         tm.assert_frame_equal(df, res2)