Re: [PR] [parquet] Allow more encryption algorithms [arrow-rs]

["hsiang-c (via GitHub)" <gi...@apache.org>]

hsiang-c commented on code in PR #9203:
URL: https://github.com/apache/arrow-rs/pull/9203#discussion_r2868326986


##########
parquet/tests/encryption/encryption_util.rs:
##########
@@ -104,6 +104,20 @@ pub(crate) fn verify_encryption_double_test_data(
     assert_eq!(row_count, file_metadata.num_rows() as usize);
 }
 
+/// Verifies parquet-java generated AES-256 data from the parquet-testing repository
+pub(crate) fn verify_encryption_test_data_from_java(metadata: &ParquetMetaData) {
+    let file_metadata = metadata.file_metadata();
+    assert_eq!(file_metadata.num_rows(), 10000);
+    assert_eq!(file_metadata.schema_descr().num_columns(), 7);
+
+    let mut total_rows = 0;
+    metadata.row_groups().iter().for_each(|rg| {
+        assert_eq!(rg.num_columns(), 7);
+        total_rows += rg.num_rows();
+    });
+    assert_eq!(total_rows, 10000);

Review Comment:
   @rok Thanks for your review.
   
   In https://github.com/apache/parquet-testing/pull/102, I match the schema and data now. So I can reuse `verify_encryption_test_data`.



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: github-unsubscribe@arrow.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org