add unit test

Yifeng-Sigma · Jun 21, 2024 · 7c1abbd · 7c1abbd
1 parent 8f140d5
commit 7c1abbd
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/go/arrow/array/data_test.go b/go/arrow/array/data_test.go
@@ -60,6 +60,16 @@ func TestSizeInBytes(t *testing.T) {
 	var arrayData arrow.ArrayData = data
 	dataWithChild := NewData(&arrow.StringType{}, 10, buffers1, []arrow.ArrayData{arrayData}, 0, 0)
 
+	buffers1[0] = nil
+	dataWithNilBuffer := NewData(&arrow.StringType{}, 10, buffers1, nil, 0, 0)
+
+	t.Run("nil buffers", func(t *testing.T) {
+		expectedSize := uint64(30)
+		if actualSize := dataWithNilBuffer.SizeInBytes(); actualSize != expectedSize {
+			t.Errorf("expected size %d, got %d", expectedSize, actualSize)
+		}
+	})
+
 	t.Run("buffers only", func(t *testing.T) {
 		expectedSize := uint64(45)
 		if actualSize := data.SizeInBytes(); actualSize != expectedSize {