"""因子表达式注册表测试."""
import pytest
from sanguo_factor.registry import register_factor, get_factor, list_factors, _REGISTRY


def test_register_and_get_factor():
    """测试注册和获取因子."""
    _REGISTRY.clear()
    register_factor("ma5", "ts_mean(close, 5)", category="trend")
    f = get_factor("ma5")
    assert f["name"] == "ma5"
    assert f["category"] == "trend"
    assert f["expression"] == "ts_mean(close, 5)"


def test_list_factors_by_category():
    """测试按分类列出因子."""
    _REGISTRY.clear()
    register_factor("ma5", "ts_mean(close, 5)", category="trend")
    register_factor("rsi", "ts_rank(close, 14)", category="momentum")
    assert len(list_factors(category="trend")) == 1
    assert len(list_factors()) == 2


def test_register_duplicate_raises():
    """测试重复注册抛出异常."""
    _REGISTRY.clear()
    register_factor("ma5", "ts_mean(close, 5)", category="trend")
    with pytest.raises(ValueError, match="因子已存在"):
        register_factor("ma5", "ts_mean(close, 10)", category="trend")


def test_get_nonexistent_factor_returns_none():
    """测试获取不存在的因子返回 None."""
    _REGISTRY.clear()
    assert get_factor("nonexistent") is None


def test_list_factors_empty_registry():
    """测试空注册表返回空列表."""
    _REGISTRY.clear()
    assert list_factors() == []
    assert list_factors(category="trend") == []