tests/test_bge_m3_migration_sql.py

from pathlib import Path

MIGRATION_SQL = (
    Path(__file__).parent.parent
    / "scripts"
    / "sql"
    / "2026-05-02-bge-m3-cutover.sql"
)


def test_bge_m3_cutover_sql_uses_1024_dim_and_hnsw():
    sql = MIGRATION_SQL.read_text(encoding="utf-8").lower()
    assert "embedding vector(1024)" in sql
    assert "using hnsw (embedding vector_cosine_ops)" in sql
    assert "query_embedding vector(1024)" in sql
    assert "1 - (knowledge_chunks.embedding <=> query_embedding)" in sql