Class TestMemoryLeaks

python/tests/test_memory_leaks.py:18–267

Test for memory leaks in various operations.

@pytest.mark.slow
class TestMemoryLeaks:
    """Test for memory leaks in various operations."""

    def test_insertion_deletion_cycle_no_leak(self):
        """Test that insertion/deletion cycles don't leak memory."""
        tree = BPlusTreeMap()

        # Track object count before operations
        gc.collect()
        initial_objects = len(gc.get_objects())

        # Perform multiple insertion/deletion cycles (reduced for CI)
        for cycle in range(3):
            # Insert items (reduced count for CI)
            for i in range(500):
                tree[i] = f"value_{i}_{cycle}"

            # Delete all items
            for i in range(500):
                del tree[i]

        # Force garbage collection
        gc.collect()
        final_objects = len(gc.get_objects())

        # Object count should not grow significantly
        # Allow some variance for internal Python operations
        growth = final_objects - initial_objects
        assert (
            growth < 50
        ), f"MEMORY LEAK DETECTED: {growth} new objects after cycles (threshold: 50)"

    def test_deleted_values_are_released(self):
        """Test that deleted values are properly released."""
        tree = BPlusTreeMap()

        # Create objects that we can track
        class TrackedObject:
            def __init__(self, value):
                self.value = value

        # Insert tracked objects
        objects = []
        weak_refs = []
        for i in range(100):
            obj = TrackedObject(f"value_{i}")
            objects.append(obj)
            weak_refs.append(weakref.ref(obj))
            tree[i] = obj

        # Clear our references but keep weak references
        objects.clear()

        # Delete from tree
        for i in range(100):
            del tree[i]

        # Force garbage collection