arrow/array/compare.go

28 func RecordEqual(left, right Record) bool {
30 	case left.NumCols() != right.NumCols():
32 	case left.NumRows() != right.NumRows():
38 		rc := right.Column(i)
48 func RecordApproxEqual(left, right Record, opts ...EqualOption) bool {
50 	case left.NumCols() != right.NumCols():
52 	case left.NumRows() != right.NumRows():
60 		rc := right.Column(i)
71 func chunkedBinaryApply(left, right *Chunked, fn func(left Interface, lbeg, lend int64, right Inter…
82 			cleft, cright = left.Chunk(leftIdx), right.Chunk(rightIdx)
108 func ChunkedEqual(left, right *Chunked) bool {
110 	case left == right:
112 	case left.length != right.length:
114 	case left.nulls != right.nulls:
116 	case !arrow.TypeEqual(left.dtype, right.dtype):
121 …chunkedBinaryApply(left, right, func(left Interface, lbeg, lend int64, right Interface, rbeg, rend…
122 		isequal = ArraySliceEqual(left, lbeg, lend, right, rbeg, rend)
131 func ChunkedApproxEqual(left, right *Chunked, opts ...EqualOption) bool {
133 	case left == right:
135 	case left.length != right.length:
137 	case left.nulls != right.nulls:
139 	case !arrow.TypeEqual(left.dtype, right.dtype):
144 …chunkedBinaryApply(left, right, func(left Interface, lbeg, lend int64, right Interface, rbeg, rend…
145 		isequal = ArraySliceApproxEqual(left, lbeg, lend, right, rbeg, rend, opts...)
153 func TableEqual(left, right Table) bool {
155 	case left.NumCols() != right.NumCols():
157 	case left.NumRows() != right.NumRows():
163 		rc := right.Column(i)
176 func TableApproxEqual(left, right Table, opts ...EqualOption) bool {
178 	case left.NumCols() != right.NumCols():
180 	case left.NumRows() != right.NumRows():
186 		rc := right.Column(i)
199 func ArrayEqual(left, right Interface) bool {
201 	case !baseArrayEqual(left, right):
217 		r := right.(*Boolean)
220 		r := right.(*FixedSizeBinary)
223 		r := right.(*Binary)
226 		r := right.(*String)
229 		r := right.(*Int8)
232 		r := right.(*Int16)
235 		r := right.(*Int32)
238 		r := right.(*Int64)
241 		r := right.(*Uint8)
244 		r := right.(*Uint16)
247 		r := right.(*Uint32)
250 		r := right.(*Uint64)
253 		r := right.(*Float16)
256 		r := right.(*Float32)
259 		r := right.(*Float64)
262 		r := right.(*Decimal128)
265 		r := right.(*Date32)
268 		r := right.(*Date64)
271 		r := right.(*Time32)
274 		r := right.(*Time64)
277 		r := right.(*Timestamp)
280 		r := right.(*List)
283 		r := right.(*FixedSizeList)
286 		r := right.(*Struct)
289 		r := right.(*MonthInterval)
292 		r := right.(*DayTimeInterval)
295 		r := right.(*MonthDayNanoInterval)
298 		r := right.(*Duration)
301 		r := right.(*Map)
304 		r := right.(ExtensionArray)
312 func ArraySliceEqual(left Interface, lbeg, lend int64, right Interface, rbeg, rend int64) bool {
315 	r := NewSlice(right, rbeg, rend)
322 func ArraySliceApproxEqual(left Interface, lbeg, lend int64, right Interface, rbeg, rend int64, opt…
325 	r := NewSlice(right, rbeg, rend)
401 func ArrayApproxEqual(left, right Interface, opts ...EqualOption) bool {
403 	return arrayApproxEqual(left, right, opt)
406 func arrayApproxEqual(left, right Interface, opt equalOption) bool {
408 	case !baseArrayEqual(left, right):
424 		r := right.(*Boolean)
427 		r := right.(*FixedSizeBinary)
430 		r := right.(*Binary)
433 		r := right.(*String)
436 		r := right.(*Int8)
439 		r := right.(*Int16)
442 		r := right.(*Int32)
445 		r := right.(*Int64)
448 		r := right.(*Uint8)
451 		r := right.(*Uint16)
454 		r := right.(*Uint32)
457 		r := right.(*Uint64)
460 		r := right.(*Float16)
463 		r := right.(*Float32)
466 		r := right.(*Float64)
469 		r := right.(*Decimal128)
472 		r := right.(*Date32)
475 		r := right.(*Date64)
478 		r := right.(*Time32)
481 		r := right.(*Time64)
484 		r := right.(*Timestamp)
487 		r := right.(*List)
490 		r := right.(*FixedSizeList)
493 		r := right.(*Struct)
496 		r := right.(*MonthInterval)
499 		r := right.(*DayTimeInterval)
502 		r := right.(*MonthDayNanoInterval)
505 		r := right.(*Duration)
508 		r := right.(*Map)
511 		r := right.(ExtensionArray)
520 func baseArrayEqual(left, right Interface) bool {
522 	case left.Len() != right.Len():
524 	case left.NullN() != right.NullN():
526 …case !arrow.TypeEqual(left.DataType(), right.DataType()): // We do not check for metadata as in th…
528 	case !validityBitmapEqual(left, right):
534 func validityBitmapEqual(left, right Interface) bool {
537 	if n != right.Len() {
541 		if left.IsNull(i) != right.IsNull(i) {
548 func arrayApproxEqualFloat16(left, right *Float16, opt equalOption) bool {
553 		if !opt.f16(left.Value(i), right.Value(i)) {
560 func arrayApproxEqualFloat32(left, right *Float32, opt equalOption) bool {
565 		if !opt.f32(left.Value(i), right.Value(i)) {
572 func arrayApproxEqualFloat64(left, right *Float64, opt equalOption) bool {
577 		if !opt.f64(left.Value(i), right.Value(i)) {
584 func arrayApproxEqualList(left, right *List, opt equalOption) bool {
592 			r := right.newListValue(i)
603 func arrayApproxEqualFixedSizeList(left, right *FixedSizeList, opt equalOption) bool {
611 			r := right.newListValue(i)
622 func arrayApproxEqualStruct(left, right *Struct, opt equalOption) bool {
624 		rf := right.fields[i]