diff --git a/zipline/finance/performance/period.py b/zipline/finance/performance/period.py index 83c0671a..0951989d 100644 --- a/zipline/finance/performance/period.py +++ b/zipline/finance/performance/period.py @@ -122,9 +122,7 @@ class PerformancePeriod(object): self.keep_transactions = keep_transactions self.keep_orders = keep_orders - # Arrays for quick calculations of positions value - self._position_amounts = OrderedDict() - self._position_last_sale_prices = OrderedDict() + self.initialize_position_calc_arrays() self.calculate_performance() @@ -142,6 +140,11 @@ class PerformancePeriod(object): self.loc_map = {} + def initialize_position_calc_arrays(self): + # Arrays for quick calculations of positions value. + self._position_amounts = OrderedDict() + self._position_last_sale_prices = OrderedDict() + def set_positions(self, positions): self.positions = positions for sid, pos in positions.iteritems():