Made some subtle fixes

Author: jprakash-db

tests/unit/test_fetches.py

@@ -80,6 +80,7 @@ def fetch_results(
             description,
             use_cloud_fetch=True,
             chunk_id=0,
+            timestamp_format=None,
         ):
             nonlocal batch_index
             results = FetchTests.make_arrow_queue(batch_list[batch_index])

tests/unit/test_sea_conversion.py

@@ -7,6 +7,7 @@
 import pytest
 import datetime
 import decimal
+import pytz
 from unittest.mock import Mock, patch
 
 from databricks.sql.backend.sea.utils.conversion import SqlType, SqlTypeConverter
@@ -158,7 +159,7 @@ def test_convert_timestamp_with_format(self):
             timestamp_format=fmt,
         )
         assert isinstance(result, datetime.datetime)
-        assert result == datetime.datetime(2023, 12, 31, 12, 30, 0, 123000)
+        assert result == datetime.datetime(2023, 12, 31, 12, 30, 0, 123000, tzinfo=pytz.UTC)
 
     def test_convert_timestamp_with_format_fallback(self):
         """Test that non-matching format falls back to dateutil."""

tests/unit/test_util.py

@@ -2,6 +2,7 @@
 import datetime
 from datetime import timezone, timedelta
 import pytest
+import pytz
 from databricks.sql.utils import (
     convert_to_assigned_datatypes_in_column_table,
     parse_timestamp,
@@ -235,7 +236,7 @@ def test_no_format_uses_dateutil(self):
     def test_matching_format_uses_strptime(self):
         fmt = "%Y-%m-%d %H:%M:%S.%f"
         result = parse_timestamp("2023-12-31 12:30:00.123000", fmt)
-        assert result == datetime.datetime(2023, 12, 31, 12, 30, 0, 123000)
+        assert result == datetime.datetime(2023, 12, 31, 12, 30, 0, 123000, tzinfo=pytz.UTC)
 
     def test_non_matching_format_falls_back_to_dateutil(self):
         fmt = "%Y-%m-%d %H:%M:%S.%f"
@@ -252,7 +253,7 @@ def test_convert_column_table_with_timestamp_format(self):
         result = convert_to_assigned_datatypes_in_column_table(
             column_table, description, timestamp_format=fmt
         )
-        assert result[0][0] == datetime.datetime(2023, 12, 31, 12, 30, 0)
+        assert result[0][0] == datetime.datetime(2023, 12, 31, 12, 30, 0, tzinfo=pytz.UTC)
 
     def test_convert_column_table_without_timestamp_format(self):
         description = [