Skip to content

Commit fedcf04

Browse files
committed
[GH-470] Address comment.
1 parent 979d56f commit fedcf04

File tree

1 file changed

+27
-0
lines changed

1 file changed

+27
-0
lines changed

vector/src/test/java/org/apache/arrow/vector/TestListViewVector.java

+27
Original file line numberDiff line numberDiff line change
@@ -2223,4 +2223,31 @@ private void writeIntValues(UnionListViewWriter writer, int[] values) {
22232223
}
22242224
writer.endListView();
22252225
}
2226+
2227+
@Test
2228+
public void testGetElementEndIndex() {
2229+
try (ListViewVector vector = ListViewVector.empty("list", allocator)) {
2230+
vector.allocateNew();
2231+
2232+
UnionListViewWriter writer = vector.getWriter();
2233+
2234+
writer.setPosition(0);
2235+
writer.startListView();
2236+
writer.integer().writeInt(1);
2237+
writer.integer().writeInt(2);
2238+
writer.endListView();
2239+
2240+
writer.setPosition(1);
2241+
writer.startListView();
2242+
writer.integer().writeInt(3);
2243+
writer.integer().writeInt(4);
2244+
writer.integer().writeInt(5);
2245+
writer.endListView();
2246+
2247+
vector.setValueCount(2);
2248+
2249+
assertEquals(2, vector.getElementEndIndex(0));
2250+
assertEquals(5, vector.getElementEndIndex(1));
2251+
}
2252+
}
22262253
}

0 commit comments

Comments
 (0)