Testing Index Operations

import pytest, uuid
from pinecone import Pinecone, ServerlessSpec

@pytest.fixture(scope="session")
def index():
    pc = Pinecone(api_key=os.environ["PINECONE_API_KEY"])
    name = f"test-{uuid.uuid4().hex[:8]}"
    pc.create_index(name=name, dimension=8, metric="cosine",
        spec=ServerlessSpec(cloud="aws", region="us-east-1"))
    yield pc.Index(name)
    pc.delete_index(name)

from unittest.mock import MagicMock

idx = MagicMock()
idx.query.return_value = {"matches": [
    {"id": "a", "score": 0.9, "metadata": {"t": "hi"}},
]}
assert my_search_fn(idx, "q")[0] == "a"

def test_upsert(index):
    index.upsert(vectors=[{"id": "t1", "values": [0.1]*8}])
    time.sleep(2)
    assert index.fetch(ids=["t1"])["vectors"]["t1"]

def test_query(index):
    index.upsert(vectors=[{"id": "q1", "values": [1.0]+[0]*7}])
    time.sleep(2)
    r = index.query(vector=[1.0]+[0]*7, top_k=1)
    assert r["matches"][0]["id"] == "q1"

def test_filter(index):
    index.upsert(vectors=[
        {"id": "a", "values": [0.1]*8, "metadata": {"cat": "x"}},
        {"id": "b", "values": [0.1]*8, "metadata": {"cat": "y"}},
    ])
    time.sleep(2)
    r = index.query(vector=[0.1]*8, top_k=5, filter={"cat": "x"})
    assert all(m["id"] == "a" for m in r["matches"])

def test_namespaces(index):
    index.upsert(vectors=[{"id": "n", "values": [0.5]*8}], namespace="ns1")
    time.sleep(2)
    assert index.fetch(ids=["n"], namespace="ns1")["vectors"]
    assert not index.fetch(ids=["n"], namespace="ns2")["vectors"]

jobs:
  test:
    runs-on: ubuntu-latest
    env:
      PINECONE_API_KEY: ${{ secrets.PINECONE_TEST_KEY }}
    steps:
      - uses: actions/checkout@v4
      - uses: actions/setup-python@v5
        with: { python-version: "3.12" }
      - run: pip install -r requirements.txt
      - run: pytest -v