Python module for backtesting trading strategies.
Support event driven backtesting, ie on_open
, on_close
, etc. Also supports multiple assets.
Very basic statistics like book cash, mtm and total value. Currently, everything else needs to be deferred to a 3rd party module like empyrical
.
There are some basic tests but use at your own peril. It's not production level code.
The core module uses pandas and scipy.
pip install yatbe
Below is an example usage (the performance of the example strategy won't be good).
import pandas as pd
from yabte.backtest import Strategy, StrategyRunner, Order, Book
from yabte.utilities.plot.plotly.strategy_runner import plot_strategy_runner
from yabte.utilities.strategy_helpers import crossover
from yabte.tests._helpers import generate_nasdaq_dataset
class SMAXO(Strategy):
def init(self):
# enhance data with simple moving averages
p = self.params
days_short = p.get("days_short", 10)
days_long = p.get("days_long", 20)
close_sma_short = (
self.data.loc[:, (slice(None), "Close")]
.rolling(days_short)
.mean()
.rename({"Close": "CloseSMAShort"}, axis=1, level=1)
)
close_sma_long = (
self.data.loc[:, (slice(None), "Close")]
.rolling(days_long)
.mean()
.rename({"Close": "CloseSMALong"}, axis=1, level=1)
)
self.data = pd.concat(
[self.data, close_sma_short, close_sma_long], axis=1
).sort_index(axis=1)
def on_close(self):
# create some orders
for symbol in ["GOOG", "MSFT"]:
df = self.data[symbol]
ix_2d = df.index[-2:]
data = df.loc[ix_2d, ("CloseSMAShort", "CloseSMALong")].dropna()
if len(data) == 2:
if crossover(data.CloseSMAShort, data.CloseSMALong):
self.orders.append(Order(asset_name=symbol, size=-100))
elif crossover(data.CloseSMALong, data.CloseSMAShort):
self.orders.append(Order(asset_name=symbol, size=100))
# load some data
assets, df_combined = generate_nasdaq_dataset()
# create a book with 100000 cash
book = Book(name="Main", cash="100000")
# run our strategy
sr = StrategyRunner(
data=df_combined,
assets=assets,
strat_classes=[SMAXO],
books=[book],
)
sr.run()
# see the trades or book history
th = sr.transaction_history
bch = sr.book_history.loc[:, (slice(None), "cash")]
# plot the trades against book value
plot_strategy_runner(sr);
Jupyter notebook examples can be found under the notebooks folder.
Documentation can be found on Read the Docs.
Before commit run following format commands in project folder:
poetry run black .
poetry run isort . --profile black
poetry run docformatter . --recursive --in-place --black --exclude _unittest_numpy_extensions.py