consider the static swap size into calculation too

Author: misrasaurabh1

codeflash/verification/pytest_plugin.py

@@ -41,22 +41,39 @@ class UnexpectedError(Exception):
 if platform.system() == "Linux":
     import resource
 
-    # We set the memory limit to 85% of total system memory when no swap exists
+    # We set the memory limit to 85% of total system memory + swap when swap exists
     swap_file_path = Path("/proc/swaps")
     swap_exists = swap_file_path.is_file()
+    swap_size = 0
+
     if swap_exists:
         with swap_file_path.open("r") as f:
-            swap_exists = len(f.readlines()) > 1  # First line is header
+            swap_lines = f.readlines()
+            swap_exists = len(swap_lines) > 1  # First line is header
+
+            if swap_exists:
+                # Parse swap size from lines after header
+                for line in swap_lines[1:]:
+                    parts = line.split()
+                    if len(parts) >= 3:
+                        # Swap size is in KB in the 3rd column
+                        try:
+                            swap_size += int(parts[2]) * 1024  # Convert KB to bytes
+                        except (ValueError, IndexError):
+                            pass
+
+    # Get total system memory
+    total_memory = os.sysconf("SC_PAGE_SIZE") * os.sysconf("SC_PHYS_PAGES")
+
+    # Add swap to total available memory if swap exists
+    if swap_exists:
+        total_memory += swap_size
 
-    if not swap_exists:
-        # Get total system memory
-        total_memory = os.sysconf("SC_PAGE_SIZE") * os.sysconf(
-            "SC_PHYS_PAGES"
-        )  # Set the memory limit to 85% of total system memory
-        memory_limit = int(total_memory * 0.85)
+    # Set the memory limit to 85% of total memory (RAM plus swap)
+    memory_limit = int(total_memory * 0.85)
 
-        # Set both soft and hard limits
-        resource.setrlimit(resource.RLIMIT_AS, (memory_limit, memory_limit))
+    # Set both soft and hard limits
+    resource.setrlimit(resource.RLIMIT_AS, (memory_limit, memory_limit))
 
 
 def pytest_addoption(parser: Parser) -> None: