@@ -134,19 +134,26 @@ TEST(HasFieldIds, PrimitiveNode) {
134134}
135135
136136TEST (HasFieldIds, GroupNode) {
137- auto group_node_without_field_id =
138- MakeGroupNode (" test_group" , {MakeInt32Node (" c1" ), MakeInt32Node (" c2" )});
139- EXPECT_FALSE (HasFieldIds (group_node_without_field_id));
140-
141- auto group_node_with_full_field_id = MakeGroupNode (
142- " test_group" ,
143- {MakeInt32Node (" c1" , /* field_id=*/ 2 ), MakeInt32Node (" c2" , /* field_id=*/ 3 )},
144- /* field_id=*/ 1 );
145- EXPECT_TRUE (HasFieldIds (group_node_with_full_field_id));
146-
147- auto group_node_with_partial_field_id = MakeGroupNode (
148- " test_group" , {MakeInt32Node (" c1" , /* field_id=*/ 1 ), MakeInt32Node (" c2" )});
149- EXPECT_TRUE (HasFieldIds (group_node_with_partial_field_id));
137+ // Group node without field id
138+ EXPECT_FALSE (HasFieldIds (MakeGroupNode (" test_group" , {
139+ MakeInt32Node (" c1" ),
140+ MakeInt32Node (" c2" ),
141+ })));
142+
143+ // Group node with full field id
144+ EXPECT_TRUE (HasFieldIds (MakeGroupNode (" test_group" ,
145+ {
146+ MakeInt32Node (" c1" , /* field_id=*/ 2 ),
147+ MakeInt32Node (" c2" , /* field_id=*/ 3 ),
148+ },
149+ /* field_id=*/ 1 )));
150+
151+ // Group node with partial field id
152+ EXPECT_TRUE (
153+ HasFieldIds (MakeGroupNode (" test_group" , {
154+ MakeInt32Node (" c1" , /* field_id=*/ 1 ),
155+ MakeInt32Node (" c2" ),
156+ })));
150157}
151158
152159TEST (ParquetSchemaProjectionTest, ProjectIdenticalSchemas) {
0 commit comments