remove outdated arrow_schema_bytes return

varun-edachali-dbx · varun-edachali-dbx · commit 73bc28267f83 · 2025-06-10T15:35:14.000Z
Signed-off-by: varun-edachali-dbx &lt;varun.edachali@databricks.com&gt;
diff --git a/tests/unit/test_thrift_backend.py b/tests/unit/test_thrift_backend.py
@@ -884,7 +884,7 @@ def test_handle_execute_response_can_handle_without_direct_results(
                 )
                 (
                     execute_response,
-                    arrow_schema_bytes,
+                    _,
                 ) = thrift_backend._handle_execute_response(execute_resp, Mock())
                 self.assertEqual(
                     execute_response.status,
@@ -990,9 +990,7 @@ def test_fall_back_to_hive_schema_if_no_arrow_schema(self, tcli_service_class):
             )
         )
         thrift_backend = self._make_fake_thrift_backend()
-        execute_response, arrow_schema_bytes = thrift_backend._handle_execute_response(
-            t_execute_resp, Mock()
-        )
+        _, _ = thrift_backend._handle_execute_response(t_execute_resp, Mock())
 
         self.assertEqual(
             hive_schema_mock,

Original file line number	Diff line number	Diff line change
`@@ -884,7 +884,7 @@ def test_handle_execute_response_can_handle_without_direct_results(`
`884`	`884`	`)`
`885`	`885`	`(`
`886`	`886`	`execute_response,`
`887`		`- arrow_schema_bytes,`
	`887`	`+ _,`
`888`	`888`	`) = thrift_backend._handle_execute_response(execute_resp, Mock())`
`889`	`889`	`self.assertEqual(`
`890`	`890`	`execute_response.status,`
`@@ -990,9 +990,7 @@ def test_fall_back_to_hive_schema_if_no_arrow_schema(self, tcli_service_class):`
`990`	`990`	`)`
`991`	`991`	`)`
`992`	`992`	`thrift_backend = self._make_fake_thrift_backend()`
`993`		`- execute_response, arrow_schema_bytes = thrift_backend._handle_execute_response(`
`994`		`- t_execute_resp, Mock()`
`995`		`- )`
	`993`	`+ _, _ = thrift_backend._handle_execute_response(t_execute_resp, Mock())`
`996`	`994`
`997`	`995`	`self.assertEqual(`
`998`	`996`	`hive_schema_mock,`