def stockGroupWeightUpdater(weightClass, weightVector): stockGroups = getAllStockGroupsByClass(weightClass) for stockGroup in stockGroups: stockGroup.setPlotBestRewardByRisk((weightClass, weightVector)) if stockGroup.isThresholdClear(): notifyExpectants(stockGroup) saveStockGroup(stockGroup) print stockGroups
def testToRemoveAllStocksAndGroups(self): stocks = getAllStocks() for stock in stocks: removeStock(stock) for i in range(0,12): stockGroups = getAllStockGroupsByClass(i) for stockGroup in stockGroups: removeStockGroups(stockGroup)
def testAddStock(self): print "Current Stocks: " print "%s" % ("\n".join([str(elem) for elem in getAllStocks()])) stockList = [] print "Adding stocks..." for i in range(0, 10): stock = Stock(str(i), "test" + str(i), (0.2 + 0.01 * i), (0.1 + 0.01 * i)) stockList.append(stock) addStock(stock) # addStock() storedStocks = getAllStocks() for stock in storedStocks: print "Stored : %s" % stock for i in range(0, 12): storedStockGroups = getAllStockGroupsByClass(i) for stockGroup in storedStockGroups: print stockGroup.getGroupId()