def test_orthogonalize_dim2(self): '''test orthogonalization in 2 dimention''' v1 = lr.Vector([3, 1]) v2 = lr.Vector([-2, 3]) normv1 = lr.Vector([3/math.sqrt(10), 1/math.sqrt(10)]) normv2 = lr.Vector([-1/math.sqrt(10), 3/math.sqrt(10)]) resv = lr.orthogonalize([v1, v2]) self.assertEqual(resv[0],normv1) self.assertEqual(resv[1],normv2)
def test_orthogonalize_dim3(self): '''test orthogonalization in 3 dimention''' v1 = lr.Vector([1, 2, 0]) v2 = lr.Vector([2, 1, 3]) v3 = lr.Vector([2, 0, 1]) d1 = math.sqrt(5) d2 = math.sqrt(30) d3 = math.sqrt(6) normv1 = lr.Vector([1/d1, 2/d1, 0/d1]) normv2 = lr.Vector([2/d2, -1/d2, 5/d2]) normv3 = lr.Vector([2/d3, -1/d3, -1/d3]) resv = lr.orthogonalize([v1, v2, v3]) self.assertEqual(resv[0],normv1) self.assertEqual(resv[1],normv2) self.assertEqual(resv[2],normv3)