SPARK-54449: add flag to UT

VindhyaG · VindhyaG · commit a2f4e27ff910 · 2025-12-09T09:04:01.000+05:30
diff --git a/core/src/test/scala/org/apache/spark/memory/UnifiedMemoryManagerSuite.scala b/core/src/test/scala/org/apache/spark/memory/UnifiedMemoryManagerSuite.scala
@@ -48,6 +48,7 @@ class UnifiedMemoryManagerSuite extends MemoryManagerSuite with PrivateMethodTes
       .set(TEST_MEMORY, maxOnHeapExecutionMemory)
       .set(MEMORY_OFFHEAP_SIZE, maxOffHeapExecutionMemory)
       .set(MEMORY_STORAGE_FRACTION, storageFraction)
+      .set(MEMORY_OFFHEAP_ENABLED, true)
     UnifiedMemoryManager(conf, numCores = 1)
   }
 

Original file line number	Diff line number	Diff line change
`@@ -48,6 +48,7 @@ class UnifiedMemoryManagerSuite extends MemoryManagerSuite with PrivateMethodTes`
`48`	`48`	`.set(TEST_MEMORY, maxOnHeapExecutionMemory)`
`49`	`49`	`.set(MEMORY_OFFHEAP_SIZE, maxOffHeapExecutionMemory)`
`50`	`50`	`.set(MEMORY_STORAGE_FRACTION, storageFraction)`
	`51`	`+ .set(MEMORY_OFFHEAP_ENABLED, true)`
`51`	`52`	`UnifiedMemoryManager(conf, numCores = 1)`
`52`	`53`	`}`
`53`	`54`