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

TST: fix bundle test discovery

Browse Source
This commit is contained in:
Joe Jevnik
2016-07-25 13:09:55 -04:00
committed by LotannaEzenwa
parent 2fe94d0c29
commit ef4eafbbb8
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/data/bundles/__init__.py
View File
tests/data/bundles/test_yahoo.py
+5 -3
View File
@@ -5,7 +5,7 @@ from toolz import flip, identity
from toolz.curried import merge_with, operator as op from toolz.curried import merge_with, operator as op
from zipline.data.bundles.core import _make_bundle_core from zipline.data.bundles.core import _make_bundle_core
from zipline.data.bundles import yahoo_equities, load from zipline.data.bundles import yahoo_equities
from zipline.lib.adjustment import Float64Multiply from zipline.lib.adjustment import Float64Multiply
from zipline.testing import test_resource_path, tmp_dir, read_compressed from zipline.testing import test_resource_path, tmp_dir, read_compressed
from zipline.testing.fixtures import WithResponses, ZiplineTestCase from zipline.testing.fixtures import WithResponses, ZiplineTestCase
@@ -30,7 +30,9 @@ class YahooBundleTestCase(WithResponses, ZiplineTestCase):
(cls.bundles, (cls.bundles,
cls.register, cls.register,
cls.unregister, cls.unregister,
cls.ingest) = map(staticmethod, _make_bundle_core()) cls.ingest,
cls.load,
cls.clean) = map(staticmethod, _make_bundle_core())
def _expected_data(self): def _expected_data(self):
sids = 0, 1, 2 sids = 0, 1, 2
@@ -166,7 +168,7 @@ class YahooBundleTestCase(WithResponses, ZiplineTestCase):
} }
self.ingest('bundle', environ=environ) self.ingest('bundle', environ=environ)
bundle = load('bundle', environ=environ) bundle = self.load('bundle', environ=environ)
sids = 0, 1, 2 sids = 0, 1, 2
equities = bundle.asset_finder.retrieve_all(sids) equities = bundle.asset_finder.retrieve_all(sids)