wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-07-12 15:37:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

Enriches message returned for test batch transform.

Browse Source 
Adds formatting of the failing values when ReturnPriceBatchTransform's
data shape does not match the window_length.
This commit is contained in:
Eddie Hebert
2013-01-30 14:04:17 -05:00
parent d52a9f3a67
commit 2587cd6512
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
zipline/test_algorithms.py
+3 -1
View File
@@ -237,7 +237,9 @@ class TestRegisterTransformAlgorithm(TradingAlgorithm):
class ReturnPriceBatchTransform(BatchTransform):
def get_value(self, data):
assert data.shape[1] == self.window_length
assert data.shape[1] == self.window_length, \
"data shape={0} does not equal window_length={1} for data={2}".\
format(data.shape[1], self.window_length, data)
return data.price