Function TEST

cpp/src/parquet/geospatial/statistics_test.cc:32–70

namespace parquet::geospatial {

TEST(TestGeoStatistics, TestDefaults) {
  GeoStatistics stats;
  EXPECT_TRUE(stats.geometry_types().has_value());
  EXPECT_EQ(stats.geometry_types().value().size(), 0);
  EXPECT_TRUE(stats.is_valid());
  EXPECT_THAT(stats.dimension_empty(), ::testing::ElementsAre(true, true, true, true));
  EXPECT_THAT(stats.dimension_valid(), ::testing::ElementsAre(true, true, true, true));
  for (int i = 0; i < kMaxDimensions; i++) {
    EXPECT_EQ(stats.lower_bound()[i], kInf);
    EXPECT_EQ(stats.upper_bound()[i], -kInf);
  }

  EXPECT_TRUE(stats.Equals(GeoStatistics()));
  EXPECT_EQ(stats.ToString(),
            "<GeoStatistics> x: empty y: empty z: empty m: empty "
            "geometry_types: []");

  // Merging empty with empty should equal empty
  stats.Merge(GeoStatistics());
  EXPECT_TRUE(stats.Equals(GeoStatistics()));

  // There's no way to encode empty ranges in Thrift, so we pretend that we
  // didn't calculate them.
  auto encoded = stats.Encode();
  EXPECT_FALSE(encoded->xy_bounds_present);
  EXPECT_FALSE(encoded->z_bounds_present);
  EXPECT_FALSE(encoded->m_bounds_present);
  EXPECT_FALSE(encoded->geospatial_types_present());

  // When imported, the statistics marked with everything uncalculated should
  // be "valid" but should have individual components marked as invalid.
  GeoStatistics valid_but_uncalculated;
  valid_but_uncalculated.Decode(*encoded);

  EXPECT_TRUE(valid_but_uncalculated.is_valid());
  EXPECT_THAT(valid_but_uncalculated.dimension_valid(),
              ::testing::ElementsAre(false, false, false, false));
  EXPECT_EQ(valid_but_uncalculated.geometry_types(), std::nullopt);
}

TEST(TestGeoStatistics, TestImportEncodedWithNaN) {
  double nan_dbl = std::numeric_limits<double>::quiet_NaN();