Function createNilIterTestFile

pkg/parquetquery/nil_iter_test.go:42–66

(t testing.TB, count int)

}

func createNilIterTestFile(t testing.TB, count int) *parquet.File {
	type Attr struct {
		Key string `parquet:",snappy,dict"`
	}
	type T struct {
		Attrs []Attr `parquet:"attrs,list"`
	}
	genericKeys := []string{"key-1", "key-2", "key-3", "key-4", "key-5"}

	rows := []T{}
	for i := range count {
		keys := genericKeys
		if i%3 == 0 {
			keys = append(keys, "special-key")
		}
		attrs := make([]Attr, 0, len(keys))
		for _, key := range keys {
			attrs = append(attrs, Attr{Key: key})
		}
		rows = append(rows, T{Attrs: attrs})
	}

	pf := createFileWith(t, context.Background(), rows)
	return pf
}