From adcae79da3059d82adc8d1a4e4d595abf423c27c Mon Sep 17 00:00:00 2001 From: Eddie Hebert Date: Fri, 10 Jan 2014 15:47:09 -0500 Subject: [PATCH] BUG: Make itervalues in empty position test Python 3 compatible. Use `six.itervalues` so that the call works in both Python 2 and 3. --- zipline/test_algorithms.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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()))