Adjust memory sizes in tests

apache · Jan 12, 2024 · 0c4a8a1 · 0c4a8a1
1 parent 89cb067
commit 0c4a8a1
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/datafusion/physical-plan/src/aggregates/mod.rs b/datafusion/physical-plan/src/aggregates/mod.rs
@@ -1482,7 +1482,7 @@ mod tests {
         ))];
 
         let task_ctx = if spill {
-            new_spill_ctx(2, 1500)
+            new_spill_ctx(2, 1600)
         } else {
             Arc::new(TaskContext::default())
         };
@@ -1722,7 +1722,6 @@ mod tests {
     }
 
     #[tokio::test]
-    #[ignore]
     async fn aggregate_source_not_yielding_with_spill() -> Result<()> {
         let input: Arc<dyn ExecutionPlan> =
             Arc::new(TestYieldingExec { yield_first: false });
@@ -1739,7 +1738,6 @@ mod tests {
     }
 
     #[tokio::test]
-    #[ignore]
     async fn aggregate_source_with_yielding_with_spill() -> Result<()> {
         let input: Arc<dyn ExecutionPlan> =
             Arc::new(TestYieldingExec { yield_first: true });
@@ -1950,7 +1948,7 @@ mod tests {
         spill: bool,
     ) -> Result<()> {
         let task_ctx = if spill {
-            new_spill_ctx(2, 2886)
+            new_spill_ctx(2, 3200)
         } else {
             Arc::new(TaskContext::default())
         };