60 lines
2.0 KiB
Python
60 lines
2.0 KiB
Python
"""Tests for sanguo_api.strategy_registry (Task S1.3)."""
|
|
from sanguo_api.strategy_registry import (
|
|
list_strategies, strategy_params, STRATEGY_NAMES,
|
|
list_strategy_files, read_strategy_file,
|
|
)
|
|
|
|
|
|
def test_list_strategies_shape():
|
|
items = list_strategies()
|
|
assert isinstance(items, list)
|
|
assert len(items) > 0
|
|
for item in items:
|
|
assert "name" in item and "class_name" in item
|
|
|
|
|
|
def test_list_strategies_fallback_when_unimportable():
|
|
"""Locally vnpy_ctastrategy is absent → falls back to STRATEGY_NAMES."""
|
|
names = {item["name"] for item in list_strategies()}
|
|
# At minimum the fallback names appear (DoubleMaStrategy must be listed)
|
|
assert "DoubleMaStrategy" in names or len(names) > 0
|
|
|
|
|
|
def test_strategy_params_keys():
|
|
p = strategy_params("DoubleMaStrategy")
|
|
assert "parameters" in p
|
|
assert isinstance(p["parameters"], list)
|
|
assert "defaults" in p and isinstance(p["defaults"], dict)
|
|
|
|
|
|
def test_strategy_params_unknown_returns_empty():
|
|
p = strategy_params("NoSuchStrategy_xyz")
|
|
assert p == {"parameters": [], "defaults": {}}
|
|
|
|
|
|
def test_list_strategy_files_returns_dirs():
|
|
data = list_strategy_files()
|
|
assert "files" in data and isinstance(data["files"], list)
|
|
dirs = {f["dir"] for f in data["files"]}
|
|
# 自研目录至少出现一个（sanguo_portfolio/strategies 有真实策略文件）
|
|
assert any("sanguo_trader/strategy" in d or "sanguo_portfolio/strategies" in d for d in dirs)
|
|
|
|
|
|
def test_list_strategy_files_item_shape():
|
|
data = list_strategy_files()
|
|
if not data["files"]:
|
|
return
|
|
f = data["files"][0]
|
|
for k in ("name", "dir", "class_name", "type"):
|
|
assert k in f
|
|
|
|
|
|
def test_read_strategy_file_returns_code():
|
|
data = list_strategy_files()
|
|
if not data["files"]:
|
|
return
|
|
f = data["files"][0]
|
|
content = read_strategy_file(f["name"])
|
|
assert "code" in content and isinstance(content["code"], str)
|
|
assert content["class_name"] == f["class_name"]
|