def test_collection(client):
    """Create a test collection (and thus a graph) for testing, then delete it
    afterwards."""
    collection_id = client.collections.create(
        name=f"Test Collection {uuid.uuid4()}",
        description="A sample collection for graph tests",
    ).results.id

    yield collection_id
    # Cleanup if needed
    # If there's a deletion endpoint for collections, call it here.
    client.collections.delete(id=collection_id)
Create a test collection (and thus a graph) for testing, then delete it
    afterwards.
py/tests/integration/test_graphs.py