Skip to content

Commit

Permalink
fix(tests): WriterTestSuite.handleNulls should not overwrite columns (#…
Browse files Browse the repository at this point in the history
  • Loading branch information
disq authored Oct 4, 2024
1 parent af8ac87 commit 08e18e2
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions plugin/nulls.go
Original file line number Diff line number Diff line change
Expand Up @@ -93,10 +93,11 @@ func (s *WriterTestSuite) replaceNullsByEmptyNestedArray(arr arrow.Array) arrow.

func (s *WriterTestSuite) handleNulls(record arrow.Record) arrow.Record {
cols := record.Columns()
newCols := make([]arrow.Array, len(cols))
for c, col := range cols {
cols[c] = s.handleNullsArray(col)
newCols[c] = s.handleNullsArray(col)
}
return array.NewRecord(record.Schema(), cols, record.NumRows())
return array.NewRecord(record.Schema(), newCols, record.NumRows())
}

func (s *WriterTestSuite) handleNullsArray(arr arrow.Array) arrow.Array {
Expand Down

0 comments on commit 08e18e2

Please sign in to comment.