update unit test

Author: jizezhang

datafusion/physical-plan/src/repartition/mod.rs

@@ -1595,7 +1595,6 @@ mod tests {
     };
 
     use arrow::array::{ArrayRef, StringArray, UInt32Array};
-    use arrow::compute::sort;
     use arrow::datatypes::{DataType, Field, Schema};
     use datafusion_common::cast::as_string_array;
     use datafusion_common::exec_err;
@@ -1983,20 +1982,16 @@ mod tests {
         });
         let batches_with_drop = crate::common::collect(output_stream1).await.unwrap();
 
-        assert_eq!(1, batches_with_drop.len());
-        assert_eq!(1, batches_without_drop.len());
+        let items_vec_with_drop = str_batches_to_vec(&batches_with_drop);
+        let items_set_with_drop: HashSet<&str> =
+            items_vec_with_drop.iter().copied().collect();
 
         assert_eq!(
-            sort(batches_without_drop[0].column(0), None)
-                .unwrap()
-                .as_any()
-                .downcast_ref::<StringArray>()
-                .unwrap(),
-            sort(batches_with_drop[0].column(0), None)
-                .unwrap()
-                .as_any()
-                .downcast_ref::<StringArray>()
-                .unwrap()
+            items_set_with_drop.symmetric_difference(&items_set).count(),
+            0,
+            "items with drop {:?} and without drop {:?} are different",
+            items_set_with_drop,
+            items_set
         );
     }