test(bigquery): verify batch count in arrow integration tests

Author: jinseopkim0

java-bigquery/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/it/ITBigQueryTest.java

@@ -7517,7 +7517,7 @@ void testQueryResultsFormatArrow() throws InterruptedException {
         totalRows += root.getRowCount();
         assertEquals(1, root.getRowCount());
       }
-      assertTrue(batchCount > 0);
+      assertEquals(1, batchCount);
       assertEquals(1, totalRows);
     }
   }
@@ -7532,10 +7532,13 @@ void testQueryResultsFormatArrowMultiPage() throws InterruptedException {
             .build();
     try (ArrowQueryResult result = bigquery.queryArrow(config)) {
       assertNotNull(result);
+      int batchCount = 0;
       long totalRows = 0;
       for (VectorSchemaRoot root : result) {
+        batchCount++;
         totalRows += root.getRowCount();
       }
+      assertTrue(batchCount > 1);
       assertEquals(15000, totalRows);
     }
   }