diff --git a/tests/test_sorting.py b/tests/test_sorting.py index bec97e31..003fc889 100644 --- a/tests/test_sorting.py +++ b/tests/test_sorting.py @@ -1,22 +1,22 @@ import pytz from unittest2 import TestCase -from itertools import cycle, chain, izip, izip_longest +from itertools import chain, izip_longest from datetime import datetime, timedelta from collections import deque from zipline import ndict -from zipline.gens.sort import \ - date_sort, \ - ready, \ - done, \ - queue_is_ready,\ +from zipline.gens.sort import ( + date_sort, + ready, + done, + queue_is_ready, queue_is_done -from zipline.gens.utils import hash_args, alternate, done_message -from zipline.gens.tradegens import date_gen, SpecificEquityTrades +) +from zipline.gens.utils import alternate, done_message +from zipline.gens.tradegens import SpecificEquityTrades from zipline.gens.composites import date_sorted_sources -import zipline.protocol as zp class HelperTestCase(TestCase): diff --git a/tests/test_transforms.py b/tests/test_transforms.py index 8e203664..3672c4e1 100644 --- a/tests/test_transforms.py +++ b/tests/test_transforms.py @@ -2,14 +2,11 @@ import pytz import numpy from datetime import timedelta, datetime -from collections import defaultdict from unittest2 import TestCase from zipline import ndict -from zipline.lines import SimulatedTrading - -from zipline.utils.test_utils import setup_logger, teardown_logger +from zipline.utils.test_utils import setup_logger from zipline.utils.date_utils import utcnow from zipline.gens.tradegens import SpecificEquityTrades