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

ENH: Replace update_asset_finder with write_data

Browse Source 
The write_data methods invokes the relevant AssetDBWriter subclass
to write data to the database. update_asset_finder is no longer
a relevant method since the AssetFinder is strictly a reader class.
This commit is contained in:
Stewart Douglas
2015-09-10 11:53:24 -04:00
committed by jfkirk
parent 9660447ed0
commit 501fd58fdf
9 changed files with 76 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_blotter.py
+1 -1
View File
@@ -38,7 +38,7 @@ class BlotterTestCase(TestCase):
@with_environment()
def setUp(self, env=None):
setup_logger(self)
env.update_asset_finder(identifiers=[24])
env.write_data(equities_identifiers=[24])
def tearDown(self):
teardown_logger(self)
tests/test_perf_tracking.py
+5 -7
View File
@@ -1894,7 +1894,7 @@ class TestPerformanceTracker(unittest.TestCase):
foosid = 1
barsid = 2
env.update_asset_finder(identifiers=[foosid, barsid])
env.write_data(equities_identifiers=[foosid, barsid])
foo_event_1 = factory.create_trade(foosid, 10.0, 20, start_dt)
order_event_1 = Order(sid=foo_event_1.sid,
@@ -1989,7 +1989,7 @@ class TestPerformanceTracker(unittest.TestCase):
@with_environment()
def test_close_position_event(self, env=None):
env.update_asset_finder(identifiers=[1, 2])
env.write_data(equities_identifiers=[1, 2])
pt = perf.PositionTracker()
dt = pd.Timestamp("1984/03/06 3:00PM")
pos1 = perf.Position(1, amount=np.float64(120.0),
@@ -2133,9 +2133,7 @@ class TestPositionTracker(unittest.TestCase):
2: {'asset_type': 'future',
'contract_multiplier': 1000}}
asset_finder = AssetFinder()
env.update_asset_finder(
asset_finder=asset_finder,
asset_metadata=metadata)
env.write_data(equities_data=metadata)
pt = perf.PositionTracker()
dt = pd.Timestamp("1984/03/06 3:00PM")
pos1 = perf.Position(1, amount=np.float64(100.0),
@@ -2163,7 +2161,7 @@ class TestPositionTracker(unittest.TestCase):
'contract_multiplier': 1000},
4: {'asset_type': 'future',
'contract_multiplier': 1000}}
env.update_asset_finder(asset_metadata=metadata)
env.write_data(equities_data=metadata)
pt = perf.PositionTracker()
dt = pd.Timestamp("1984/03/06 3:00PM")
pos1 = perf.Position(1, amount=np.float64(10.0),
@@ -2197,7 +2195,7 @@ class TestPositionTracker(unittest.TestCase):
metadata = {1: {'asset_type': 'equity'},
2: {'asset_type': 'future',
'contract_multiplier': 1000}}
env.update_asset_finder(asset_metadata=metadata)
env.write_data(equities_data=metadata)
pt = perf.PositionTracker()
dt = pd.Timestamp("1984/03/06 3:00PM")
pos1 = perf.Position(1, amount=np.float64(120.0),
tests/test_security_list.py
+2 -3
View File
@@ -66,9 +66,8 @@ class SecurityListTestCase(TestCase):
self.trading_day_before_first_kd = datetime(
2015, 1, 23, 0, 0, tzinfo=pytz.utc)
env.update_asset_finder(
clear_metadata=True,
identifiers=["BZQ", "URTY", "JFT", "AAPL", "GOOG"]
env.write_data(
equities_identifiers=["BZQ", "URTY", "JFT", "AAPL", "GOOG"]
)
setup_logger(self)