diff --git a/tests/test_ndict.py b/tests/test_ndict.py index e2cb8a84..2fac4f56 100644 --- a/tests/test_ndict.py +++ b/tests/test_ndict.py @@ -1,4 +1,4 @@ -from zipline.protocol_utils import ndict, namedict +from zipline.utils.protocol_utils import ndict, namedict def test_ndict(): nd = ndict({}) diff --git a/tests/test_perf_tracking.py b/tests/test_perf_tracking.py index e952fed4..6e958014 100644 --- a/tests/test_perf_tracking.py +++ b/tests/test_perf_tracking.py @@ -4,18 +4,19 @@ import random import datetime import pytz -import zipline.test.factory as factory -import zipline.test.algorithms -import zipline.util as qutil +import zipline.utils.factory as factory +import zipline.test_algorithms +#import zipline.util as qutil import zipline.finance.performance as perf import zipline.finance.risk as risk import zipline.protocol as zp from zipline.finance.trading import TradeSimulationClient, TradingEnvironment, \ -SIMULATION_STYLE + SIMULATION_STYLE + class PerformanceTestCase(unittest.TestCase): def setUp(self): - qutil.configure_logging() + #qutil.configure_logging() self.benchmark_returns, self.treasury_curves = \ factory.load_market_data() @@ -565,4 +566,4 @@ shares in position" cumulative_pos = perf_tracker.cumulative_performance.positions[sid] expected_size = txn_count / 2 * -25 self.assertEqual(cumulative_pos.amount, expected_size) - \ No newline at end of file + diff --git a/tests/test_protocol.py b/tests/test_protocol.py index c0d4a7c7..ec39a352 100644 --- a/tests/test_protocol.py +++ b/tests/test_protocol.py @@ -9,8 +9,8 @@ from collections import defaultdict from nose.tools import timed -import zipline.test.factory as factory -import zipline.util as qutil +import zipline.utils.factory as factory +from zipline.utils import logger import zipline.protocol as zp from zipline.sources import SpecificEquityTrades @@ -22,7 +22,7 @@ class ProtocolTestCase(TestCase): leased_sockets = defaultdict(list) def setUp(self): - qutil.configure_logging() + #qutil.configure_logging() self.trading_environment = factory.create_trading_environment() @timed(DEFAULT_TIMEOUT) diff --git a/tests/test_risk.py b/tests/test_risk.py index 25685143..21b5785b 100644 --- a/tests/test_risk.py +++ b/tests/test_risk.py @@ -4,15 +4,14 @@ import datetime import calendar import pytz import zipline.finance.risk as risk -import zipline.test.factory as factory -import zipline.util as qutil +from zipline.utils import factory from zipline.finance.trading import TradingEnvironment class Risk(unittest.TestCase): def setUp(self): - qutil.configure_logging() + #qutil.configure_logging() start_date = datetime.datetime( year=2006, month=1, @@ -354,4 +353,4 @@ RETURNS = [ 0.048 , -0.0307, -0.0357, 0.0033, -0.0412, -0.0407, 0.0455, 0.0159, -0.0051, -0.0274, -0.0213, 0.0361, 0.0051, -0.0378, 0.0084, 0.0066, -0.0103, -0.0037, 0.0478, -0.0278 -] \ No newline at end of file +]