chore

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

@@ -970,7 +970,7 @@ mod tests {
         let schema = batch1.schema();
         let expected_batches = vec![batch1.clone(), batch2.clone()];
         let input = MockExec::new(vec![Ok(batch1), Ok(batch2)], schema);
-        let partitioning = Partitioning::RoundRobinBatch(1);
+        let partitioning = Partitioning::OnDemand(1);
 
         let exec =
             OnDemandRepartitionExec::try_new(Arc::new(input), partitioning).unwrap();
@@ -1143,7 +1143,7 @@ mod tests {
         let schema = test_schema();
         let partition = create_vec_batches(50);
         let input_partitions = vec![partition];
-        let partitioning = Partitioning::RoundRobinBatch(4);
+        let partitioning = Partitioning::OnDemand(4);
 
         // setup up context
         let runtime = RuntimeEnvBuilder::default()
@@ -1198,7 +1198,7 @@ mod tests {
         let union = UnionExec::new(vec![source1, source2]);
         let exec = OnDemandRepartitionExec::try_new(
             Arc::new(union),
-            Partitioning::RoundRobinBatch(10),
+            Partitioning::OnDemand(10),
         )
         .unwrap()
         .with_preserve_order();