Function test_stress_c_extension

python/tests/test_c_extension.py:174–206

Stress test the C extension with large dataset.

()

def test_stress_c_extension():
    """Stress test the C extension with large dataset."""
    if not HAS_C_EXTENSION:
        return

    print("\nC Extension Stress Test")
    print("=" * 40)

    size = 100000
    tree = bplustree_c.BPlusTree(capacity=128)

    # Insert random data
    keys = list(range(size))
    random.shuffle(keys)

    start = time.perf_counter()
    for key in keys:
        tree[key] = key * 2
    insert_time = time.perf_counter() - start

    print(f"Inserted {size:,} items in {insert_time:.3f}s")
    print(f"Rate: {size/insert_time:,.0f} insertions/sec")

    # Verify all items
    start = time.perf_counter()
    for key in range(size):
        assert tree[key] == key * 2
    lookup_time = time.perf_counter() - start

    print(f"Verified {size:,} lookups in {lookup_time:.3f}s")
    print(f"Rate: {size/lookup_time:,.0f} lookups/sec")

    print("✓ Stress test passed")


if __name__ == "__main__":