import pytest.py

import pytest
import pinecone

def test_upsert_vector():
    # Initialize Pinecone connection
    pinecone.init(api_key="your-pinecone-api-key", environment="us-west1-gcp")
    index = pinecone.Index("example-index")

    # Upsert a vector
    response = index.upsert(
        vectors=[{"id": "test1", "values": [1.0, 2.0], "metadata": {"genre": "comedy"}}],
        namespace="example-namespace"
    )

    assert response["status"] == "success"

def test_query_vector():
    # Initialize Pinecone connection
    pinecone.init(api_key="your-pinecone-api-key", environment="us-west1-gcp")
    index = pinecone.Index("example-index")

    # Query a vector
    query_result = index.query(
        vector=[1.0, 2.0],
        top_k=1,
        namespace="example-namespace"
    )

    assert len(query_result["matches"]) > 0