55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
|
"""批次索引：列表走 DB，重建走扫盘。"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import time
|
|||
|
|
|
|||
|
|
from as_platform.labeling.batch_index import (
|
|||
|
|
index_is_empty,
|
|||
|
|
list_batches_from_index,
|
|||
|
|
rebuild_batch_index,
|
|||
|
|
)
|
|||
|
|
from as_platform.labeling.service import list_labeling_batches
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_rebuild_and_list_from_index():
|
|||
|
|
r = rebuild_batch_index()
|
|||
|
|
assert r["ok"] is True
|
|||
|
|
assert r["count"] >= 0
|
|||
|
|
|
|||
|
|
t0 = time.perf_counter()
|
|||
|
|
out = list_batches_from_index(limit=100)
|
|||
|
|
elapsed_ms = (time.perf_counter() - t0) * 1000
|
|||
|
|
assert out["source"] == "index"
|
|||
|
|
assert "items" in out
|
|||
|
|
assert elapsed_ms < 500, f"index list too slow: {elapsed_ms:.0f}ms"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_list_labeling_batches_uses_index():
|
|||
|
|
if index_is_empty():
|
|||
|
|
rebuild_batch_index()
|
|||
|
|
t0 = time.perf_counter()
|
|||
|
|
out = list_labeling_batches(limit=50)
|
|||
|
|
elapsed_ms = (time.perf_counter() - t0) * 1000
|
|||
|
|
assert "items" in out
|
|||
|
|
assert elapsed_ms < 800, f"list_labeling_batches too slow: {elapsed_ms:.0f}ms"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_archive_batch_hides_from_list():
|
|||
|
|
from as_platform.db.engine import session_scope
|
|||
|
|
from as_platform.db.models import BatchIndex
|
|||
|
|
from as_platform.labeling.batch_index import archive_batch, list_batches_from_index
|
|||
|
|
|
|||
|
|
with session_scope() as db:
|
|||
|
|
rec = (
|
|||
|
|
db.query(BatchIndex)
|
|||
|
|
.filter(BatchIndex.archived.is_(False), BatchIndex.stage == "raw_pool")
|
|||
|
|
.first()
|
|||
|
|
)
|
|||
|
|
if not rec:
|
|||
|
|
return
|
|||
|
|
cid = rec.campaign_id
|
|||
|
|
|
|||
|
|
archive_batch(cid)
|
|||
|
|
out = list_batches_from_index(stage="raw_pool", limit=500)
|
|||
|
|
assert all(r.get("campaign_id") != cid for r in out["items"])
|