mirror of
https://github.com/wassname/catalyst.git
synced 2026-08-14 12:10:24 +08:00
ENH: Make data available in before_trading_start.
This commit is contained in:
@@ -197,9 +197,13 @@ class FFCAlgorithmTestCase(TestCase):
|
||||
# unfortunate.
|
||||
assert_almost_equal(computed, expected, decimal=2)
|
||||
|
||||
# Do the same checks in before_trading_start
|
||||
before_trading_start = handle_data
|
||||
|
||||
algo = TradingAlgorithm(
|
||||
initialize=initialize,
|
||||
handle_data=handle_data,
|
||||
before_trading_start=before_trading_start,
|
||||
data_frequency='daily',
|
||||
ffc_loader=self.ffc_loader,
|
||||
asset_finder=self.asset_finder,
|
||||
|
||||
@@ -27,7 +27,7 @@ class BeforeTradingAlgorithm(TradingAlgorithm):
|
||||
self.before_trading_at = []
|
||||
super(BeforeTradingAlgorithm, self).__init__(*args, **kwargs)
|
||||
|
||||
def before_trading_start(self):
|
||||
def before_trading_start(self, data):
|
||||
self.before_trading_at.append(self.datetime)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user