mirror of
https://github.com/wassname/options_backtester.git
synced 2026-07-12 12:34:39 +08:00
Juan Pablo Amoroso
57 lines
1.3 KiB
Python
57 lines
1.3 KiB
Python
import os
|
|
|
|
import pytest
|
|
|
|
from backtester.datahandler import HistoricalOptionsData, TiingoData
|
|
from backtester.enums import Stock
|
|
|
|
TEST_DIR = os.path.abspath(os.path.dirname(__file__))
|
|
|
|
SAMPLE_DATA_STOCKS = os.path.join(TEST_DIR, 'test_data', 'test_data_stocks.csv')
|
|
IVY_PORTFOLIO_DATA = os.path.join(TEST_DIR, 'test_data', 'ivy_portfolio.csv')
|
|
SAMPLE_DATA_OPTIONS = os.path.join(TEST_DIR, 'test_data', 'test_data_options.csv')
|
|
|
|
# DataHandler fixtures
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def sample_stocks_datahandler():
|
|
data = TiingoData(SAMPLE_DATA_STOCKS)
|
|
return data
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def ivy_portfolio_datahandler():
|
|
data = TiingoData(IVY_PORTFOLIO_DATA)
|
|
return data
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def constant_price_stocks():
|
|
data = TiingoData(SAMPLE_DATA_STOCKS)
|
|
data['adjClose'] = data['close'] = 10.0
|
|
return data
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def sample_options_datahandler():
|
|
data = HistoricalOptionsData(SAMPLE_DATA_OPTIONS)
|
|
return data
|
|
|
|
|
|
# Stock Porfolio fixtures
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def ivy_portfolio():
|
|
return [Stock('VTI', 0.2), Stock('VEU', 0.2), Stock('BND', 0.2), Stock('VNQ', 0.2), Stock('DBC', 0.2)]
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def sample_stock_portfolio():
|
|
VOO = Stock('VOO', 0.4)
|
|
TUR = Stock('TUR', 0.1)
|
|
RSX = Stock('RSX', 0.5)
|
|
|
|
return [VOO, TUR, RSX]
|