Skip to content

Commit 7f0292e

Browse files
authored
[tests] Improve test coverage on VectorizedColumnBatch (#1049)
1 parent c6c9a17 commit 7f0292e

File tree

2 files changed

+35
-3
lines changed

2 files changed

+35
-3
lines changed

fluss-common/src/test/java/com/alibaba/fluss/row/arrow/ArrowReaderWriterTest.java

Lines changed: 35 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,42 @@ void testReaderWriter() throws IOException {
160160
for (int i = 0; i < rowCount; i++) {
161161
row.setRowId(i);
162162
assertThatRow(row).withSchema(rowType).isEqualTo(TEST_DATA.get(i));
163+
164+
InternalRow rowData = TEST_DATA.get(i);
165+
assertThat(row.getBoolean(0)).isEqualTo(rowData.getBoolean(0));
166+
assertThat(row.getByte(1)).isEqualTo(rowData.getByte(1));
167+
assertThat(row.getShort(2)).isEqualTo(rowData.getShort(2));
168+
if (!row.isNullAt(3)) {
169+
assertThat(row.getInt(3)).isEqualTo(rowData.getInt(3));
170+
}
171+
assertThat(row.getLong(4)).isEqualTo(rowData.getLong(4));
172+
assertThat(row.getFloat(5)).isEqualTo(rowData.getFloat(5));
173+
assertThat(row.getDouble(6)).isEqualTo(rowData.getDouble(6));
174+
assertThat(row.getDecimal(7, 10, 3)).isEqualTo(rowData.getDecimal(7, 10, 3));
175+
assertThat(row.getChar(8, 3)).isEqualTo(rowData.getChar(8, 3));
176+
if (!row.isNullAt(9)) {
177+
assertThat(row.getString(9)).isEqualTo(rowData.getString(9));
178+
}
179+
assertThat(row.getBinary(10, 5)).isEqualTo(rowData.getBinary(10, 5));
180+
assertThat(row.getBytes(11)).isEqualTo(rowData.getBytes(11));
181+
assertThat(row.getInt(12)).isEqualTo(rowData.getInt(12));
182+
assertThat(row.getInt(13)).isEqualTo(rowData.getInt(13));
183+
if (!row.isNullAt(14)) {
184+
assertThat(row.getTimestampNtz(14, 0))
185+
.isEqualTo(rowData.getTimestampNtz(14, 0));
186+
}
187+
assertThat(row.getTimestampNtz(15, 3)).isEqualTo(rowData.getTimestampNtz(15, 3));
188+
assertThat(row.getTimestampNtz(16, 6)).isEqualTo(rowData.getTimestampNtz(16, 6));
189+
assertThat(row.getTimestampNtz(17, 9)).isEqualTo(rowData.getTimestampNtz(17, 9));
190+
if (!row.isNullAt(18)) {
191+
assertThat(row.getTimestampLtz(18, 0))
192+
.isEqualTo(rowData.getTimestampLtz(18, 0));
193+
}
194+
assertThat(row.getTimestampLtz(19, 3)).isEqualTo(rowData.getTimestampLtz(19, 3));
195+
assertThat(row.getTimestampLtz(20, 6)).isEqualTo(rowData.getTimestampLtz(20, 6));
196+
assertThat(row.getTimestampLtz(21, 9)).isEqualTo(rowData.getTimestampLtz(21, 9));
163197
}
198+
reader.close();
164199
}
165200
}
166201

fluss-test-coverage/pom.xml

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -243,9 +243,6 @@
243243
<exclude>
244244
com.alibaba.fluss.row.columnar.BytesColumnVector.Bytes
245245
</exclude>
246-
<exclude>
247-
com.alibaba.fluss.row.columnar.VectorizedColumnBatch
248-
</exclude>
249246
<exclude>com.alibaba.fluss.row.encode.RowEncoder</exclude>
250247
<exclude>com.alibaba.fluss.table.*</exclude>
251248
<exclude>com.alibaba.fluss.record.*</exclude>

0 commit comments

Comments
 (0)