quantcli 0.1.10__tar.gz → 0.1.11__tar.gz

{quantcli-0.1.10/quantcli.egg-info → quantcli-0.1.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quantcli
-Version: 0.1.10
+Version: 0.1.11
 Summary: 面向AI的多因子量化选股策略挖掘工具，AI Agent 友好 CLI
 Author-email: QuantCLI Team <quantcli@example.com>
 Project-URL: repository, https://github.com/wumu2013/quantcli

{quantcli-0.1.10 → quantcli-0.1.11}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "quantcli"
-version = "0.1.10"
+version = "0.1.11"
 description = "面向AI的多因子量化选股策略挖掘工具，AI Agent 友好 CLI"
 readme = "README.md"
 requires-python = ">=3.9"

{quantcli-0.1.10 → quantcli-0.1.11}/quantcli/cli.py

@@ -277,68 +277,6 @@ def data_sync_fundamentals(ctx, years):
         sys.exit(1)
 
 
-# =============================================================================
-# Factors 命令 - 列出内置因子
-# =============================================================================
-
-
-@quantcli.group()
-def factors():
-    """内置因子管理"""
-    pass
-
-
-@factors.command("list")
-@click.option("--json", is_flag=True, help="Output as JSON")
-@click.pass_context
-def factors_list(ctx, json):
-    """列出所有内置 Alpha101 因子"""
-    from .utils import builtin_factors_dir
-
-    builtin_dir = builtin_factors_dir() / "alpha101"
-
-    if not builtin_dir.exists():
-        click.echo("No built-in factors found")
-        return
-
-    # 加载所有因子
-    factors = []
-    for f in sorted(builtin_dir.glob("alpha_*.yaml")):
-        import yaml
-
-        with open(f, "r", encoding="utf-8") as fp:
-            data = yaml.safe_load(fp)
-            factors.append(
-                {
-                    "name": data.get("name", f.stem),
-                    "type": data.get("type", "technical"),
-                    "direction": data.get("direction", "neutral"),
-                    "description": data.get("description", ""),
-                    "file": f"alpha101/{f.name}",
-                }
-            )
-
-    if json:
-        import json
-
-        click.echo(
-            json.dumps(
-                {"status": "success", "count": len(factors), "factors": factors},
-                ensure_ascii=False,
-                indent=2,
-            )
-        )
-        return
-
-    click.echo(f"Built-in Alpha101 Factors ({len(factors)} total):\n")
-    click.echo(f"{'File':<22} {'Type':<12} {'Direction':<10} Description")
-    click.echo("-" * 60)
-    for f in factors:
-        click.echo(
-            f"{f['file']:<22} {f['type']:<12} {f['direction']:<10} {f['description']}"
-        )
-
-
 # =============================================================================
 # Factor 命令
 # =============================================================================
@@ -469,21 +407,53 @@ def factor_eval(ctx, name, symbol, start, end, method):
 
 
 @factor.command("list")
+@click.option("--json", is_flag=True, help="Output as JSON")
 @click.pass_context
-def factor_list(ctx):
-    """列出已注册的因子"""
-    engine = FactorEngine()
-    factors = engine.registry.list_all()
+def factor_list(ctx, json):
+    """列出所有内置 Alpha101 因子"""
+    from .utils import builtin_factors_dir
 
-    if not factors:
-        click.echo("No factors registered")
+    builtin_dir = builtin_factors_dir() / "alpha101"
+
+    if not builtin_dir.exists():
+        click.echo("No built-in factors found")
         return
 
-    click.echo("Registered factors:")
-    for name in factors:
-        factor = engine.registry.get(name)
-        if factor:
-            click.echo(f"  - {name}: {factor.formula[:50]}...")
+    factors = []
+    for f in sorted(builtin_dir.glob("alpha_*.yaml")):
+        import yaml
+
+        with open(f, "r", encoding="utf-8") as fp:
+            data = yaml.safe_load(fp)
+            factors.append(
+                {
+                    "name": data.get("name", f.stem),
+                    "type": data.get("type", "technical"),
+                    "direction": data.get("direction", "neutral"),
+                    "description": data.get("description", ""),
+                    "file": f"alpha101/{f.name}",
+                }
+            )
+
+    if json:
+        import json
+
+        click.echo(
+            json.dumps(
+                {"status": "success", "count": len(factors), "factors": factors},
+                ensure_ascii=False,
+                indent=2,
+            )
+        )
+        return
+
+    click.echo(f"Built-in Alpha101 Factors ({len(factors)} total):\n")
+    click.echo(f"{'File':<22} {'Type':<12} {'Direction':<10} Description")
+    click.echo("-" * 60)
+    for f in factors:
+        click.echo(
+            f"{f['file']:<22} {f['type']:<12} {f['direction']:<10} {f['description']}"
+        )
 
 
 @factor.command("run-file")

{quantcli-0.1.10 → quantcli-0.1.11/quantcli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quantcli
-Version: 0.1.10
+Version: 0.1.11
 Summary: 面向AI的多因子量化选股策略挖掘工具，AI Agent 友好 CLI
 Author-email: QuantCLI Team <quantcli@example.com>
 Project-URL: repository, https://github.com/wumu2013/quantcli

{quantcli-0.1.10 → quantcli-0.1.11}/tests/test_builtin_factors.py

@@ -18,7 +18,11 @@ import numpy as np
 from click.testing import CliRunner
 
 from quantcli import cli
-from quantcli.factors.loader import load_strategy, load_all_factors, load_factor_from_ref
+from quantcli.factors.loader import (
+    load_strategy,
+    load_all_factors,
+    load_factor_from_ref,
+)
 from quantcli.factors.pipeline import FactorPipeline
 
 
@@ -38,15 +42,17 @@ def sample_price_data():
     open_ = close * (1 + np.random.randn(100) * 0.01)
     volume = np.random.randint(1000000, 10000000, 100)
 
-    return pd.DataFrame({
-        "date": dates,
-        "symbol": ["600519"] * 100,
-        "open": open_,
-        "high": close * 1.02,
-        "low": close * 0.98,
-        "close": close,
-        "volume": volume,
-    })
+    return pd.DataFrame(
+        {
+            "date": dates,
+            "symbol": ["600519"] * 100,
+            "open": open_,
+            "high": close * 1.02,
+            "low": close * 0.98,
+            "close": close,
+            "volume": volume,
+        }
+    )
 
 
 @pytest.fixture
@@ -60,25 +66,27 @@ def multi_symbol_price_data():
 
     for symbol in symbols:
         close = 100 + np.cumsum(np.random.randn(60) * 0.5)
-        data[symbol] = pd.DataFrame({
-            "date": dates,
-            "symbol": [symbol] * 60,
-            "open": close * (1 + np.random.randn(60) * 0.01),
-            "high": close * 1.02,
-            "low": close * 0.98,
-            "close": close,
-            "volume": np.random.randint(1000000, 10000000, 60),
-        })
+        data[symbol] = pd.DataFrame(
+            {
+                "date": dates,
+                "symbol": [symbol] * 60,
+                "open": close * (1 + np.random.randn(60) * 0.01),
+                "high": close * 1.02,
+                "low": close * 0.98,
+                "close": close,
+                "volume": np.random.randint(1000000, 10000000, 60),
+            }
+        )
 
     return data
 
 
-class TestFactorsListCommand:
-    """quantcli factors list 命令测试"""
+class TestFactorBuiltinListCommand:
+    """quantcli factor list 命令测试"""
 
-    def test_factors_list_basic(self, runner):
-        """测试 factors list 基本功能"""
-        result = runner.invoke(cli.factors_list, ["--json"])
+    def test_factor_list_builtin_basic(self, runner):
+        """测试 factor list 基本功能"""
+        result = runner.invoke(cli.factor_list, ["--json"])
 
         assert result.exit_code == 0
         output = json.loads(result.output)
@@ -87,20 +95,19 @@ class TestFactorsListCommand:
         assert output["count"] == 40
         assert len(output["factors"]) == 40
 
-    def test_factors_list_shows_alpha101(self, runner):
-        """测试 factors list 显示 Alpha101 因子"""
-        result = runner.invoke(cli.factors_list, ["--json"])
+    def test_factor_list_builtin_shows_alpha101(self, runner):
+        """测试 factor list 显示 Alpha101 因子"""
+        result = runner.invoke(cli.factor_list, ["--json"])
 
         assert result.exit_code == 0
         output = json.loads(result.output)
 
-        # 检查包含 alpha_001
         factor_files = [f["file"] for f in output["factors"]]
         assert "alpha101/alpha_001.yaml" in factor_files
 
-    def test_factors_list_human_format(self, runner):
-        """测试 factors list 人类友好格式"""
-        result = runner.invoke(cli.factors_list, [])
+    def test_factor_list_builtin_human_format(self, runner):
+        """测试 factor list 人类友好格式"""
+        result = runner.invoke(cli.factor_list, [])
 
         assert result.exit_code == 0
         assert "Built-in Alpha101 Factors" in result.output
@@ -113,7 +120,9 @@ class TestBuiltinFactors:
 
     def test_load_single_builtin_factor(self):
         """测试加载单个内置因子"""
-        factor = load_factor_from_ref("/Users/apple/quantcli/examples/strategies", "alpha101/alpha_001")
+        factor = load_factor_from_ref(
+            "/Users/apple/quantcli/examples/strategies", "alpha101/alpha_001"
+        )
 
         assert factor is not None
         assert "alpha_001" in factor.name.lower() or "反转" in factor.description
@@ -135,7 +144,9 @@ class TestBuiltinFactors:
 
     def test_builtin_factor_fields(self):
         """测试内置因子字段完整"""
-        factor = load_factor_from_ref("/Users/apple/quantcli/examples/strategies", "alpha101/alpha_008")
+        factor = load_factor_from_ref(
+            "/Users/apple/quantcli/examples/strategies", "alpha101/alpha_008"
+        )
 
         assert factor is not None
         assert factor.name is not None
@@ -156,7 +167,9 @@ class TestBuiltinFactors:
         # 测试每个因子都能加载
         for yaml_file in yaml_files:
             alpha_name = yaml_file.stem  # alpha_001
-            factor = load_factor_from_ref("/Users/apple/quantcli/examples/strategies", f"alpha101/{alpha_name}")
+            factor = load_factor_from_ref(
+                "/Users/apple/quantcli/examples/strategies", f"alpha101/{alpha_name}"
+            )
             assert factor is not None, f"Failed to load {yaml_file.name}"
 
 
@@ -248,8 +261,8 @@ class TestAnalyzeCommands:
         window = 20
         ic_rolling = []
         for i in range(window, len(factor)):
-            f = factor.iloc[i-window:i]
-            r = forward_returns.iloc[i-window:i]
+            f = factor.iloc[i - window : i]
+            r = forward_returns.iloc[i - window : i]
             valid_mask = ~(f.isna() | r.isna())
             if valid_mask.sum() > 5:
                 ic, _ = spearmanr(f[valid_mask], r[valid_mask])
@@ -268,7 +281,7 @@ class TestFilterRunCommand:
 
     def test_filter_config_loading(self, tmp_path):
         """测试 filter 配置加载"""
-        with tempfile.NamedTemporaryFile(mode='w', suffix='.yaml', delete=False) as f:
+        with tempfile.NamedTemporaryFile(mode="w", suffix=".yaml", delete=False) as f:
             f.write("""
 name: 测试策略
 version: 1.0.0
@@ -293,6 +306,7 @@ output:
             assert config.name == "测试策略"
         finally:
             import os
+
             os.unlink(config_path)
 
     def test_filter_with_weights(self, tmp_path, multi_symbol_price_data):
@@ -300,7 +314,7 @@ output:
         from quantcli.factors.compute import FactorComputer
         from quantcli.factors.ranking import ScoringEngine
 
-        with tempfile.NamedTemporaryFile(mode='w', suffix='.yaml', delete=False) as f:
+        with tempfile.NamedTemporaryFile(mode="w", suffix=".yaml", delete=False) as f:
             f.write("""
 name: 测试策略
 version: 1.0.0
@@ -332,7 +346,7 @@ output:
                 factors,
                 multi_symbol_price_data,
                 {},
-                list(multi_symbol_price_data.keys())
+                list(multi_symbol_price_data.keys()),
             )
 
             assert "close_price" in factor_df.columns
@@ -345,6 +359,7 @@ output:
             assert len(result) == len(multi_symbol_price_data)
         finally:
             import os
+
             os.unlink(config_path)
 
     def test_filter_with_conditions(self, tmp_path, multi_symbol_price_data):
@@ -352,7 +367,7 @@ output:
         from quantcli.factors.compute import FactorComputer
         from quantcli.factors.ranking import ScoringEngine
 
-        with tempfile.NamedTemporaryFile(mode='w', suffix='.yaml', delete=False) as f:
+        with tempfile.NamedTemporaryFile(mode="w", suffix=".yaml", delete=False) as f:
             f.write("""
 name: 测试条件筛选
 version: 1.0.0
@@ -387,7 +402,7 @@ output:
                 factors,
                 multi_symbol_price_data,
                 {},
-                list(multi_symbol_price_data.keys())
+                list(multi_symbol_price_data.keys()),
             )
 
             scorer = ScoringEngine(normalize="zscore")
@@ -397,6 +412,7 @@ output:
             assert "score" in result.columns
         finally:
             import os
+
             os.unlink(config_path)
 
     def test_filter_json_output_format(self, tmp_path, multi_symbol_price_data):
@@ -406,7 +422,7 @@ output:
         from quantcli.factors.compute import FactorComputer
         from quantcli.factors.ranking import ScoringEngine
 
-        with tempfile.NamedTemporaryFile(mode='w', suffix='.yaml', delete=False) as f:
+        with tempfile.NamedTemporaryFile(mode="w", suffix=".yaml", delete=False) as f:
             f.write("""
 name: 测试策略
 version: 1.0.0
@@ -435,7 +451,7 @@ output:
                 factors,
                 multi_symbol_price_data,
                 {},
-                list(multi_symbol_price_data.keys())
+                list(multi_symbol_price_data.keys()),
             )
 
             scorer = ScoringEngine(normalize="zscore")
@@ -445,7 +461,7 @@ output:
             output = {
                 "status": "success",
                 "count": len(result),
-                "results": result.to_dict(orient="records")
+                "results": result.to_dict(orient="records"),
             }
 
             assert output["status"] == "success"
@@ -454,6 +470,7 @@ output:
             assert len(parsed) > 0
         finally:
             import os
+
             os.unlink(config_path)
 
 
@@ -712,17 +729,22 @@ class TestFormulaSyntax:
         computer = FactorComputer()
 
         factors = {
-            "ma20": FactorDefinition(name="ma20", type="technical", expr="ma(close, 20)"),
-            "ema12": FactorDefinition(name="ema12", type="technical", expr="ema(close, 12)"),
-            "delay5": FactorDefinition(name="delay5", type="technical", expr="delay(close, 5)"),
-            "std20": FactorDefinition(name="std20", type="technical", expr="rolling_std(close, 20)"),
+            "ma20": FactorDefinition(
+                name="ma20", type="technical", expr="ma(close, 20)"
+            ),
+            "ema12": FactorDefinition(
+                name="ema12", type="technical", expr="ema(close, 12)"
+            ),
+            "delay5": FactorDefinition(
+                name="delay5", type="technical", expr="delay(close, 5)"
+            ),
+            "std20": FactorDefinition(
+                name="std20", type="technical", expr="rolling_std(close, 20)"
+            ),
         }
 
         result = computer.compute_all_factors(
-            factors,
-            {"600519": sample_price_data},
-            {},
-            ["600519"]
+            factors, {"600519": sample_price_data}, {}, ["600519"]
         )
 
         assert "ma20" in result.columns
@@ -739,14 +761,13 @@ class TestFormulaSyntax:
 
         factors = {
             "rank": FactorDefinition(name="rank", type="technical", expr="rank(close)"),
-            "zscore": FactorDefinition(name="zscore", type="technical", expr="zscore(close)"),
+            "zscore": FactorDefinition(
+                name="zscore", type="technical", expr="zscore(close)"
+            ),
         }
 
         result = computer.compute_all_factors(
-            factors,
-            {"600519": sample_price_data},
-            {},
-            ["600519"]
+            factors, {"600519": sample_price_data}, {}, ["600519"]
         )
 
         assert "rank" in result.columns
@@ -761,14 +782,13 @@ class TestFormulaSyntax:
 
         # correlation 计算 - 价格与成交量的相关性
         factors = {
-            "corr": FactorDefinition(name="corr", type="technical", expr="correlation(close, volume, 10)"),
+            "corr": FactorDefinition(
+                name="corr", type="technical", expr="correlation(close, volume, 10)"
+            ),
         }
 
         result = computer.compute_all_factors(
-            factors,
-            {"600519": sample_price_data},
-            {},
-            ["600519"]
+            factors, {"600519": sample_price_data}, {}, ["600519"]
         )
 
         assert "corr" in result.columns
@@ -781,15 +801,16 @@ class TestFormulaSyntax:
         computer = FactorComputer()
 
         factors = {
-            "is_up": FactorDefinition(name="is_up", type="technical", expr="where(close > open, 1, 0)"),
-            "is_yinliang": FactorDefinition(name="is_yinliang", type="technical", expr="where(close < open, 1, 0)"),
+            "is_up": FactorDefinition(
+                name="is_up", type="technical", expr="where(close > open, 1, 0)"
+            ),
+            "is_yinliang": FactorDefinition(
+                name="is_yinliang", type="technical", expr="where(close < open, 1, 0)"
+            ),
         }
 
         result = computer.compute_all_factors(
-            factors,
-            {"600519": sample_price_data},
-            {},
-            ["600519"]
+            factors, {"600519": sample_price_data}, {}, ["600519"]
         )
 
         assert "is_up" in result.columns