def test_index_in_sorted_list(): """Test the index_in_sorted_list function.""" list = [1, 3, 5, 7, 9, 11] x = 5 expected = 2 assert_equal(index_in_sorted_list(x, list), expected) x = 4 expected = -1 assert_equal(index_in_sorted_list(x, list), expected)
def get_end_of_day(self, symbol): """Gets the EndOfDay object associated with the given stock. symbol -- The ticker symbol of the stock to get. The start_date and end_date of the returned object will match the start_date and end_date of the simulation. """ if not self.__end_of_day_items.has_key(symbol): self.__end_of_day_items[symbol] = self.end_of_day_class(symbol, self.start_date, self.end_date) if not self.__date_offsets.has_key(symbol): dates = self.__end_of_day_items[Config().BACKTEST_SYMBOL_FOR_ALL_DATES].dates self.__date_offsets[symbol] = index_in_sorted_list(self.__end_of_day_items[symbol].dates[0], dates) return self.__end_of_day_items[symbol]