Python StockCoRelation.getCorelation Examples

Example #1

File: GenericTest.py Project: arshadansari27/Portfolio-optimization

 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()

Example #2

File: stockcombinationgenerator.py Project: arshadansari27/Portfolio-optimization

def getCorelation(stock1, stock2):
    val = StockCoRelation.getCorelation(stock1.uid, stock2.uid)
    if val is None:
        val = getCorelation(stock1.uid, stock2.uid)
    return val