Daywalker is a stock market backtesting framework. I wrote it mainly for my own personal use because I was unsatisfied with existing frameworks.
There's a lot of frameworks like these, so let me outline my design goals/methodology explicitly.
- I want to do as much of my analysis as possible in Python/
pandas
/numpy
. - I only trade daily (no intraday), almost 100% limit-on-open and limit-on-close. Many backtesting frameworks don't support the opening auction.
- My strategies run on a wide range of stocks and use alternative data sources. Providing a censored view of many symbols is important to me. The view should be in
pandas
. - Transaction costs, dividends, and taxes need to be tracked properly.
- I want to easily add tracking information and useful statistics to my trades and get this out at the end.
Here are a couple of novelties of this framework, which make up a significant chunk of the reason I built it.
Taxes matter a lot. There are a number of trading strategies that sacrifice a little alpha in order to minimize taxes, and justifiably so. Tax loss harvesting, as practiced by assorted robo-advisors (e.g. Wealthfront) is a good example.
In order to measure taxes, the source of income must be tracked. Dividends are treated differently from long and short term capital gains, for example.
For this reason, Daywalker tracks the cost-basis of every purchase you make.
For example, suppose you buy 100@25 of X at 100 on Jan 1, 1990 and another 100@30 shares on June 1, 1990. On Jan 10, 1991, you sell 150@35. The result, in terms of capital gains, is:
- A long term capital gain of
100 * (35 - 25) = 1000
. - A short term capital gain of
50 x (35 - 30) = 250
.
The $1000 cap gain is taxed at a low rate (in the US), while the $250 cap gain is taxed at a much higher rate. It might be a worthwhile improvement to the strategy to sell only the 100@35 to minimize capital gains.
The tracking information is an important piece here that I want to explain further. Suppose I am running a long/short strategy of the following nature - when an event occurs, I will open both a long and a short.
The way this works in daywalker is as follows. Inside my Strategy
object, I will enter trades in the following manner:
def pre_open(self, dt, broker, trades, other_data):
story = get_wallstreetbets_recommendation(dt)
story_score = get_story_score(story)
if story_score > self.threshold:
long_symbol, short_symbol = get_symbols_from_story(story)
broker.limit_on_open(long_symbol, price=..., size=..., is_buy=True, meta={ 'trade_story_id': story.id })
broker.limit_on_open(short_symbol, price=..., size=..., is_buy=False, meta={ 'trade_story_id': story.id })
The key observation here is that what's important to me is the P&L of the combined long/short position. One metric I will want to track is the Sharpe ratio of my strategy, computed as the average over individual long/short trades (rather than over time).
After we run the simulation we can use the metadata to track our trades. If we call the function m.broker.capital_gains()
(I'll discuss later on how to get here), we will get a dataframe that looks like this:
open_price close_price size open_trade_story_id open_date close_trade_id close_date symbol
0 17.50 17.51 1 1 2004-08-12 09:30:00-04:00 0 2004-08-13 16:00:00-04:00 "long_symbol"
1 17.50 17.50 2 1 2004-08-13 09:30:00-04:00 1 2004-08-16 16:00:00-04:00 "short_symbol"
2 17.54 17.34 3 2 2004-08-16 09:30:00-04:00 0 2004-08-17 16:00:00-04:00 "another_long_symbol_from_a_diff_event"
...
We can now take our cap_gains
dataframe and do cap_gains.groupby('open_trade_story_id')
to do an analysis on the level of an individual story (where each story triggered multiple trades). This makes it easy to analyze our capital gains from the combined long/short position.
To begin with, we will need some market data. Daywalker does not include market data.
>>> import pandas as pd
>>> from daywalker import TradeableAsset, Strategy, Market
>>> from daywalker.broker import InteractiveBrokers
>>> prices = pd.DataFrame({'date': [pd.Timestamp('2004-08-12 00:00:00-0400', tz='America/New_York'),
... pd.Timestamp('2004-08-13 00:00:00-0400', tz='America/New_York'),
... pd.Timestamp('2004-08-16 00:00:00-0400', tz='America/New_York'),
... pd.Timestamp('2004-08-17 00:00:00-0400', tz='America/New_York'),
... pd.Timestamp('2004-08-18 00:00:00-0400', tz='America/New_York')],
... 'open': [17.5, 17.5, 17.54, 17.35, 17.25],
... 'high': [17.58, 17.51, 17.54, 17.4, 17.29],
... 'low': [17.5, 17.5, 17.5, 17.15, 17.0],
... 'close': [17.5, 17.51, 17.5, 17.34, 17.11],
... 'volume': [2545100, 593000, 684700, 295900, 121300],
... 'divCash': [0.0, 0.0, 0.0, 0.0, 0.0],
... 'splitFactor': [1.0, 1.0, 1.0, 1.0, 1.0]})
Once you have market data, it simply needs to be brought to the above format. Then it should be wrapped in a TradeableAsset
and given a symbol:
>>> ta = TradeableAsset('acc', prices)
A TradeableAsset
represents a symbol that can be bought or sold. It also provides a time-censored view of itself. For example, on 2004-08-17, after the open, we can see the open price for 2004-08-17:
>>> history, open_price = ta.get_censored(pd.Timestamp('2004-08-17'), after_open=True)
>>> open_price
17.35
We can also see the history up to (and including) 2004-08-16.
# Note that we are doing this weird printing of a CSV because
# for some reason, doctests is failing even though exact strings seem to match. WTF.
>>> print(history[['open', 'high', 'low', 'close', 'volume', 'divCash', 'splitFactor']].to_csv(None))
date,open,high,low,close,volume,divCash,splitFactor
2004-08-12,17.5,17.58,17.5,17.5,2545100,0.0,1.0
2004-08-13,17.5,17.51,17.5,17.51,593000,0.0,1.0
2004-08-16,17.54,17.54,17.5,17.5,684700,0.0,1.0
<BLANKLINE>
The next step in running a backtest is getting a Broker
. I use Interactive Brokers.
>>> from daywalker.broker import InteractiveBrokers
>>> broker = InteractiveBrokers(10000, assets={'acc': ta})
This broker is initialized with $10,000 in our account and acc
as the only tradeable asset.
>>> from daywalker.market import _TestStrategy # _TestStrategy is a test-only strategy
>>> m = Market(prices['date'].min(), prices['date'].max(), strategy=_TestStrategy('acc'), broker=broker)
>>> m.run()
At the end of the backtest, we hold these positions:
>>> m.broker.positions()[['price', 'size', 'symbol', 'trade_id', 'date']]
price size symbol trade_id date
0 17.35 4 acc 0 2004-08-17 09:30:00-04:00
1 17.25 5 acc 1 2004-08-18 09:30:00-04:00
You can also get the positions marked to market at any given time:
>>> m.broker.positions_marked_to_market(pd.Timestamp('2004-08-17'), False)
commission_per_share date price size symbol trade_id current_value market_value mark_to_market_time
0 0.1735 2004-08-17 09:30:00-04:00 17.35 4 acc 0 17.5 70.0 2004-08-17 09:30:00-04:00
1 0.1725 2004-08-18 09:30:00-04:00 17.25 5 acc 1 17.5 87.5 2004-08-17 09:30:00-04:00
The following capital gains were achieved:
>>> cg = m.broker.capital_gains()
>>> cg[['open_price', 'close_price', 'size', 'open_trade_id', 'open_date', 'close_trade_id', 'close_date', 'symbol']]
open_price close_price size open_trade_id open_date close_trade_id close_date symbol
0 17.50 17.51 1 1 2004-08-12 09:30:00-04:00 0 2004-08-13 16:00:00-04:00 acc
1 17.50 17.50 2 0 2004-08-13 09:30:00-04:00 1 2004-08-16 16:00:00-04:00 acc
2 17.54 17.34 3 1 2004-08-16 09:30:00-04:00 0 2004-08-17 16:00:00-04:00 acc
And the following commissions were paid, as per the InteractiveBrokers schedule:
>>> trades = m.broker.trades()
>>> trades[['price', 'size', 'symbol', 'date', 'trade_id', 'commission']]
price size symbol date trade_id commission
0 17.50 1 acc 2004-08-12 09:30:00-04:00 1 0.1750
1 17.50 2 acc 2004-08-13 09:30:00-04:00 0 0.3500
2 17.51 -1 acc 2004-08-13 16:00:00-04:00 0 0.1751
3 17.54 3 acc 2004-08-16 09:30:00-04:00 1 0.5262
4 17.50 -2 acc 2004-08-16 16:00:00-04:00 1 0.3500
5 17.35 4 acc 2004-08-17 09:30:00-04:00 0 0.6940
6 17.34 -3 acc 2004-08-17 16:00:00-04:00 0 0.5202
7 17.25 5 acc 2004-08-18 09:30:00-04:00 1 0.8625
After the strategy has run, we can observe how it performed over time:
>>> m.broker.strategy_values()
cash date long_equities short_equities
0 9982.3250 2004-08-13 17.50 0.0
1 9964.3099 2004-08-16 35.02 0.0
2 9945.8137 2004-08-17 52.50 0.0
3 9927.2195 2004-08-18 69.36 0.0
4 9840.1070 2004-08-19 156.06 0.0
A strategy is a very simple class. Lets build a strategy which attempts to buy a stock, at the open, whenever the price is less than 5% of the previous maximum. It will do this at a rate of 100 shares/day until it owns 1000 shares.
It will attempt to sell 100 whenever the price is greater than 5% of the previous max.
class MyStrategy(Strategy):
def __init__(self, symbol):
self.symbol = symbol
def pre_open(self, dt, broker, trades, commissions):
positions = broker.positions()
total_shares = positions['size'].sum()
price = broker.historical_prices(self.symbol)
prev_max = price['close'].max()
if total_shares < 1000:
broker.limit_on_open(self.symbol, prev_max * 0.95, min(100, 1000 - total_shares), is_buy=True, meta={'prev_max': prev_max})
if total_shares > 0:
broker.limit_on_open(self.symbol, prev_max * 1.05, min(100, total_shares), is_buy=False, meta={'prev_max': prev_max})
def pre_close(self, dt, broker, trades, commissions):
positions = broker.positions()
total_shares = positions['size'].sum()
price = broker.historical_prices(self.symbol)
prev_max = price['close'].max()
if total_shares < 1000:
broker.limit_on_close(self.symbol, prev_max * 0.95, min(100, 1000 - total_shares), is_buy=True, meta={'prev_max': prev_max})
if total_shares > 0:
broker.limit_on_close(self.symbol, prev_max * 1.05, min(100, total_shares), is_buy=False, meta={'prev_max': prev_max})
At the moment, only the opening/closing auction are supported. The reason is that these are what I use.
Another interesting piece is the meta={...}
argument. This argument allows you to attach diagnostic information to each trade order. This diagnostic information carries over to capital gains as columns in the resulting dataframe.
An important aspect of trading (particularly inter-day strategies) is using data going beyond tick data. Here's an example:
>>> wsb = pd.DataFrame({'date': [pd.Timestamp('2004-08-12 00:00:00-0400', tz='America/New_York'),
... pd.Timestamp('2004-08-13 00:00:00-0400', tz='America/New_York'),
... pd.Timestamp('2004-08-16 00:00:00-0400', tz='America/New_York'),
... pd.Timestamp('2004-08-17 00:00:00-0400', tz='America/New_York'),
... pd.Timestamp('2004-08-18 00:00:00-0400', tz='America/New_York')],
... 'rating': [0, 1, 0, 1, 0]})
This data represents an /r/wallstreetbets buy/sell rating - 1 is good, 0 is bad. And /r/wallstreetbets is never wrong, right?
Daywalker allows the use of such data in strategies. As before, we'll create a market. We'll then add this data:
>>> broker = InteractiveBrokers(initial_cash=10000) # Not real tesla prices
>>> broker.add_asset('tsla', prices)
>>> m = Market(prices['date'].min() + pd.offsets.BDay(), prices['date'].max(), strategy=None, broker=broker)
>>> m.add_data('wallstreetbets', wsb, censor_on_index=False, censor_column='date')
By default add_data
will censor the data on it's index (which must be a datetime index). Alternately, we can choose to censor it based on a date time column.
What we mean by censorship is the following - the censorship column represents the date at which the data became available. For example, a person posting "TSLA is awesome!" on 2020/2/6 is not usable information until 2020/2/6. In contrast, a person posting "Buy TSLA" on 2019/12/1 is usable information anytime after 2019/12/1.
Thus, when building one's alternative data strategy, it is vitally important to build a correct censorship column. Many data sources revise their data after reporting it - things like unemployment numbers are famous for this. Thus, when backtesting a strategy, one must use the data which was available in the past rather than the revised version.
Here's a strategy that uses this information:
>>> class WallStreetBetsStrategy(Strategy):
... """This is used just for the doctests."""
... def __init__(self, size, sell_size):
... self.size = size
... self.sell_size = sell_size
...
... def pre_open(self, dt, broker, trades, other_data):
... wsb_rec = other_data.get_data('wallstreetbets')
... wsb_rec = wsb_rec.set_index('date')
... if (dt == pd.Timestamp('2004-08-16')): # For illustrative purposes
... print("We can only see data from /r/wallstreetbets from " + str(wsb_rec.index.min()) + " to " + str(wsb_rec.index.max()) + ", which makes sense since today is " + str(dt))
... prices, _ = broker.historical_prices('tsla') # Open price is always null inside pre_open
... last_trade_dt = prices.index.max()
... last_trade_price = prices['close'][last_trade_dt]
... if wsb_rec['rating'][dt] > 0:
... buy_order_price = last_trade_price*1.05
... broker.limit_on_open('tsla', price=buy_order_price, size=self.size, is_buy=True, meta={'trade_id': str(self.size % 2)})
... self.log('my_custom_log', {'buy_order_price': buy_order_price, 'sell_order_price': None}, dt)
... else:
... sell_order_price = last_trade_price*0.95
... broker.limit_on_open('tsla', price=sell_order_price, size=self.sell_size, is_buy=False, meta={'trade_id': str(self.size % 2)})
... self.log('my_custom_log', {'buy_order_price': None, 'sell_order_price': sell_order_price}, dt)
...
... def pre_close(self, dt, broker, trades, other_data):
... pass
>>> m.set_strategy(WallStreetBetsStrategy(size=10, sell_size=7))
>>> m.run()
We can only see data from /r/wallstreetbets from 2004-08-12 00:00:00-04:00 to 2004-08-16 00:00:00-04:00, which makes sense since today is 2004-08-16 00:00:00
As we can see from the result of the print(...)
statement, the alternative data is properly censored.
>>> m.broker.trades()
commission date price size symbol trade_id
0 1.0 2004-08-13 09:30:00-04:00 17.50 10 tsla 0
1 1.0 2004-08-16 09:30:00-04:00 17.54 -7 tsla 0
2 1.0 2004-08-17 09:30:00-04:00 17.35 10 tsla 0
3 1.0 2004-08-18 09:30:00-04:00 17.25 -7 tsla 0
As expected, we opened a trade on the 13'th (due to the /r/wallstreetbets recommendation) and sold most of it on the 16'th (next trading day).
>>> m.broker.capital_gains()
close_commission_per_share close_date close_price close_trade_id open_commission_per_share open_date open_price open_trade_id size symbol gain
0 0.142857 2004-08-16 09:30:00-04:00 17.54 0 0.100000 2004-08-13 09:30:00-04:00 17.50 0 7 tsla -1.420000
1 0.142857 2004-08-18 09:30:00-04:00 17.25 0 0.142857 2004-08-16 09:30:00-04:00 17.50 0 3 tsla -1.607143
2 0.142857 2004-08-18 09:30:00-04:00 17.25 0 0.100000 2004-08-17 09:30:00-04:00 17.35 0 4 tsla -1.371429
In spite of buying and selling twice, there are actually 3 capital gains. We bought one batch of 10 shares at $17.50 (on 2004-08-13), and sold 7 of them (on 2004-08-16). Then on the 16'th, we bought 10 more shares at $17.35. On the 18'th, when we sold our shares, we first sold the remaining 3 shares we bought on 2004-08-13 and only then sold 4 more shares purchased on 2004-08-17.
This is the FIFO accounting system used by the American Internal Revenue Service.
Note our use of the log
method on Strategy
. The result of this can be accessed as follows:
>>> m.strategy_log('my_custom_log')
buy_order_price date sell_order_price
0 18.375 2004-08-13 NaN
1 NaN 2004-08-16 16.6345
2 18.375 2004-08-17 NaN
3 NaN 2004-08-18 16.4730
This can be used to track debugging information from inside the strategy.
For example, I am currently developing a market-neutral strategy that seems to randomly go long or short a few times a year. To debug, I will keep a log of the desired allocation as described by the strategy. If the desired allocation matches the actual, this tells me if the problem is in the core allocation logic. In contrast, if they differ, then some trades are simply failing to execute.