use of org.apache.arrow.vector.types.pojo.Schema in project parquet-mr by apache.
the class TestSchemaConverter method testParquetInt64TimestampMillisToArrow.
@Test
public void testParquetInt64TimestampMillisToArrow() {
MessageType parquet = Types.buildMessage().addField(Types.optional(INT64).as(TIMESTAMP_MILLIS).named("a")).named("root");
Schema expected = new Schema(asList(field("a", new ArrowType.Timestamp(TimeUnit.MILLISECOND, "UTC"))));
Assert.assertEquals(expected, converter.fromParquet(parquet).getArrowSchema());
}
use of org.apache.arrow.vector.types.pojo.Schema in project parquet-mr by apache.
the class TestSchemaConverter method testParquetInt32TimeMillisToArrow.
@Test
public void testParquetInt32TimeMillisToArrow() {
MessageType parquet = Types.buildMessage().addField(Types.optional(INT32).as(TIME_MILLIS).named("a")).named("root");
Schema expected = new Schema(asList(field("a", new ArrowType.Time(TimeUnit.MILLISECOND, 32))));
Assert.assertEquals(expected, converter.fromParquet(parquet).getArrowSchema());
}
use of org.apache.arrow.vector.types.pojo.Schema in project parquet-mr by apache.
the class TestSchemaConverter method testParquetFixedBinaryToArrowDecimal.
@Test
public void testParquetFixedBinaryToArrowDecimal() {
MessageType parquet = Types.buildMessage().addField(Types.optional(FIXED_LEN_BYTE_ARRAY).length(5).as(DECIMAL).precision(8).scale(2).named("a")).named("root");
Schema expected = new Schema(asList(field("a", new ArrowType.Decimal(8, 2))));
Assert.assertEquals(expected, converter.fromParquet(parquet).getArrowSchema());
}
use of org.apache.arrow.vector.types.pojo.Schema in project parquet-mr by apache.
the class TestSchemaConverter method testRepeatedParquetToArrow.
@Test
public void testRepeatedParquetToArrow() throws IOException {
Schema arrow = converter.fromParquet(Paper.schema).getArrowSchema();
assertEquals(paperArrowSchema, arrow);
}
use of org.apache.arrow.vector.types.pojo.Schema in project parquet-mr by apache.
the class TestSchemaConverter method testArrowTimestampMicrosecondToParquet.
@Test
public void testArrowTimestampMicrosecondToParquet() {
MessageType expected = converter.fromArrow(new Schema(asList(field("a", new ArrowType.Timestamp(TimeUnit.MICROSECOND, "UTC"))))).getParquetSchema();
Assert.assertEquals(expected, Types.buildMessage().addField(Types.optional(INT64).as(TIMESTAMP_MICROS).named("a")).named("root"));
}
Aggregations