diff --git a/zipline/test_algorithms.py b/zipline/test_algorithms.py index 30528510..f9a958e2 100644 --- a/zipline/test_algorithms.py +++ b/zipline/test_algorithms.py @@ -75,6 +75,7 @@ from copy import deepcopy import numpy as np from six.moves import range +from six import itervalues from zipline.algorithm import TradingAlgorithm from zipline.finance.slippage import FixedSlippage @@ -653,7 +654,7 @@ class EmptyPositionsAlgorithm(TradingAlgorithm): if not self.exited: amounts = [pos.amount for pos - in self.portfolio.positions.itervalues()] + in itervalues(self.portfolio.positions)] if ( all([(amount == 100) for amount in amounts]) and (len(amounts) == len(data.keys()))