nit

Author: jprakash-db

src/databricks/sql/client.py

@@ -49,8 +49,6 @@
 from databricks.sql.thrift_api.TCLIService.ttypes import (
     TSparkParameter,
     TOperationState,
-    TSparkParameterValue,
-    TSparkParameterValueArg,
 )
 
 

tests/e2e/test_complex_types.py

@@ -22,7 +22,7 @@ def table_fixture(self, connection_details):
                     array_array_col ARRAY<ARRAY<STRING>>,
                     array_map_col ARRAY<MAP<STRING, INTEGER>>,
                     map_array_col MAP<STRING, ARRAY<STRING>>
-                )
+                ) USING DELTA
                 """
             )
             # Insert a record
@@ -41,7 +41,7 @@ def table_fixture(self, connection_details):
             )
             yield
             # Clean up the table after the test
-            cursor.execute("DROP TABLE IF EXISTS pysql_test_complex_types_table")
+            cursor.execute("DELETE FROM pysql_test_complex_types_table")
 
     @pytest.mark.parametrize(
         "field,expected_type",

tests/e2e/test_parameterized_queries.py

@@ -179,9 +179,9 @@ def _inline_roundtrip(self, params: dict, paramstyle: ParamStyle, target_column)
         :paramstyle:
             This is a no-op but is included to make the test-code easier to read.
         """
-        INSERT_QUERY = f"INSERT INTO ___________________first.jprakash.pysql_e2e_inline_param_test_table (`{target_column}`) VALUES (%(p)s)"
-        SELECT_QUERY = f"SELECT {target_column} `col` FROM ___________________first.jprakash.pysql_e2e_inline_param_test_table LIMIT 1"
-        DELETE_QUERY = "DELETE FROM ___________________first.jprakash.pysql_e2e_inline_param_test_table"
+        INSERT_QUERY = f"INSERT INTO pysql_e2e_inline_param_test_table (`{target_column}`) VALUES (%(p)s)"
+        SELECT_QUERY = f"SELECT {target_column} `col` FROM pysql_e2e_inline_param_test_table LIMIT 1"
+        DELETE_QUERY = "DELETE FROM pysql_e2e_inline_param_test_table"
 
         with self.connection(extra_params={"use_inline_params": True}) as conn:
             with conn.cursor() as cursor: