mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-30 05:52:23 +00:00
83 lines
2.5 KiB
Python
83 lines
2.5 KiB
Python
"""Integration-test fixtures for openbb-econometrics."""
|
|
|
|
import threading
|
|
import time
|
|
|
|
import numpy as np
|
|
import pandas as pd
|
|
import pytest
|
|
from openbb_core.app.utils import df_to_basemodel
|
|
|
|
from openbb_econometrics.utils import mock_multi_index_data
|
|
|
|
_SEED = 20240517
|
|
_N_ROWS = 120
|
|
|
|
|
|
@pytest.fixture(scope="session", autouse=True)
|
|
def api_server(pytestconfig):
|
|
"""Run the OpenBB REST API on 0.0.0.0:8000 for the integration session."""
|
|
if pytestconfig.getoption("markexpr") == "not integration":
|
|
yield None
|
|
return
|
|
|
|
import uvicorn
|
|
from openbb_core.api.rest_api import app
|
|
|
|
config = uvicorn.Config(app, host="0.0.0.0", port=8000, log_level="warning") # noqa: S104
|
|
server = uvicorn.Server(config)
|
|
thread = threading.Thread(target=server.run, daemon=True)
|
|
thread.start()
|
|
|
|
deadline = time.time() + 60
|
|
while time.time() < deadline:
|
|
if server.started:
|
|
break
|
|
time.sleep(0.25)
|
|
else: # pragma: no cover
|
|
raise RuntimeError("The OpenBB REST API did not start within 60s.")
|
|
|
|
yield server
|
|
|
|
server.should_exit = True
|
|
thread.join(timeout=15)
|
|
|
|
|
|
def _timeseries_records() -> list[dict]:
|
|
"""Build a deterministic OHLCV time series as JSON-ready records."""
|
|
rng = np.random.default_rng(_SEED)
|
|
t = np.arange(_N_ROWS)
|
|
base = 100 + 0.5 * t + rng.normal(0, 2, _N_ROWS).cumsum() * 0.3
|
|
close = base + rng.normal(0, 1, _N_ROWS)
|
|
open_ = close + rng.normal(0, 0.5, _N_ROWS)
|
|
high = np.maximum(open_, close) + np.abs(rng.normal(1, 0.5, _N_ROWS))
|
|
low = np.minimum(open_, close) - np.abs(rng.normal(1, 0.5, _N_ROWS))
|
|
volume = 1_000_000 + rng.normal(0, 100_000, _N_ROWS) + 5_000 * t
|
|
dates = pd.date_range("2021-01-04", periods=_N_ROWS, freq="B")
|
|
return [
|
|
{
|
|
"date": day.strftime("%Y-%m-%d"),
|
|
"open": round(float(o), 4),
|
|
"high": round(float(h), 4),
|
|
"low": round(float(lo), 4),
|
|
"close": round(float(c), 4),
|
|
"volume": round(float(v), 2),
|
|
}
|
|
for day, o, h, lo, c, v in zip(dates, open_, high, low, close, volume)
|
|
]
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def timeseries_data() -> list[dict]:
|
|
"""Deterministic OHLCV records for the time-series command tests."""
|
|
return _timeseries_records()
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def panel_data() -> list[dict]:
|
|
"""Deterministic entity/time panel records as JSON-safe dicts."""
|
|
np.random.seed(_SEED)
|
|
return [
|
|
row.model_dump(mode="json") for row in df_to_basemodel(mock_multi_index_data())
|
|
]
|