From 0c4a8a1423ea787b89b8bbb8d83550bbc227849e Mon Sep 17 00:00:00 2001
From: Andrew Lamb <andrew@nerdnetworks.org>
Date: Fri, 12 Jan 2024 16:29:02 -0500
Subject: [PATCH] Adjust memory sizes in tests

---
 datafusion/physical-plan/src/aggregates/mod.rs | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/datafusion/physical-plan/src/aggregates/mod.rs b/datafusion/physical-plan/src/aggregates/mod.rs
index 48b7284c7cfd..facd601955b6 100644
--- 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())
         };