This is an automated email from the ASF dual-hosted git repository.
tustvold pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/arrow-rs.git
The following commit(s) were added to refs/heads/master by this push:
new 6144f1995 Add two `from` methods for `FixedSizeBinaryArray` (#1854)
6144f1995 is described below
commit 6144f1995fc09ec8613c6624fb76a4e75f172c2d
Author: Remzi Yang <[email protected]>
AuthorDate: Mon Jun 13 19:09:51 2022 +0800
Add two `from` methods for `FixedSizeBinaryArray` (#1854)
* add from
Signed-off-by: remzi <[email protected]>
* fmt
Signed-off-by: remzi <[email protected]>
* add tests
Signed-off-by: remzi <[email protected]>
---
arrow/src/array/array_binary.rs | 70 +++++++++++++++++++++++++++++++++++++++++
1 file changed, 70 insertions(+)
diff --git a/arrow/src/array/array_binary.rs b/arrow/src/array/array_binary.rs
index af55854a5..b1fc06d36 100644
--- a/arrow/src/array/array_binary.rs
+++ b/arrow/src/array/array_binary.rs
@@ -700,6 +700,18 @@ impl From<FixedSizeListArray> for FixedSizeBinaryArray {
}
}
+impl From<Vec<Option<&[u8]>>> for FixedSizeBinaryArray {
+ fn from(v: Vec<Option<&[u8]>>) -> Self {
+ Self::try_from_sparse_iter(v.into_iter()).unwrap()
+ }
+}
+
+impl From<Vec<&[u8]>> for FixedSizeBinaryArray {
+ fn from(v: Vec<&[u8]>) -> Self {
+ Self::try_from_iter(v.into_iter()).unwrap()
+ }
+}
+
impl fmt::Debug for FixedSizeBinaryArray {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "FixedSizeBinaryArray<{}>\n[\n", self.value_length())?;
@@ -1713,6 +1725,64 @@ mod tests {
assert_eq!(5, arr.len())
}
+ #[test]
+ fn test_fixed_size_binary_array_from_vec() {
+ let values = vec!["one".as_bytes(), b"two", b"six", b"ten"];
+ let array = FixedSizeBinaryArray::from(values);
+ assert_eq!(array.len(), 4);
+ assert_eq!(array.null_count(), 0);
+ assert_eq!(array.value(0), b"one");
+ assert_eq!(array.value(1), b"two");
+ assert_eq!(array.value(2), b"six");
+ assert_eq!(array.value(3), b"ten");
+ assert!(!array.is_null(0));
+ assert!(!array.is_null(1));
+ assert!(!array.is_null(2));
+ assert!(!array.is_null(3));
+ }
+
+ #[test]
+ #[should_panic(expected = "Nested array size mismatch: one is 3, and the
other is 5")]
+ fn test_fixed_size_binary_array_from_vec_incorrect_length() {
+ let values = vec!["one".as_bytes(), b"two", b"three", b"four"];
+ let _ = FixedSizeBinaryArray::from(values);
+ }
+
+ #[test]
+ fn test_fixed_size_binary_array_from_opt_vec() {
+ let values = vec![
+ Some("one".as_bytes()),
+ Some(b"two"),
+ None,
+ Some(b"six"),
+ Some(b"ten"),
+ ];
+ let array = FixedSizeBinaryArray::from(values);
+ assert_eq!(array.len(), 5);
+ assert_eq!(array.value(0), b"one");
+ assert_eq!(array.value(1), b"two");
+ assert_eq!(array.value(3), b"six");
+ assert_eq!(array.value(4), b"ten");
+ assert!(!array.is_null(0));
+ assert!(!array.is_null(1));
+ assert!(array.is_null(2));
+ assert!(!array.is_null(3));
+ assert!(!array.is_null(4));
+ }
+
+ #[test]
+ #[should_panic(expected = "Nested array size mismatch: one is 3, and the
other is 5")]
+ fn test_fixed_size_binary_array_from_opt_vec_incorrect_length() {
+ let values = vec![
+ Some("one".as_bytes()),
+ Some(b"two"),
+ None,
+ Some(b"three"),
+ Some(b"four"),
+ ];
+ let _ = FixedSizeBinaryArray::from(values);
+ }
+
#[test]
fn test_binary_array_all_null() {
let data = vec![None];
