Spaces:
Sleeping
Sleeping
File size: 1,942 Bytes
3ab07bd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 | """TDD for spaceutil.encoder. Requires the real Qwen3 model - marked
`model` and skipped by default. Run with `pytest -m model`."""
from __future__ import annotations
import numpy as np
import pytest
pytestmark = pytest.mark.model
def test_encode_query_returns_normalized_vector(synthetic_embed_provenance):
from spaceutil.encoder import encode_query_via_optcg
vec = encode_query_via_optcg("red blocker with draw", synthetic_embed_provenance)
arr = np.asarray(vec, dtype=np.float32)
assert arr.shape == (synthetic_embed_provenance.embedding_dim,)
norm = float(np.linalg.norm(arr))
assert abs(norm - 1.0) < 1e-4
def test_paraphrases_are_close(synthetic_embed_provenance):
"""Loose floor - the model is small. We assert > 0.4 cosine, well
above random."""
from spaceutil.encoder import encode_query_via_optcg
a = np.asarray(
encode_query_via_optcg("red blocker", synthetic_embed_provenance),
dtype=np.float32,
)
b = np.asarray(
encode_query_via_optcg("crimson defender", synthetic_embed_provenance),
dtype=np.float32,
)
cosine = float(np.dot(a, b))
assert cosine > 0.4, f"paraphrases too distant: cosine={cosine}"
def test_model_cached_across_calls(synthetic_embed_provenance):
"""Second call must reuse the loaded model - the warmup pattern is
load-bearing for cold-start UX."""
import time
from spaceutil.encoder import encode_query_via_optcg, get_encoder
_ = get_encoder(synthetic_embed_provenance) # warmup
t0 = time.time()
_ = encode_query_via_optcg("hello world", synthetic_embed_provenance)
first = time.time() - t0
t0 = time.time()
_ = encode_query_via_optcg("goodbye world", synthetic_embed_provenance)
second = time.time() - t0
assert second < first * 5, (
f"second call ({second:.3f}s) much slower than first ({first:.3f}s) - "
"model likely re-loaded"
)
|