Function makeTestFile

tempodb/encoding/vparquet5/row_number_iterator_test.go:167–183

(t testing.TB, traces []Trace)

}

func makeTestFile(t testing.TB, traces []Trace) *parquet.File {
	var buf bytes.Buffer

	w := parquet.NewGenericWriter[Trace](&buf)
	n, err := w.Write(traces)
	require.NoError(t, err)
	require.Equal(t, len(traces), n)

	err = w.Close()
	require.NoError(t, err)

	data := buf.Bytes()
	pf, err := parquet.OpenFile(bytes.NewReader(data), int64(len(data)))
	require.NoError(t, err)

	return pf
}