Add test for array builder

Author: junli1026

common/datavalues/src/arrays/builders/boolean_test.rs

@@ -0,0 +1,68 @@
+// Copyright 2020-2021 The Datafuse Authors.
+//
+// SPDX-License-Identifier: Apache-2.0.
+
+use crate::prelude::*;
+
+#[test]
+fn test_empty_boolean_array() {
+	let mut builder = BooleanArrayBuilder::with_capacity(16);
+	let data_array: DataArray<BooleanType> = builder.finish();
+	let mut iter = data_array.into_iter();
+	assert_eq!(None, iter.next());
+	assert_eq!(true, data_array.is_empty());
+	assert_eq!(DataType::Boolean, data_array.data_type());
+}
+
+#[test]
+fn test_fill_data() {
+	let mut builder = BooleanArrayBuilder::with_capacity(16);
+	builder.append_value(true);
+	builder.append_value(false);
+	builder.append_null();
+	builder.append_option(Some(false));
+	builder.append_option(None);
+
+	let data_array: DataArray<BooleanType> = builder.finish();
+	let mut iter = data_array.into_iter();
+
+	assert_eq!(Some(Some(true)), iter.next());
+	assert_eq!(Some(Some(false)), iter.next());
+	assert_eq!(Some(None), iter.next());
+	assert_eq!(Some(Some(false)), iter.next());
+	assert_eq!(Some(None), iter.next());
+	assert_eq!(None, iter.next());
+}
+
+#[test]
+fn test_new_from_slice() {
+	let data_array: DataArray<BooleanType> = NewDataArray::new_from_slice(&[true, false]);
+	let mut iter = data_array.into_iter();
+	assert_eq!(Some(Some(true)), iter.next());
+	assert_eq!(Some(Some(false)), iter.next());
+	assert_eq!(None, iter.next());
+}
+
+#[test]
+fn test_new_from_opt_slice() {
+	let v = [Some(true), Some(false), None];
+	let data_array: DataArray<BooleanType> = NewDataArray::new_from_opt_slice(&v);
+	let mut iter = data_array.into_iter();
+	assert_eq!(Some(Some(true)), iter.next());
+	assert_eq!(Some(Some(false)), iter.next());
+	assert_eq!(Some(None), iter.next());
+	assert_eq!(None, iter.next());
+}
+
+#[test]
+fn test_new_from_opt_iter() {
+	let v = vec![None, Some(true), Some(false), None];
+	let mut iter = v.into_iter();
+	iter.next(); // move iterator and create data_array from second element
+	let data_array: DataArray<BooleanType> = NewDataArray::new_from_opt_iter(iter);
+	let mut iter = data_array.into_iter();
+	assert_eq!(Some(Some(true)), iter.next());
+	assert_eq!(Some(Some(false)), iter.next());
+	assert_eq!(Some(None), iter.next());
+	assert_eq!(None, iter.next());
+}

common/datavalues/src/arrays/builders/mod.rs

@@ -12,6 +12,15 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+#[cfg(test)]
+mod boolean_test;
+
+#[cfg(test)]
+mod primitive_test;
+
+#[cfg(test)]
+mod utf8_test;
+
 mod boolean;
 mod builder;
 mod primitive;

common/datavalues/src/arrays/builders/primitive_test.rs

@@ -0,0 +1,70 @@
+// Copyright 2020-2021 The Datafuse Authors.
+//
+// SPDX-License-Identifier: Apache-2.0.
+
+use crate::prelude::*;
+
+#[test]
+fn test_empty_array() {
+	let mut builder = PrimitiveArrayBuilder::<Int64Type>::with_capacity(16);
+	let data_array = builder.finish();
+
+	// verify empty data array and data type
+	assert_eq!(true, data_array.is_empty());
+	assert_eq!(DataType::Int64, data_array.data_type());
+}
+
+#[test]
+fn test_fill_data() {
+	let mut builder = PrimitiveArrayBuilder::<Int64Type>::with_capacity(16);
+	builder.append_value(1);
+	builder.append_option(Some(2));
+	builder.append_null();
+
+	let data_array: DataArray<Int64Type> = builder.finish();
+	let mut iter = data_array.into_iter();
+
+	// verify data array with value, option value and null
+	assert_eq!(Some(Some(1)), iter.next());
+	assert_eq!(Some(Some(2)), iter.next());
+	assert_eq!(Some(None), iter.next());
+	assert_eq!(None, iter.next());
+}
+
+#[test]
+fn test_new_from_slice() {
+	let data_array: DataArray<Int64Type> = NewDataArray::new_from_slice(&[1, 2]);
+	let mut iter = data_array.into_iter();
+
+	// verify NewDataArray::new_from_slice
+	assert_eq!(Some(Some(1)), iter.next());
+	assert_eq!(Some(Some(2)), iter.next());
+	assert_eq!(None, iter.next());
+}
+
+#[test]
+fn test_new_from_opt_slice() {
+	let v = [Some(1), Some(2), None];
+	let data_array: DataArray<Int64Type> = NewDataArray::new_from_opt_slice(&v);
+	let mut iter = data_array.into_iter();
+
+	// verify NewDataArray::new_from_opt_slice
+	assert_eq!(Some(Some(1)), iter.next());
+	assert_eq!(Some(Some(2)), iter.next());
+	assert_eq!(Some(None), iter.next());
+	assert_eq!(None, iter.next());
+}
+
+#[test]
+fn test_new_from_opt_iter() {
+	let v = vec![None, Some(1), Some(2), None];
+	let mut iter = v.into_iter();
+	iter.next(); // move iterator and create data_array from second element
+	let data_array: DataArray<Int64Type> = NewDataArray::new_from_opt_iter(iter);
+	let mut iter = data_array.into_iter();
+
+	assert_eq!(Some(Some(1)), iter.next());
+	assert_eq!(Some(Some(2)), iter.next());
+	assert_eq!(Some(None), iter.next());
+	assert_eq!(None, iter.next());
+}

common/datavalues/src/arrays/builders/utf8_test.rs

@@ -0,0 +1,55 @@
+// Copyright 2020-2021 The Datafuse Authors.
+//
+// SPDX-License-Identifier: Apache-2.0.
+
+use crate::prelude::*;
+
+#[test]
+fn test_empty_array() {
+	let mut builder = Utf8ArrayBuilder::with_capacity(16);
+	let data_array = builder.finish();
+	assert_eq!(true, data_array.is_empty());
+	assert_eq!(DataType::Utf8, data_array.data_type());
+}
+
+#[test]
+fn test_fill_data() {
+	let mut builder = Utf8ArrayBuilder::with_capacity(16);
+	builder.append_value("你好");
+	builder.append_option(Some("\u{1F378}"));
+	builder.append_null();
+
+	let data_array: DataArray<Utf8Type> = builder.finish();
+	let mut iter = data_array.into_iter();
+
+	assert_eq!(3, data_array.len());
+	assert_eq!(Some(Some("你好")), iter.next());
+	assert_eq!(Some(Some("🍸")), iter.next());
+	assert_eq!(Some(None), iter.next());
+	assert_eq!(None, iter.next());
+}
+
+#[test]
+fn test_new_from_opt_slice() {
+	let data_array: DataArray<Utf8Type> = DataArray::new_from_opt_slice(&[Some("你好"), None]);
+	let mut iter = data_array.into_iter();
+
+	assert_eq!(2, data_array.len());
+	assert_eq!(Some(Some("你好")), iter.next());
+	assert_eq!(Some(None), iter.next());
+	assert_eq!(None, iter.next());
+}
+
+#[test]
+fn test_new_from_opt_iter() {
+	let v = vec![None, Some("你好"), None];
+	let mut iter = v.into_iter();
+	iter.next(); // move iterator and create data array from second element
+	let data_array: DataArray<Utf8Type> = DataArray::new_from_opt_iter(iter);
+	let mut iter = data_array.into_iter();
+
+	assert_eq!(2, data_array.len());
+	assert_eq!(Some(Some("你好")), iter.next());
+	assert_eq!(Some(None), iter.next());
+	assert_eq!(None, iter.next());
+}