mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-01 15:01:18 +00:00
36 lines
905 B
Python
36 lines
905 B
Python
import pytest
|
|
|
|
from openbb_terminal.stocks import databento_model
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def vcr_config():
|
|
return {
|
|
"filter_headers": [
|
|
("User-Agent", None),
|
|
("Authorization", "MOCK_AUTHORIZATION"),
|
|
],
|
|
}
|
|
|
|
|
|
def test_databento_model():
|
|
"""Tests the DataBento model."""
|
|
model = databento_model.DataBento(
|
|
symbol="AAPL", start="2021-08-10", end="2021-08-11"
|
|
)
|
|
assert model.symbol == "AAPL"
|
|
assert model.exchange == "XNAS.ITCH"
|
|
assert model.stype == "raw_symbol"
|
|
assert model.start == "2021-08-10"
|
|
assert model.end == "2021-08-11"
|
|
|
|
|
|
@pytest.mark.vcr
|
|
def test_stock_load(recorder):
|
|
"""Tests the stock load function."""
|
|
result = databento_model.get_historical_stock(
|
|
symbol="AAPL", start_date="2022-08-01", end_date="2022-10-01"
|
|
)
|
|
assert result.empty is False
|
|
recorder.capture(result)
|