|
28 | 28 | import java.util.HashMap; |
29 | 29 | import java.util.List; |
30 | 30 | import java.util.Map; |
| 31 | +import org.apache.arrow.memory.ArrowBuf; |
31 | 32 | import org.apache.arrow.memory.BufferAllocator; |
32 | 33 | import org.apache.arrow.memory.RootAllocator; |
| 34 | +import org.apache.arrow.vector.complex.BaseRepeatedValueVector; |
33 | 35 | import org.apache.arrow.vector.complex.DenseUnionVector; |
34 | 36 | import org.apache.arrow.vector.complex.FixedSizeListVector; |
35 | 37 | import org.apache.arrow.vector.complex.LargeListViewVector; |
@@ -115,6 +117,8 @@ public void testWithEmptyVector() { |
115 | 117 | TransferPair transferPair = listVector.getTransferPair(allocator); |
116 | 118 | transferPair.splitAndTransfer(0, 0); |
117 | 119 | assertEquals(0, transferPair.getTo().getValueCount()); |
| 120 | + transferPair.getTo().clear(); |
| 121 | + listVector.clear(); |
118 | 122 | // BaseFixedWidthVector |
119 | 123 | IntVector intVector = new IntVector("", allocator); |
120 | 124 | transferPair = intVector.getTransferPair(allocator); |
@@ -911,6 +915,11 @@ public void testListVectorZeroStartIndexAndLength() { |
911 | 915 |
|
912 | 916 | tp.splitAndTransfer(0, 0); |
913 | 917 | assertEquals(valueCount, newListVector.getValueCount()); |
| 918 | + List<ArrowBuf> buffers = newListVector.getFieldBuffers(); |
| 919 | + ArrowBuf offsetBuffer = buffers.get(1); |
| 920 | + assertEquals(BaseRepeatedValueVector.OFFSET_WIDTH, offsetBuffer.readableBytes()); |
| 921 | + assertTrue(offsetBuffer.capacity() >= BaseRepeatedValueVector.OFFSET_WIDTH); |
| 922 | + assertEquals(0, offsetBuffer.getInt(0)); |
914 | 923 |
|
915 | 924 | newListVector.clear(); |
916 | 925 | } |
|
0 commit comments