record type begin and end

This commit is contained in:
chrislu
2024-04-28 13:00:52 -07:00
parent 0da1b7b0f9
commit 189a7fc90e
4 changed files with 55 additions and 41 deletions

View File

@@ -18,16 +18,21 @@ func TestToParquetLevels(t *testing.T) {
{
name: "nested type",
args: args{
NewRecordTypeBuilder().
RecordTypeBegin().
SetField("ID", TypeLong).
SetField("CreatedAt", TypeLong).
SetRecordField("Person", NewRecordTypeBuilder().
SetField("zName", TypeString).
SetField("emails", ListOf(TypeString))).
SetRecordField("Person",
RecordTypeBegin().
SetField("zName", TypeString).
SetField("emails", ListOf(TypeString)).
RecordTypeEnd()).
SetField("Company", TypeString).
SetRecordField("Address", NewRecordTypeBuilder().
SetField("Street", TypeString).
SetField("City", TypeString)).Build(),
SetRecordField("Address",
RecordTypeBegin().
SetField("Street", TypeString).
SetField("City", TypeString).
RecordTypeEnd()).
RecordTypeEnd(),
},
want: &ParquetLevels{
startColumnIndex: 0,