Verify MapVector schema is not null in ArrowBridgeSchemaTest.cpp

facebookincubator · Oct 11, 2024 · edd660c · edd660c
1 parent 27c5e1f
commit edd660c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/velox/vector/arrow/tests/ArrowBridgeSchemaTest.cpp b/velox/vector/arrow/tests/ArrowBridgeSchemaTest.cpp
@@ -87,6 +87,11 @@ class ArrowBridgeSchemaExportTest : public testing::Test {
       EXPECT_STREQ("+m", schema->format);
       ASSERT_EQ(schema->n_children, 1);
       schema = schema->children[0];
+      // Map data should be a non-nullable struct type
+      ASSERT_EQ(schema->flags & ARROW_FLAG_NULLABLE, 0);
+      ASSERT_EQ(schema->n_children, 2);
+      // Map data key type should be a non-nullable
+      ASSERT_EQ(schema->children[0]->flags & ARROW_FLAG_NULLABLE, 0);
     } else if (type->kind() == TypeKind::ROW) {
       EXPECT_STREQ("+s", schema->format);
     }