Skip to content

Commit e293989

Browse files
MINOR: Reset list test helper capacity
1 parent 10b259b commit e293989

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

vector/src/test/java/org/apache/arrow/vector/testing/ValueVectorDataPopulator.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -633,9 +633,10 @@ public static void setVector(LargeVarCharVector vector, String... values) {
633633

634634
/** Populate values for {@link ListVector}. */
635635
public static void setVector(ListVector vector, List<Integer>... values) {
636-
vector.allocateNewSafe();
637636
Types.MinorType type = Types.MinorType.INT;
638637
vector.addOrGetVector(FieldType.nullable(type.getType()));
638+
vector.setInitialCapacity(values.length);
639+
vector.allocateNewSafe();
639640

640641
IntVector dataVector = (IntVector) vector.getDataVector();
641642
dataVector.allocateNew();
@@ -662,9 +663,10 @@ public static void setVector(ListVector vector, List<Integer>... values) {
662663

663664
/** Populate values for {@link LargeListVector}. */
664665
public static void setVector(LargeListVector vector, List<Integer>... values) {
665-
vector.allocateNewSafe();
666666
Types.MinorType type = Types.MinorType.INT;
667667
vector.addOrGetVector(FieldType.nullable(type.getType()));
668+
vector.setInitialCapacity(values.length);
669+
vector.allocateNewSafe();
668670

669671
IntVector dataVector = (IntVector) vector.getDataVector();
670672
dataVector.allocateNew();

0 commit comments

Comments
 (0)