alihan/backtesting.py
1
0
Fork 0
mirror of https://github.com/kernc/backtesting.py.git synced 2024-01-28 15:29:30 +03:00
Code Issues Packages Projects Releases Wiki Activity

REF: Slightly more robust input handling in places

Browse Source
This commit is contained in:
Kernc
2019-08-04 02:16:45 +02:00
parent daa1da96ea
commit 5051b61618
2 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
backtesting/lib.py
View File

@@ -197,9 +197,10 @@ def resample_apply(rule: str,
self.sma = self.I(SMA, daily, 10, plot=False)
"""
if not isinstance(series, pd.Series):
if not isinstance(series, (pd.Series, pd.DataFrame)):
assert isinstance(series, _Array), \
'resample_apply() takes either a `pd.Series` or a `Strategy.data.*` array'
'resample_apply() takes either a `pd.Series`, `pd.DataFrame`, ' \
'or a `Strategy.data.*` array'
series = series.to_series()
resampled = series.resample(rule, label='right').agg('last').dropna()