Merge pull request #1468 from junli1026/jun/add-test

Add test for array builder

Author: databend-bot

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

@@ -0,0 +1,78 @@
+// Copyright 2020 Datafuse Labs.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+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,80 @@
+// Copyright 2020 Datafuse Labs.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+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,65 @@
+// Copyright 2020 Datafuse Labs.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+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());
+}