fmt

datafusion/core/tests/dataframe/mod.rs

@@ -29,7 +29,10 @@ use arrow::{
     },
     record_batch::RecordBatch,
 };
-use arrow_array::{Array, BooleanArray, DictionaryArray, Float32Array, Float64Array, Int8Array, UnionArray};
+use arrow_array::{
+    Array, BooleanArray, DictionaryArray, Float32Array, Float64Array, Int8Array,
+    UnionArray,
+};
 use arrow_buffer::ScalarBuffer;
 use arrow_schema::{ArrowError, UnionFields, UnionMode};
 use datafusion_functions_aggregate::count::count_udaf;
@@ -2364,17 +2367,14 @@ async fn dense_union_is_null() {
     assert_batches_sorted_eq!(expected, &result_df.collect().await.unwrap());
 }
 
-
 #[tokio::test]
 async fn boolean_dictionary_as_filter() {
     let values = vec![Some(true), Some(false), None, Some(true)];
     let keys = vec![0, 0, 1, 2, 1, 3, 1];
     let values_array = BooleanArray::from(values);
     let keys_array = Int8Array::from(keys);
-    let array = DictionaryArray::new(
-        keys_array,
-        Arc::new(values_array) as Arc<dyn Array>,
-    );
+    let array =
+        DictionaryArray::new(keys_array, Arc::new(values_array) as Arc<dyn Array>);
     let array = Arc::new(array);
 
     let field = Field::new(
@@ -2423,14 +2423,17 @@ async fn boolean_dictionary_as_filter() {
     // test nested dictionary
     let keys = vec![0, 2]; // 0 -> true, 2 -> false
     let keys_array = Int8Array::from(keys);
-    let nested_array = DictionaryArray::new(
-        keys_array,
-        array,
-    );
+    let nested_array = DictionaryArray::new(keys_array, array);
 
     let field = Field::new(
         "my_nested_dict",
-        DataType::Dictionary(Box::new(DataType::Int8), Box::new(DataType::Dictionary(Box::new(DataType::Int8), Box::new(DataType::Boolean)))),
+        DataType::Dictionary(
+            Box::new(DataType::Int8),
+            Box::new(DataType::Dictionary(
+                Box::new(DataType::Int8),
+                Box::new(DataType::Boolean),
+            )),
+        ),
         true,
     );
 
@@ -2464,5 +2467,4 @@ async fn boolean_dictionary_as_filter() {
     ];
 
     assert_batches_eq!(expected, &result_df.collect().await.unwrap());
-
 }