Fix test imports

Author: Shiv Bhatia

datafusion/physical-plan/src/sorts/sort.rs

@@ -1441,8 +1441,10 @@ mod tests {
     use super::*;
     use crate::coalesce_partitions::CoalescePartitionsExec;
     use crate::collect;
+    use crate::empty::EmptyExec;
     use crate::execution_plan::Boundedness;
     use crate::expressions::col;
+    use crate::filter_pushdown::{FilterPushdownPhase, PushedDown};
     use crate::test;
     use crate::test::TestMemoryExec;
     use crate::test::exec::{BlockingExec, assert_strong_count_converges_to_zero};
@@ -1453,14 +1455,18 @@ mod tests {
     use arrow::datatypes::*;
     use datafusion_common::cast::as_primitive_array;
     use datafusion_common::test_util::batches_to_string;
+    use datafusion_common::config::ConfigOptions;
     use datafusion_common::{DataFusionError, Result, ScalarValue};
     use datafusion_execution::RecordBatchStream;
+    use datafusion_execution::memory_pool::{
+        GreedyMemoryPool, MemoryConsumer, MemoryPool,
+    };
     use datafusion_execution::config::SessionConfig;
     use datafusion_execution::runtime_env::RuntimeEnvBuilder;
     use datafusion_physical_expr::EquivalenceProperties;
     use datafusion_physical_expr::expressions::{Column, Literal};
 
-    use futures::{FutureExt, Stream};
+    use futures::{FutureExt, Stream, TryStreamExt};
     use insta::assert_snapshot;
 
     #[derive(Debug, Clone)]
@@ -2759,11 +2765,6 @@ mod tests {
     /// those bytes become unaccounted-for reserved memory that nobody uses.
     #[tokio::test]
     async fn test_sort_merge_reservation_transferred_not_freed() -> Result<()> {
-        use datafusion_execution::memory_pool::{
-            GreedyMemoryPool, MemoryConsumer, MemoryPool,
-        };
-        use futures::TryStreamExt;
-
         let sort_spill_reservation_bytes: usize = 10 * 1024; // 10 KB
 
         // Pool: merge reservation (10KB) + enough room for sort to work.
@@ -2877,11 +2878,6 @@ mod tests {
 
     #[test]
     fn test_sort_with_fetch_blocks_filter_pushdown() -> Result<()> {
-        use crate::empty::EmptyExec;
-        use crate::filter_pushdown::{FilterPushdownPhase, PushedDown};
-        use datafusion_common::config::ConfigOptions;
-        use datafusion_physical_expr::expressions::Column;
-
         let schema = Arc::new(Schema::new(vec![Field::new("a", DataType::Int32, false)]));
         let input = Arc::new(EmptyExec::new(Arc::clone(&schema)));
         let sort = SortExec::new(
@@ -2914,11 +2910,6 @@ mod tests {
 
     #[test]
     fn test_sort_without_fetch_allows_filter_pushdown() -> Result<()> {
-        use crate::empty::EmptyExec;
-        use crate::filter_pushdown::{FilterPushdownPhase, PushedDown};
-        use datafusion_common::config::ConfigOptions;
-        use datafusion_physical_expr::expressions::Column;
-
         let schema = Arc::new(Schema::new(vec![Field::new("a", DataType::Int32, false)]));
         let input = Arc::new(EmptyExec::new(Arc::clone(&schema)));
         let sort = SortExec::new(