mirror of
https://github.com/wassname/simpeg.git
synced 2026-06-27 19:48:52 +08:00
Brendan Smithyman
63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
from __future__ import division
|
|
from __future__ import unicode_literals
|
|
from __future__ import print_function
|
|
from __future__ import absolute_import
|
|
from future import standard_library
|
|
standard_library.install_aliases()
|
|
import unittest, os
|
|
import numpy as np
|
|
from SimPEG import Utils
|
|
|
|
tol = 1e-15
|
|
|
|
class coorUtilsTest(unittest.TestCase):
|
|
|
|
def test_rotationMatrixFromNormals(self):
|
|
np.random.seed(0)
|
|
v0 = np.random.rand(3)
|
|
v0 *= old_div(1.,np.linalg.norm(v0))
|
|
|
|
np.random.seed(5)
|
|
v1 = np.random.rand(3)
|
|
v1 *= old_div(1.,np.linalg.norm(v1))
|
|
|
|
Rf = Utils.coordutils.rotationMatrixFromNormals(v0,v1)
|
|
Ri = Utils.coordutils.rotationMatrixFromNormals(v1,v0)
|
|
|
|
self.assertTrue(np.linalg.norm(Utils.mkvc(Rf.dot(v0) - v1)) < tol)
|
|
self.assertTrue(np.linalg.norm(Utils.mkvc(Ri.dot(v1) - v0)) < tol)
|
|
|
|
def test_rotatePointsFromNormals(self):
|
|
np.random.seed(10)
|
|
v0 = np.random.rand(3)
|
|
v0*= old_div(1.,np.linalg.norm(v0))
|
|
|
|
np.random.seed(15)
|
|
v1 = np.random.rand(3)
|
|
v1*= old_div(1.,np.linalg.norm(v1))
|
|
|
|
v2 = Utils.mkvc(Utils.coordutils.rotatePointsFromNormals(Utils.mkvc(v0,2).T,v0,v1))
|
|
|
|
self.assertTrue(np.linalg.norm(v2-v1) < tol)
|
|
|
|
def test_rotateMatrixFromNormals(self):
|
|
np.random.seed(20)
|
|
n0 = np.random.rand(3)
|
|
n0 *= old_div(1.,np.linalg.norm(n0))
|
|
|
|
np.random.seed(25)
|
|
n1 = np.random.rand(3)
|
|
n1 *= old_div(1.,np.linalg.norm(n1))
|
|
|
|
np.random.seed(30)
|
|
scale = np.random.rand(100,1)
|
|
XYZ0 = scale * n0
|
|
XYZ1 = scale * n1
|
|
|
|
XYZ2 = Utils.coordutils.rotatePointsFromNormals(XYZ0,n0,n1)
|
|
self.assertTrue(old_div(np.linalg.norm(Utils.mkvc(XYZ1) - Utils.mkvc(XYZ2)),np.linalg.norm(Utils.mkvc(XYZ1))) < tol)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|
|
|