Examples with Schema org.apache.arrow.vector.types.pojo.Schema used on opensource projects

Example 11 with Schema

use of org.apache.arrow.vector.types.pojo.Schema in project parquet-mr by apache.

the class TestSchemaConverter method testArrowTimestampMillisecondToParquet.

@Test
public void testArrowTimestampMillisecondToParquet() {
    MessageType expected = converter.fromArrow(new Schema(asList(field("a", new ArrowType.Timestamp(TimeUnit.MILLISECOND, "UTC"))))).getParquetSchema();
    Assert.assertEquals(expected, Types.buildMessage().addField(Types.optional(INT64).as(TIMESTAMP_MILLIS).named("a")).named("root"));
}

Example 12 with Schema

use of org.apache.arrow.vector.types.pojo.Schema in project parquet-mr by apache.

the class TestSchemaConverter method testParquetInt64TimeMicrosToArrow.

@Test
public void testParquetInt64TimeMicrosToArrow() {
    MessageType parquet = Types.buildMessage().addField(Types.optional(INT64).as(TIME_MICROS).named("a")).named("root");
    Schema expected = new Schema(asList(field("a", new ArrowType.Time(TimeUnit.MICROSECOND, 64))));
    Assert.assertEquals(expected, converter.fromParquet(parquet).getArrowSchema());
}

Example 13 with Schema

use of org.apache.arrow.vector.types.pojo.Schema in project parquet-mr by apache.

the class TestSchemaConverter method testArrowTimeMillisecondToParquet.

@Test
public void testArrowTimeMillisecondToParquet() {
    MessageType expected = converter.fromArrow(new Schema(asList(field("a", new ArrowType.Time(TimeUnit.MILLISECOND, 32))))).getParquetSchema();
    Assert.assertEquals(expected, Types.buildMessage().addField(Types.optional(INT32).as(timeType(false, MILLIS)).named("a")).named("root"));
}

Example 14 with Schema

use of org.apache.arrow.vector.types.pojo.Schema in project parquet-mr by apache.

the class TestSchemaConverter method testParquetInt64TimestampMicrosToArrow.

@Test
public void testParquetInt64TimestampMicrosToArrow() {
    MessageType parquet = Types.buildMessage().addField(Types.optional(INT64).as(TIMESTAMP_MICROS).named("a")).named("root");
    Schema expected = new Schema(asList(field("a", new ArrowType.Timestamp(TimeUnit.MICROSECOND, "UTC"))));
    Assert.assertEquals(expected, converter.fromParquet(parquet).getArrowSchema());
}

Example 15 with Schema

use of org.apache.arrow.vector.types.pojo.Schema in project parquet-mr by apache.

the class TestSchemaConverter method testParquetFixedBinaryToArrow.

@Test
public void testParquetFixedBinaryToArrow() {
    MessageType parquet = Types.buildMessage().addField(Types.optional(FIXED_LEN_BYTE_ARRAY).length(12).named("a")).named("root");
    Schema expected = new Schema(asList(field("a", new ArrowType.Binary())));
    Assert.assertEquals(expected, converter.fromParquet(parquet).getArrowSchema());
}