diff --git a/cpp/cmake/thirdparty/CUDF_GetArrow.cmake b/cpp/cmake/thirdparty/CUDF_GetArrow.cmake index b2ddac55667..79d3c0770a3 100644 --- a/cpp/cmake/thirdparty/CUDF_GetArrow.cmake +++ b/cpp/cmake/thirdparty/CUDF_GetArrow.cmake @@ -124,4 +124,4 @@ endfunction() set(CUDF_VERSION_Arrow 1.0.1) -find_and_configure_arrow(${CUDF_VERSION_Arrow} ${CUDF_USE_ARROW_STATIC}) \ No newline at end of file +find_and_configure_arrow(${CUDF_VERSION_Arrow} ${CUDF_USE_ARROW_STATIC}) diff --git a/cpp/tests/io/arrow_io_source_test.cpp b/cpp/tests/io/arrow_io_source_test.cpp index f5236d8ddcc..92bb3d9862e 100644 --- a/cpp/tests/io/arrow_io_source_test.cpp +++ b/cpp/tests/io/arrow_io_source_test.cpp @@ -61,27 +61,24 @@ TEST_F(ArrowIOTest, URIFileSystem) ASSERT_EQ(2, tbl.tbl->num_rows()); } -/* -// This S3 object is currently to large to run in CI. This test can be ran locally however -TEST_F(ArrowIOTest, URIFileSystem) +TEST_F(ArrowIOTest, S3FileSystem) { - std::string s3_uri = "s3://ursa-labs-taxi-data/2010/06/data.parquet?region=us-east-2"; + std::string s3_uri = "s3://rapidsai-data/cudf/test/tips.parquet?region=us-east-2"; std::unique_ptr datasource = std::make_unique(s3_uri); // Populate the Parquet Reader Options cudf::io::source_info src(datasource.get()); std::vector single_column; - single_column.insert(single_column.begin(), "dropoff_at"); + single_column.insert(single_column.begin(), "total_bill"); cudf::io::parquet_reader_options_builder builder(src); cudf::io::parquet_reader_options options = builder.columns(single_column).build(); // Read the Parquet file from S3 cudf::io::table_with_metadata tbl = cudf::io::read_parquet(options); - ASSERT_EQ(1, tbl.tbl->num_columns()); // Only single column specified in reader_options - ASSERT_EQ(14825128, tbl.tbl->num_rows()); // known number of rows from the S3 file + ASSERT_EQ(1, tbl.tbl->num_columns()); // Only single column specified in reader_options + ASSERT_EQ(244, tbl.tbl->num_rows()); // known number of rows from the S3 file } -*/ CUDF_TEST_PROGRAM_MAIN()