Fix test_segments_cursor

Author: damian3031

tests/integration/test_dbapi_integration.py

@@ -1884,8 +1884,14 @@ def test_segments_cursor(trino_connection):
     total = 0
     for segment in segments:
         assert segment.encoding == trino_connection._client_session.encoding
-        assert isinstance(segment.segment.uri, str), f"Expected string for uri, got {segment.segment.uri}"
-        assert isinstance(segment.segment.ack_uri, str), f"Expected string for ack_uri, got {segment.segment.ack_uri}"
+        if hasattr(segment.segment, 'uri'):
+            assert isinstance(segment.segment.uri, str), (
+                f"Expected string for uri, got {segment.segment.uri}"
+            )
+        if hasattr(segment.segment, 'ack_uri'):
+            assert isinstance(segment.segment.ack_uri, str), (
+                f"Expected string for ack_uri, got {segment.segment.ack_uri}"
+            )
         total += len(list(SegmentIterator(segment, row_mapper)))
     assert total == 300875, f"Expected total rows 300875, got {total}"