Skip to content

Commit 15e8ee2

Browse files
committed
fix style
1 parent ba07795 commit 15e8ee2

File tree

1 file changed

+16
-22
lines changed
  • native/spark-expr/src/conversion_funcs

1 file changed

+16
-22
lines changed

native/spark-expr/src/conversion_funcs/cast.rs

Lines changed: 16 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ use crate::{EvalMode, SparkError, SparkResult};
2121
use arrow::array::builder::StringBuilder;
2222
use arrow::array::{DictionaryArray, ListArray, StringArray, StructArray};
2323
use arrow::compute::can_cast_types;
24+
use arrow::datatypes::DataType::Utf8;
2425
use arrow::datatypes::{ArrowDictionaryKeyType, ArrowNativeType, DataType, Schema};
2526
use arrow::{
2627
array::{
@@ -59,7 +60,6 @@ use std::{
5960
num::Wrapping,
6061
sync::Arc,
6162
};
62-
use arrow::datatypes::DataType::Utf8;
6363

6464
static TIMESTAMP_FORMAT: Option<&str> = Some("%Y-%m-%d %H:%M:%S%.f");
6565

@@ -1180,8 +1180,9 @@ fn cast_array_to_string(
11801180
} else {
11811181
str.clear();
11821182
let value_ref = array.value(row_index);
1183-
let native_cast_result= cast_array(value_ref, &Utf8, &spark_cast_options).unwrap();
1184-
let string_array = native_cast_result.as_any()
1183+
let native_cast_result = cast_array(value_ref, &Utf8, &spark_cast_options).unwrap();
1184+
let string_array = native_cast_result
1185+
.as_any()
11851186
.downcast_ref::<StringArray>()
11861187
.unwrap();
11871188
let mut any_fields_written = false;
@@ -1196,7 +1197,6 @@ fn cast_array_to_string(
11961197
str.push_str("]");
11971198
builder.append_value(&str);
11981199
}
1199-
12001200
}
12011201
Ok(Arc::new(builder.finish()))
12021202
}
@@ -2790,24 +2790,22 @@ mod tests {
27902790

27912791
#[test]
27922792
fn test_cast_string_array_to_string() {
2793-
use arrow::buffer::OffsetBuffer;
27942793
use arrow::array::ListArray;
2795-
let values_array = StringArray::from(vec![
2796-
Some("a"), Some("b"), Some("c"),
2797-
Some("a"), None,
2798-
None
2799-
]);
2794+
use arrow::buffer::OffsetBuffer;
2795+
let values_array =
2796+
StringArray::from(vec![Some("a"), Some("b"), Some("c"), Some("a"), None, None]);
28002797
let offsets_buffer = OffsetBuffer::<i32>::new(vec![0, 3, 5, 6, 6].into());
28012798
let item_field = Arc::new(Field::new("item", DataType::Utf8, true));
28022799
let list_array = Arc::new(ListArray::new(
28032800
item_field,
28042801
offsets_buffer,
28052802
Arc::new(values_array),
2806-
None
2803+
None,
28072804
));
28082805
let string_array = cast_array_to_string(
28092806
&list_array,
2810-
&SparkCastOptions::new(EvalMode::Legacy, "UTC", false))
2807+
&SparkCastOptions::new(EvalMode::Legacy, "UTC", false),
2808+
)
28112809
.unwrap();
28122810
let string_array = string_array.as_string::<i32>();
28132811
assert_eq!(r#"[a, b, c]"#, string_array.value(0));
@@ -2818,30 +2816,26 @@ mod tests {
28182816

28192817
#[test]
28202818
fn test_cast_i32_array_to_string() {
2821-
use arrow::buffer::OffsetBuffer;
28222819
use arrow::array::ListArray;
2823-
let values_array = Int32Array::from(vec![
2824-
Some(1), Some(2), Some(3),
2825-
Some(1), None,
2826-
None
2827-
]);
2820+
use arrow::buffer::OffsetBuffer;
2821+
let values_array = Int32Array::from(vec![Some(1), Some(2), Some(3), Some(1), None, None]);
28282822
let offsets_buffer = OffsetBuffer::<i32>::new(vec![0, 3, 5, 6, 6].into());
28292823
let item_field = Arc::new(Field::new("item", DataType::Int32, true));
28302824
let list_array = Arc::new(ListArray::new(
28312825
item_field,
28322826
offsets_buffer,
28332827
Arc::new(values_array),
2834-
None
2828+
None,
28352829
));
28362830
let string_array = cast_array_to_string(
28372831
&list_array,
2838-
&SparkCastOptions::new(EvalMode::Legacy, "UTC", false))
2839-
.unwrap();
2832+
&SparkCastOptions::new(EvalMode::Legacy, "UTC", false),
2833+
)
2834+
.unwrap();
28402835
let string_array = string_array.as_string::<i32>();
28412836
assert_eq!(r#"[1, 2, 3]"#, string_array.value(0));
28422837
assert_eq!(r#"[1, NULL]"#, string_array.value(1));
28432838
assert_eq!(r#"[NULL]"#, string_array.value(2));
28442839
assert_eq!(r#"[]"#, string_array.value(3));
28452840
}
2846-
28472841
}

0 commit comments

Comments
 (0)