class EquationTestCase(unittest.TestCase):
def setUp(self):
self.equation = Equation(1,4.6,{2:-5.7,3:5.7})
def test_equation(self):
self.assertIsNotNone(self.equation)
def test_copy(self):
copyEquation= self.equation.copy()
self.assertNotEquals(self.equation,copyEquation)
def test_copyIntensive(self):
copyEquation= self.equation.copy()
self.assertEquals(self.equation.basicVar,copyEquation.basicVar)
self.assertEquals(self.equation.bValue,copyEquation.bValue)
self.assertEquals(self.equation.rhsDict,copyEquation.rhsDict)
self.assertNotEquals(id(self.equation.rhsDict),id(copyEquation.rhsDict))
def test_valueOfEntry(self):
equation = Equation(1,4.0,{2:-2.0,3:2.0})
self.assertEquals(equation.valueOfEntry(2),2.0)
self.assertEquals(equation.valueOfEntry(3),-2.0)
def test_valueOfEntryInvalidEntry(self):
self.assertIsNone(self.equation.valueOfEntry(9))
def test_valueOfEntryZeroCoeff(self):
eq = Equation(6,0.0,{2:-1.0,4:0.0,5:-2.0,7:0.0})
self.assertIsNone(eq.valueOfEntry(4))
def test_exitVarRebalanceInvalidIndex(self):
self.assertFalse(self.equation.exitVarRebalance(103))
def test_exitVarRebalanceInvalidCoeff(self):
self.equation.rhsDict[103] = 0.0
self.assertFalse(self.equation.exitVarRebalance(103))
def test_exitVarRebalance(self):
equation = Equation(1,4.0,{2:-2.0,3:2.0})
self.assertTrue(equation.exitVarRebalance(2))
self.assertEquals(equation.basicVar,2)
self.assertEquals(equation.bValue,2.0)
self.assertEquals(equation.rhsDict[1],-0.5)
self.assertEquals(equation.rhsDict[3],1.0)
self.assertFalse(equation.exitVarRebalance(2))
equation = Equation(4,0.0,{3:0.0,6:-1.0,2:1.0,7:3.0})
equation.exitVarRebalance(6)
self.assertTrue(Equation.equals(equation,Equation(6,0.0,{3:0.0,4:-1.0,2:1.0,7:3.0})))
def test_substituteEquationNone(self):
self.assertFalse(self.equation.substituteEquation(None))
def test_substituteEquationZeroCoeff(self):
equation1 = Equation(1,4.0,{2:-2.0,3:2.0})
equation2 = Equation(4,4.0,{1:0.0,3:2.0})
self.assertTrue(equation2.substituteEquation(equation1))
def test_substituteEquation(self):
equation1 = Equation(1,4.0,{2:-2.0,3:2.0})
equation2 = Equation(4,4.0,{1:-2.0,3:2.0})
self.assertTrue(equation2.substituteEquation(equation1))
self.assertEquals(equation2.basicVar,4)
self.assertEquals(equation2.bValue,-4.0)
self.assertEquals(equation2.rhsDict[2],4.0)
self.assertEquals(equation2.rhsDict[3],-2.0)
self.assertEquals(len(equation2.rhsDict.keys()),2)
def test_equationEqualsTrue(self):
self.assertTrue(Equation.equals(Equation(1,4.0,{2:-2.0,3:2.0}),Equation(1,4.0,{2:-2.0,3:2.0})))
self.assertTrue(Equation.equals(Equation(1,4.0,{3:2.0}),Equation(1,4.0,{3:2.0})))
self.assertTrue(Equation.equals(Equation(1,4.0,{4:0.0,3:2.0}),Equation(1,4.0,{3:2.0})))
def test_equationEqualsFalse(self):
self.assertFalse(Equation.equals(Equation(1,4.0,{6:-2.0,3:2.0}),Equation(1,4.0,{2:-2.0,3:2.0})))
self.assertFalse(Equation.equals(Equation(1,4.0,{2:0.0,3:2.0}),Equation(1,4.0,{2:-2.0,3:2.0})))
def test_valueOfEntryZeroCoeff(self):
eq = Equation(6,0.0,{2:-1.0,4:0.0,5:-2.0,7:0.0})
self.assertIsNone(eq.valueOfEntry(4))
def test_valueOfEntry(self):
equation = Equation(1,4.0,{2:-2.0,3:2.0})
self.assertEquals(equation.valueOfEntry(2),2.0)
self.assertEquals(equation.valueOfEntry(3),-2.0)
