def testStockCorelation(self): StockCoRelation.addCorelation('1', '2', 0) assert(StockCoRelation.getCorelation( '1', '2')==0) assert(StockCoRelation.getCorelation( '2', '1')==0) StockCoRelation.addCorelation('1', '3', 1) assert(StockCoRelation.getCorelation( '1', '3')==1) assert(StockCoRelation.getCorelation( '3', '1')==1) StockCoRelation.addCorelation('2', '1', 2) assert(StockCoRelation.getCorelation( '1', '2')==2) assert(StockCoRelation.getCorelation( '2', '1')==2) StockCoRelation.addCorelation('3', '2', 3) assert(StockCoRelation.getCorelation( '2', '3')==3) assert(StockCoRelation.getCorelation( '3', '2')==3) StockCoRelation.resetMatrix()
def getCorelation(stock1, stock2): val = StockCoRelation.getCorelation(stock1.uid, stock2.uid) if val is None: val = getCorelation(stock1.uid, stock2.uid) return val