Yet another backtesting engine
Project description
yabte - Yet Another BackTesting Engine
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.
Core dependencies
The core module uses pandas and scipy.
Installation
pip install yatbe
Usage
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.matplotlib.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);
Examples
Jupyter notebook examples can be found under the notebooks folder.
Documentation
Documentation can be found on Read the Docs.
Development
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
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distributions
Hashes for yabte-0.3.6-cp310-cp310-manylinux_2_35_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | dc67f1d867c443cf12c3819877ec633299f52cec0d8d60d0981d8eedc4f1f417 |
|
MD5 | f2b7f3995b9f7bbbcb3058588c304f4c |
|
BLAKE2b-256 | bf4dc732d9b6881c7189a9a0b172fff52c477d72ffd8dfc8a546a52580813511 |