You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

25 lines
587 B

10 months ago
  1. package schema
  2. import (
  3. "fmt"
  4. "testing"
  5. )
  6. func TestWriteParquet(t *testing.T) {
  7. // create a schema_pb.RecordType
  8. recordType := NewRecordTypeBuilder().
  9. AddLongField("ID").
  10. AddLongField("CreatedAt").
  11. AddRecordField("Person", NewRecordTypeBuilder().
  12. AddStringField("Name").
  13. AddListField("emails", TypeString)).Build()
  14. fmt.Printf("RecordType: %v\n", recordType)
  15. // create a parquet schema
  16. parquetSchema, err := ToParquetSchema("example", recordType)
  17. if err != nil {
  18. t.Fatalf("ToParquetSchema failed: %v", err)
  19. }
  20. fmt.Printf("ParquetSchema: %v\n", parquetSchema)
  21. }