扩展指标

2026-07-09 05:08:16 +08:00
commit 308c46ab9a
537 changed files with 152299 additions and 0 deletions
@@ -0,0 +1,124 @@
+"""
+Derivatives benchmark hooks.
+
+These are intentionally optional and skip when `py_vollib` is unavailable.
+Run with:
+
+    uv run pytest benchmarks/test_derivatives_speed.py --benchmark-only -v
+"""
+
+from __future__ import annotations
+
+import importlib.util
+import sys
+from pathlib import Path
+
+import numpy as np
+import pytest
+
+# Ensure direct benchmark test runs can import local package from `python/`.
+ROOT = Path(__file__).resolve().parents[1]
+PYTHON_SRC = ROOT / "python"
+if str(PYTHON_SRC) not in sys.path:
+    sys.path.insert(0, str(PYTHON_SRC))
+
+HAS_FERRO_EXTENSION = True
+try:
+    from ferro_ta.analysis.options import implied_volatility, option_price
+except ModuleNotFoundError:
+    HAS_FERRO_EXTENSION = False
+
+pytestmark = pytest.mark.skipif(
+    not HAS_FERRO_EXTENSION, reason="ferro_ta extension is not built"
+)
+
+
+def _sample_chain(n: int = 1000) -> tuple[np.ndarray, ...]:
+    spot = np.linspace(90.0, 110.0, n)
+    strike = np.full(n, 100.0)
+    rate = np.full(n, 0.02)
+    time_to_expiry = np.full(n, 0.5)
+    volatility = np.full(n, 0.2)
+    return spot, strike, rate, time_to_expiry, volatility
+
+
+def test_ferro_ta_option_price_speed(benchmark):
+    spot, strike, rate, time_to_expiry, volatility = _sample_chain()
+
+    benchmark.pedantic(
+        lambda: option_price(
+            spot,
+            strike,
+            rate,
+            time_to_expiry,
+            volatility,
+            option_type="call",
+            model="bsm",
+        ),
+        iterations=5,
+        rounds=20,
+        warmup_rounds=2,
+    )
+
+
+def test_ferro_ta_implied_vol_speed(benchmark):
+    spot, strike, rate, time_to_expiry, volatility = _sample_chain()
+    prices = option_price(
+        spot,
+        strike,
+        rate,
+        time_to_expiry,
+        volatility,
+        option_type="call",
+        model="bsm",
+    )
+
+    benchmark.pedantic(
+        lambda: implied_volatility(
+            prices,
+            spot,
+            strike,
+            rate,
+            time_to_expiry,
+            option_type="call",
+            model="bsm",
+        ),
+        iterations=5,
+        rounds=20,
+        warmup_rounds=2,
+    )
+
+
+@pytest.mark.skipif(
+    importlib.util.find_spec("py_vollib") is None,
+    reason="py_vollib is optional",
+)
+def test_py_vollib_scalar_loop_baseline(benchmark):
+    from py_vollib.black_scholes_merton import black_scholes_merton as py_vollib_bsm
+    from py_vollib.black_scholes_merton.implied_volatility import (
+        implied_volatility as py_vollib_iv,
+    )
+
+    spot, strike, rate, time_to_expiry, volatility = _sample_chain(250)
+    prices = [
+        py_vollib_bsm("c", float(s), float(k), float(t), float(r), float(vol), 0.0)
+        for s, k, r, t, vol in zip(spot, strike, rate, time_to_expiry, volatility)
+    ]
+
+    benchmark.pedantic(
+        lambda: [
+            py_vollib_iv(
+                float(price),
+                "c",
+                float(s),
+                float(k),
+                float(t),
+                float(r),
+                0.0,
+            )
+            for price, s, k, r, t in zip(prices, spot, strike, rate, time_to_expiry)
+        ],
+        iterations=3,
+        rounds=10,
+        warmup_rounds=1,
+    )