test(profiling): make memalloc test less flaky (#15769)

Author: KowalskiThomas

ddtrace/profiling/collector/memalloc.py

@@ -94,14 +94,15 @@ def snapshot(self) -> None:
             # DEV: This can happen if either _memalloc has not been started or has been stopped.
             LOG.debug("Unable to collect heap events from process %d", os.getpid(), exc_info=True)
 
-    def snapshot_and_parse_pprof(self, output_filename: str) -> Any:
+    def snapshot_and_parse_pprof(self, output_filename: str, assert_samples: bool = True) -> Any:
         """Export samples to profile, upload, and parse the pprof profile.
 
         This is similar to test_snapshot() but exports to the profile and returns
         the parsed pprof profile instead of Python objects.
 
         Args:
             output_filename: The pprof output filename prefix (without .pid.counter suffix)
+            assert_samples: Whether to assert that the profile contains samples
 
         Returns:
             Parsed pprof profile object (pprof_pb2.Profile)
@@ -123,4 +124,4 @@ def snapshot_and_parse_pprof(self, output_filename: str) -> Any:
                 "pprof_utils is not available. snapshot_and_parse_pprof() is only available in test environment."
             )
 
-        return pprof_utils.parse_newest_profile(output_filename)
+        return pprof_utils.parse_newest_profile(output_filename, assert_samples=assert_samples)

tests/profiling/collector/test_memalloc.py

@@ -550,7 +550,7 @@ def test_memory_collector_python_interface_with_allocation_tracking(tmp_path):
         tmp_path, "test_memory_collector_python_interface_with_allocation_tracking"
     )
 
-    mc = memalloc.MemoryCollector(heap_sample_size=128)
+    mc = memalloc.MemoryCollector(heap_sample_size=32)
 
     with mc:
         first_batch = []
@@ -623,11 +623,11 @@ def test_memory_collector_python_interface_with_allocation_tracking_no_deletion(
         tmp_path, "test_memory_collector_python_interface_with_allocation_tracking_no_deletion"
     )
 
-    mc = memalloc.MemoryCollector(heap_sample_size=128)
+    mc = memalloc.MemoryCollector(heap_sample_size=32)
 
     with mc:
-        # Take initial snapshot to reset allocation tracking
-        mc.snapshot_and_parse_pprof(output_filename)
+        # Take initial snapshot to reset allocation tracking (may have no samples)
+        mc.snapshot_and_parse_pprof(output_filename, assert_samples=False)
 
         first_batch = []
         for i in range(20):