diff --git a/tests/data/bundles/__init__.py b/tests/data/bundles/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/tests/data/bundles/test_yahoo.py b/tests/data/bundles/test_yahoo.py index 8c41d9d3..f2054e09 100644 --- a/tests/data/bundles/test_yahoo.py +++ b/tests/data/bundles/test_yahoo.py @@ -5,7 +5,7 @@ from toolz import flip, identity from toolz.curried import merge_with, operator as op 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.testing import test_resource_path, tmp_dir, read_compressed from zipline.testing.fixtures import WithResponses, ZiplineTestCase @@ -30,7 +30,9 @@ class YahooBundleTestCase(WithResponses, ZiplineTestCase): (cls.bundles, cls.register, cls.unregister, - cls.ingest) = map(staticmethod, _make_bundle_core()) + cls.ingest, + cls.load, + cls.clean) = map(staticmethod, _make_bundle_core()) def _expected_data(self): sids = 0, 1, 2 @@ -166,7 +168,7 @@ class YahooBundleTestCase(WithResponses, ZiplineTestCase): } self.ingest('bundle', environ=environ) - bundle = load('bundle', environ=environ) + bundle = self.load('bundle', environ=environ) sids = 0, 1, 2 equities = bundle.asset_finder.retrieve_all(sids)