Skip to content

Commit

Permalink
Fixed formatting for TestAceroSubstraitConsumer
Browse files Browse the repository at this point in the history
  • Loading branch information
normanj-bitquill committed Jun 17, 2024
1 parent d351b90 commit 887d215
Showing 1 changed file with 12 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -516,8 +516,10 @@ public void testRunExtendedExpressionsProjectionWithFilterInsteadOfProjectionExc
int rowcount = 0;
while (reader.loadNextBatch()) {
final ValueIterableVector<Boolean> booleanVector =
(ValueIterableVector<Boolean>) reader.getVectorSchemaRoot().getVector("filter_id_lower_than_20");
assertThat(booleanVector.getValueIterable(),
(ValueIterableVector<Boolean>)
reader.getVectorSchemaRoot().getVector("filter_id_lower_than_20");
assertThat(
booleanVector.getValueIterable(),
IsIterableContainingInOrder.contains(true, true, true, false, false));
rowcount += reader.getVectorSchemaRoot().getRowCount();
}
Expand Down Expand Up @@ -625,13 +627,17 @@ public void testRunExtendedExpressionsProjectAndFilter() throws Exception {
int rowcount = 0;
while (reader.loadNextBatch()) {
final ValueIterableVector<Integer> sumVector =
(ValueIterableVector<Integer>) reader.getVectorSchemaRoot().getVector("add_two_to_column_a");
(ValueIterableVector<Integer>)
reader.getVectorSchemaRoot().getVector("add_two_to_column_a");
assertThat(sumVector.getValueIterable(), IsIterableContainingInOrder.contains(21, 3, 13));
final ValueIterableVector<Text> nameVector =
(ValueIterableVector<Text>) reader.getVectorSchemaRoot().getVector("conccat_column_a_and_b");
assertThat(nameVector.getValueIterable(),
(ValueIterableVector<Text>)
reader.getVectorSchemaRoot().getVector("conccat_column_a_and_b");
assertThat(
nameVector.getValueIterable(),
IsIterableContainingInOrder.contains(
new Text("value_19 - value_19"), new Text("value_1 - value_1"),
new Text("value_19 - value_19"),
new Text("value_1 - value_1"),
new Text("value_11 - value_11")));
rowcount += reader.getVectorSchemaRoot().getRowCount();
}
Expand Down

0 comments on commit 887d215

Please sign in to comment.