| from pathlib import Path |
| from services.context_compressor import extract_request_spec, merge_request_spec, render_request_spec |
| from services.region_registry import resolve_region |
| from services.ocean_batch import parse_ocean_batch_request |
|
|
| ROOT=Path(__file__).resolve().parents[1] |
|
|
| def test_version(): |
| assert (ROOT/"VERSION").read_text(encoding="utf-8").strip()=="4.6.15" |
|
|
| def test_squid_center_uses_repository_membership(): |
| js=(ROOT/"static"/"js"/"app.js").read_text(encoding="utf-8") |
| api=(ROOT/"routes"/"datasets.py").read_text(encoding="utf-8") |
| assert "repository_domains" in api |
| assert "file_count_by_repository" in api |
| assert "domains.includes(domain)" in js |
| assert "squid_file_count" in js |
| assert "实时仓库文件树口径" in js |
|
|
| def test_single_ocean_export_forces_managed_proxy(): |
| js=(ROOT/"static"/"js"/"app.js").read_text(encoding="utf-8") |
| assert "data-ocean-managed-download" in js |
| assert "downloadWithProgress(urls[0],name" in js |
| assert "浏览器不直接访问学校/ngrok" in js |
|
|
| def test_followup_depth_layer_context(): |
| base=merge_request_spec(None,"查询2002年8月西北太平洋 SST，导出 CSV",revision=False) |
| updated=merge_request_spec(base,"其中200米层的",revision=True) |
| assert updated.get("depth") |
| rendered=render_request_spec(updated) |
| assert "200" in rendered |
| assert "SST" in rendered or "sst" in rendered.lower() |
| assert "2002" in rendered |
| assert "西北太平洋" in rendered |
|
|
| def test_layer_is_structured(): |
| p=extract_request_spec("还是这个区域，第5层，换成SSH") |
| assert p["layer"]=="5" |
| assert "zos" in p["variables"] |
|
|
| def test_region_is_deterministic(): |
| a=resolve_region("查询西太平洋 SST") |
| b=resolve_region("再查询一次西太平洋 SSH") |
| assert a==b |
| assert a["bbox"]=={"lon_min":120.0,"lon_max":180.0,"lat_min":-30.0,"lat_max":30.0} |
|
|
| def test_explicit_bbox_overrides_registry(): |
| x=resolve_region("西太平洋 130°E-140°E 20°N-30°N") |
| assert x["source"]=="explicit_coordinates" |
| assert x["bbox"]["lon_min"]==130.0 |
|
|
| def test_batch_uses_registry(): |
| p=parse_ocean_batch_request("查询2002年8月西北太平洋 SST 并导出 CSV") |
| assert p is not None |
| assert p["region_name"]=="西北太平洋" |
| assert p["bbox"]["lat_max"]==60.0 |
|
|
| def test_multitask_parser_accepts_task_colon_and_isolates_context(): |
| js=(ROOT/"static"/"js"/"app.js").read_text(encoding="utf-8") |
| assert "任务\\s*)?(\\d+)" in js |
| assert "[MULTITASK_CHILD]" in js |
| assert "[TARGET_CHILD_TASK]" in js |
| assert "history:[]" in js |
| assert "data-parallel-active" in js |
| assert "runParallelChild(task,i,board,originalPrompt)" in js |
|
|
