chore: Rename method to set_query_execution_args

Author: paxcodes

clickhouse_backend/driver/connection.py

@@ -114,7 +114,7 @@ def use_numpy(self, value):
             self._client.progress_query_result_cls = ProgressQueryResult
 
     @contextmanager
-    def set_query_args(
+    def set_query_execution_args(
         self, columnar: T.Optional[bool] = None, use_numpy: T.Optional[bool] = None
     ):
         original_use_numpy = self.use_numpy

tests/backends/tests.py

@@ -578,7 +578,7 @@ def test_columnar_query(self):
             LIMIT 10
         """
         with connections["s2r1"].cursor() as cursorWrapper:
-            with cursorWrapper.cursor.set_query_args(columnar=True) as cursor:
+            with cursorWrapper.cursor.set_query_execution_args(columnar=True) as cursor:
                 cursor.execute(sql)
                 self.assertEqual(
                     cursor.fetchall(),
@@ -630,7 +630,7 @@ def test_use_numpy_query(self):
         import numpy as np
 
         with connections["s2r1"].cursor() as cursorWrapper:
-            with cursorWrapper.cursor.set_query_args(
+            with cursorWrapper.cursor.set_query_execution_args(
                 columnar=True, use_numpy=True
             ) as cursor:
                 cursor.execute(sql)
@@ -701,7 +701,7 @@ def test_use_numpy_but_not_columnar_format(self):
         import numpy as np
 
         with connections["s2r1"].cursor() as cursorWrapper:
-            with cursorWrapper.cursor.set_query_args(
+            with cursorWrapper.cursor.set_query_execution_args(
                 columnar=False, use_numpy=True
             ) as cursor:
                 cursor.execute(sql)