align SeaResultSet with ext-links-sea

Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>

Author: varun-edachali-dbx

examples/experimental/tests/test_sea_async_query.py

@@ -69,8 +69,12 @@ def test_sea_async_query_with_cloud_fetch():
 
         logger.info("Query is no longer pending, getting results...")
         cursor.get_async_execution_result()
-        rows = cursor.fetchall()
-        logger.info(f"Retrieved rows: {rows}")
+        
+        results = [cursor.fetchone()]
+        results.extend(cursor.fetchmany(10))
+        results.extend(cursor.fetchall())
+        logger.info(f"{len(results)} rows retrieved against 100 requested")
+
         logger.info(
             "Successfully retrieved asynchronous query results with cloud fetch enabled"
         )
@@ -150,8 +154,11 @@ def test_sea_async_query_without_cloud_fetch():
 
         logger.info("Query is no longer pending, getting results...")
         cursor.get_async_execution_result()
-        rows = cursor.fetchall()
-        logger.info(f"Retrieved rows: {rows}")
+        results = [cursor.fetchone()]
+        results.extend(cursor.fetchmany(10))
+        results.extend(cursor.fetchall())
+        logger.info(f"{len(results)} rows retrieved against 100 requested")
+
         logger.info(
             "Successfully retrieved asynchronous query results with cloud fetch disabled"
         )

examples/experimental/tests/test_sea_sync_query.py

@@ -55,8 +55,10 @@ def test_sea_sync_query_with_cloud_fetch():
         cursor.execute(
             "SELECT id, 'test_value_' || CAST(id as STRING) as test_value FROM range(1, 101)"
         )
-        rows = cursor.fetchall()
-        logger.info(f"Retrieved rows: {rows}")
+        results = [cursor.fetchone()]
+        results.extend(cursor.fetchmany(10))
+        results.extend(cursor.fetchall())
+        logger.info(f"{len(results)} rows retrieved against 100 requested")
 
         # Close resources
         cursor.close()
@@ -121,10 +123,11 @@ def test_sea_sync_query_without_cloud_fetch():
         cursor.execute(
             "SELECT id, 'test_value_' || CAST(id as STRING) as test_value FROM range(1, 101)"
         )
-        logger.info("Query executed successfully with cloud fetch disabled")
 
-        rows = cursor.fetchall()
-        logger.info(f"Retrieved rows: {rows}")
+        results = [cursor.fetchone()]
+        results.extend(cursor.fetchmany(10))
+        results.extend(cursor.fetchall())
+        logger.info(f"{len(results)} rows retrieved against 100 requested")
 
         # Close resources
         cursor.close()