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 addStock(newStock):
listOfStocks = getAllStocks()
saveStock(newStock)
print "Added stock %s" % newStock
if len(listOfStocks) is not 0:
combinationsOfStockGroup = getCombination(newStock, listOfStocks)
print combinationsOfStockGroup
for stockGroup in combinationsOfStockGroup.values():
saveStockGroup(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()
def testStockAddingAndRemoving(self):
stock = Stock(str(9),'test9', (0.2+ 0.01), (0.1 + 0.01))
assert(saveStock(stock) is True)
stock2 = getStock(stock)
assert(stock2 is not None and stock2 == stock)
val = removeStock(stock2)
assert(val is True)
stock2 = getStock(stock)
assert(stock2 is None)
assert(saveStock(stock) is True)
stock2 = Stock(str(8),'test8', (0.2+ 0.01), (0.1 + 0.01))
assert(saveStock(stock2) is True)
stockList = getAllStocks()
assert(len(stockList) >=2 )
val = False
for stockReturned in stockList:
val = val or stockReturned.uid in ['9','8']
assert(val is True)
for stockReturned in stockList:
assert(removeStock(stockReturned) is True)
