def testStockCorelation(self): stock1 = Stock('T1','test1', (0.2+ 0.01), (0.1 + 0.01)) stock2 = Stock('T2','test2', (0.2+ 0.01), (0.1 + 0.01)) assert( saveCorelation(stock1.uid, stock2.uid, -1) is True) assert( getCorelation(stock1.uid, stock2.uid) == -1) assert( removeCorelation(stock1.uid, stock2.uid)) assert( getCorelation(stock1.uid, stock2.uid) is None)
def addStockCorelation(stock1, stock2, correlationValue): StockCoRelation.addCorelation(stock1.uid, stock2.uid, correlationValue) saveCorelation(stock1.uid, stock2.uid, correlationValue)