wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-07-28 11:18:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

heartbeating for handle_data

Browse Source
This commit is contained in:
scottsanderson
2012-08-19 02:22:16 -04:00
parent b5cb08ceed
commit 69ac68af2e
3 changed files with 110 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_exception_handling.py
+34 -1
View File
@@ -3,7 +3,8 @@ import zmq
from unittest2 import TestCase
from collections import defaultdict
from zipline.test_algorithms import ExceptionAlgorithm, DivByZeroAlgorithm
from zipline.test_algorithms import ExceptionAlgorithm, DivByZeroAlgorithm, \
InitializeTimeoutAlgorithm, TooMuchProcessingAlgorithm
from zipline.finance.trading import SIMULATION_STYLE
from zipline.core.devsimulator import AddressAllocator
from zipline.lines import SimulatedTrading
@@ -143,3 +144,35 @@ class ExceptionTestCase(TestCase):
# make sure our path shortening is working
self.assertEqual(payload['stack'][0]['filename'], '/zipline/lines.py')
self.assertEqual(payload['stack'][-1]['filename'], '/zipline/test_algorithms.py')
def test_initialize_timeout(self):
self.zipline_test_config['algorithm'] = \
InitializeTimeoutAlgorithm(
self.zipline_test_config['sid']
)
zipline = SimulatedTrading.create_test_zipline(
**self.zipline_test_config
)
output, _ = drain_zipline(self, zipline)
self.assertEqual(output[-1]['prefix'], 'EXCEPTION')
payload = output[-1]['payload']
self.assertEqual(payload['name'],'Timeout')
self.assertEqual(payload['message'], 'Call to initialize timed out')
# def test_heartbeat(self):
# self.zipline_test_config['algorithm'] = \
# TooMuchProcessingAlgorithm(
# self.zipline_test_config['sid']
# )
# zipline = SimulatedTrading.create_test_zipline(
# **self.zipline_test_config
# )
# output, _ = drain_zipline(self, zipline)
# self.assertEqual(output[-1]['prefix'], 'EXCEPTION')
# payload = output[-1]['payload']
# self.assertEqual(payload['name'],'Timeout')
# self.assertEqual(payload['message'], 'Too much time spent in handle_data call')