88use std:: { collections:: HashMap , path:: PathBuf , time:: SystemTime } ;
99
1010use amadeus:: prelude:: * ;
11- use amadeus_parquet:: get_row_predicate;
1211
1312fn assert_columns_in_row < T > (
14- expected_column_names : & Vec < String > , row : Result < Value , T > ,
13+ expected_column_names : & [ String ] , row : Result < Value , T > ,
1514) -> Result < Value , T > {
1615 let row_unwrapped = row. ok ( ) . unwrap ( ) ;
1716 let row_group = row_unwrapped. clone ( ) . into_group ( ) . ok ( ) . unwrap ( ) ;
@@ -38,20 +37,17 @@ async fn dynamic_reads() {
3837
3938 let column_names = vec ! [ "uri" . to_string( ) , "location" . to_string( ) ] ;
4039
41- #[ cfg( feature = "aws" ) ]
42- {
43- let rows = Parquet :: < _ , Value > :: new ( vec ! [
44- S3File :: new_with( AwsRegion :: UsEast1 , "us-east-1.data-analytics" , "cflogworkshop/optimized/cf-accesslogs/year=2018/month=11/day=02/part-00176-17868f39-cd99-4b60-bb48-8daf9072122e.c000.snappy.parquet" , AwsCredentials :: Anonymous ) ,
45- ] , get_row_predicate ( column_names. clone ( ) ) ) . await . unwrap ( ) ;
40+ let rows = Parquet :: < _ , Value > :: new ( vec ! [
41+ PathBuf :: from( "amadeus-testing/parquet/cf-accesslogs/year=2018/month=11/day=02/part-00176-17868f39-cd99-4b60-bb48-8daf9072122e.c000.snappy.parquet" ) ,
42+ ] , amadeus_parquet:: get_row_predicate ( column_names. clone ( ) ) ) . await . unwrap ( ) ;
4643
47- assert_eq ! (
48- rows. par_stream( )
49- . map( move |row: Result <Value , _>| assert_columns_in_row( & column_names, row) )
50- . count( pool)
51- . await ,
52- 4172
53- ) ;
54- }
44+ assert_eq ! (
45+ rows. par_stream( )
46+ . map( move |row: Result <Value , _>| assert_columns_in_row( & column_names, row) )
47+ . count( pool)
48+ . await ,
49+ 4172
50+ ) ;
5551
5652 println ! ( "in {:?}" , start. elapsed( ) . unwrap( ) ) ;
5753}
0 commit comments