gh-161008: Fix missing statistics in heatmap replay output

Author: deadlovelll

Lib/profiling/sampling/_heatmap_assets/heatmap_index_template.html

@@ -66,7 +66,7 @@
                 <div class="stat-card">
                     <div class="stat-icon">&#9201;</div>
                     <div class="stat-data">
-                        <span class="stat-value"><!-- DURATION --></span>
+                        <span class="stat-value"><!-- DURATION --> sec</span>
                         <span class="stat-label">Duration</span>
                     </div>
                     <div class="stat-sparkline"></div>

Lib/profiling/sampling/heatmap_collector.py

@@ -407,6 +407,20 @@ def set_stats(self, sample_interval_usec, duration_sec, sample_rate, error_rate=
         }
         self.stats.update(kwargs)
 
+    def set_replay_stats(self, info):
+        """Restore measured statistics stored in a binary profile."""
+        duration_sec = info.get("duration_sec")
+        sample_rate = info.get("sample_rate")
+        if duration_sec is None or sample_rate is None:
+            return
+        self.set_stats(
+            self.sample_interval_usec,
+            duration_sec,
+            sample_rate,
+            error_rate=info.get("error_rate"),
+            missed_samples=info.get("missed_samples"),
+        )
+
     def process_frames(self, frames, thread_id, weight=1):
         """Process stack frames and count samples per line.
 

Lib/test/test_profiling/test_sampling_profiler/test_binary_format.py

@@ -7,10 +7,11 @@
 import struct
 import tempfile
 import unittest
+from types import SimpleNamespace
 from unittest import mock
 from collections import defaultdict
 
-from test.support import captured_stderr
+from test.support import captured_stderr, captured_stdout, run_with_locale
 
 try:
     import _remote_debugging
@@ -28,6 +29,7 @@
     )
     from profiling.sampling.binary_collector import BinaryCollector
     from profiling.sampling.binary_reader import BinaryReader, convert_binary_to_format
+    from profiling.sampling.cli import _replay_with_reader
     from profiling.sampling.constants import PROFILING_MODE_CPU
     from profiling.sampling.gecko_collector import GeckoCollector
 
@@ -1937,6 +1939,41 @@ def test_original_stats_extension_remains_readable(self):
         self.assertIsNone(info["missed_samples"])
 
 
+class TestBinaryReplayToHeatmap(unittest.TestCase):
+    @run_with_locale("LC_NUMERIC", "C")
+    def test_replay_with_and_without_persisted_stats(self):
+        frame = make_frame(__file__, 1, "hot_func")
+        for with_stats in (True, False):
+            with (
+                self.subTest(with_stats=with_stats),
+                tempfile.TemporaryDirectory() as tmp,
+                captured_stdout(),
+            ):
+                bin_path = os.path.join(tmp, "profile.bin")
+                with BinaryCollector(bin_path, 2000, compression="none") as collector:
+                    collector.collect([
+                        make_interpreter(0, [make_thread(1, [frame])])
+                    ])
+                    if with_stats:
+                        collector.set_stats(
+                            2000, 2.5, 0.4, error_rate=2.0, missed_samples=1.5
+                        )
+                args = SimpleNamespace(
+                    format="heatmap", input_file=bin_path,
+                    outfile=tmp, diff_baseline=None,
+                )
+                with BinaryReader(bin_path) as reader:
+                    _replay_with_reader(args, reader)
+                content = pathlib.Path(tmp, "index.html").read_text(
+                    encoding="utf-8"
+                )
+                if with_stats:
+                    self.assertIn('<span class="stat-value">2.5 sec</span>', content)
+                    self.assertIn('<span class="stat-value">0.4</span>', content)
+                    self.assertIn('<span class="rate-value">2.0%</span>', content)
+                    self.assertIn('<span class="rate-value">1.5%</span>', content)
+
+
 class TestBinaryReplayToJsonl(BinaryFormatTestBase):
     """Tests for binary -> JSONL replay via convert_binary_to_format."""
 

Misc/NEWS.d/next/Library/2026-10-31-18-25-26.gh-issue-161008.YotuUZ.rst

@@ -0,0 +1,2 @@
+Fix zero duration and sample rate in heatmap replay output. Patch by Timofei
+Ivankov.