tool tips

Author: drisspg

test/test_memory_viz.py

@@ -12,12 +12,6 @@
 )
 
 
-def _unpack(snapshot, device=0):
-    timeline, allocs, frames, stacks, categories, max_ts = process_snapshot(snapshot, device)
-    resolved_stacks = [[frames[fi] for fi in s] for s in stacks]
-    return timeline, allocs, resolved_stacks, categories, max_ts
-
-
 DATA_DIR = Path(__file__).parent / "data"
 SNAPSHOT_PATH = DATA_DIR / "mini_snapshot.pickle"
 
@@ -85,12 +79,11 @@ def test_is_cpython_c_frame(self):
 
 class TestProcessSnapshot:
     def test_returns_correct_tuple_shape(self, snapshot):
-        timeline, allocs, frames, stacks, categories, max_ts = process_snapshot(snapshot)
+        timeline, allocs, frames, stacks, max_ts = process_snapshot(snapshot)
         assert len(timeline) > 0
         assert len(allocs) > 0
         assert len(frames) > 0
         assert len(stacks) > 0
-        assert len(categories) == len(stacks)
         assert max_ts > 0
 
     def test_stacks_reference_valid_frame_indices(self, snapshot):
@@ -131,7 +124,7 @@ def test_stack_indices_valid(self, snapshot):
 
     def test_empty_device_returns_empty(self, snapshot):
         result = process_snapshot(snapshot, device=99)
-        assert result == ([], [], [], [], [], 0)
+        assert result == ([], [], [], [], 0)
 
     def test_polygon_offsets_non_negative(self, snapshot):
         _, allocs, *_ = process_snapshot(snapshot)
@@ -152,7 +145,6 @@ def test_no_remaining_placeholders(self, snapshot):
             "__TIMELINE__",
             "__ALLOCS__",
             "__STACKS__",
-            "__CATEGORIES__",
             "__META__",
         ]:
             assert placeholder not in html