223031012
/
DAMASK_EICMD
mirror of https://github.com/achalhp/DAMASK_EICMD.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

testing some special cases

This commit is contained in:
Martin Diehl 2020-04-08 13:41:46 +02:00
parent 8f88480790
commit 10d5b2e791
1 changed files with 48 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
python/tests/test_Rotation.py
View File

@ -4,13 +4,39 @@ import pytest
import numpy as np
from damask import Rotation
n = 1000
@pytest.fixture
def default():
"""A set of n random rotations."""
return [Rotation.fromRandom() for r in range(n)]
specials =[np.array([ 1.0, 0.0, 0.0, 0.0]),
#-----------------------------------------------
np.array([ 1.0, 1.0, 0.0, 0.0])*np.sqrt(2.)*.5,
np.array([ 1.0, 0.0, 1.0, 0.0])*np.sqrt(2.)*.5,
np.array([ 1.0, 0.0, 0.0, 1.0])*np.sqrt(2.)*.5,
np.array([ 0.0, 1.0, 1.0, 0.0])*np.sqrt(2.)*.5,
np.array([ 0.0, 1.0, 0.0, 1.0])*np.sqrt(2.)*.5,
np.array([ 0.0, 0.0, 1.0, 1.0])*np.sqrt(2.)*.5,
#-----------------------------------------------
np.array([ 1.0,-1.0, 0.0, 0.0])*np.sqrt(2.)*.5,
np.array([ 1.0, 0.0,-1.0, 0.0])*np.sqrt(2.)*.5,
np.array([ 1.0, 0.0, 0.0,-1.0])*np.sqrt(2.)*.5,
np.array([ 0.0, 1.0,-1.0, 0.0])*np.sqrt(2.)*.5,
np.array([ 0.0, 1.0, 0.0,-1.0])*np.sqrt(2.)*.5,
np.array([ 0.0, 0.0, 1.0,-1.0])*np.sqrt(2.)*.5,
#-----------------------------------------------
np.array([ 0.0, 1.0,-1.0, 0.0])*np.sqrt(2.)*.5,
np.array([ 0.0, 1.0, 0.0,-1.0])*np.sqrt(2.)*.5,
np.array([ 0.0, 0.0, 1.0,-1.0])*np.sqrt(2.)*.5,
#-----------------------------------------------
np.array([ 0.0,-1.0,-1.0, 0.0])*np.sqrt(2.)*.5,
np.array([ 0.0,-1.0, 0.0,-1.0])*np.sqrt(2.)*.5,
np.array([ 0.0, 0.0,-1.0,-1.0])*np.sqrt(2.)*.5,
#-----------------------------------------------
]
return [Rotation.fromQuaternion(s) for s in specials] + \
[Rotation.fromRandom() for r in range(n-len(specials))]
@pytest.fixture
def reference_dir(reference_dir_base):
@ -22,35 +48,41 @@ class TestRotation:
def test_Eulers(self,default):
for rot in default:
assert np.allclose(rot.asQuaternion(),
Rotation.fromEulers(rot.asEulers()).asQuaternion())
c = Rotation.fromEulers(rot.asEulers())
ok = np.allclose(rot.asQuaternion(),c.asQuaternion())
if np.isclose(rot.asQuaternion()[0],0.0,atol=1.e-13,rtol=0.0):
ok = ok or np.allclose(rot.asQuaternion(),c.asQuaternion()*-1.)
assert ok
def test_AxisAngle(self,default):
for rot in default:
assert np.allclose(rot.asEulers(),
Rotation.fromAxisAngle(rot.asAxisAngle()).asEulers())
c = Rotation.fromAxisAngle(rot.asAxisAngle())
assert np.allclose(rot.asEulers(),c.asEulers())
def test_Matrix(self,default):
for rot in default:
assert np.allclose(rot.asAxisAngle(),
Rotation.fromMatrix(rot.asMatrix()).asAxisAngle())
c = Rotation.fromMatrix(rot.asMatrix())
ok = np.allclose(rot.asAxisAngle(),c.asAxisAngle())
if np.isclose(rot.asAxisAngle()[3],np.pi):
ok = ok or np.allclose(rot.asAxisAngle(),c.asAxisAngle()*np.array([-1.,-1.,-1.,1.]))
assert ok
def test_Rodriques(self,default):
for rot in default:
assert np.allclose(rot.asMatrix(),
Rotation.fromRodrigues(rot.asRodrigues()).asMatrix())
c = Rotation.fromRodrigues(rot.asRodrigues())
assert np.allclose(rot.asMatrix(),c.asMatrix())
def test_Homochoric(self,default):
for rot in default:
assert np.allclose(rot.asRodrigues(),
Rotation.fromHomochoric(rot.asHomochoric()).asRodrigues(),rtol=1.e-4)
c = Rotation.fromHomochoric(rot.asHomochoric())
assert np.allclose(np.clip(rot.asRodrigues(),None,1.e9),np.clip(c.asRodrigues(),None,1.e9))
def test_Cubochoric(self,default):
for rot in default:
assert np.allclose(rot.asHomochoric(),
Rotation.fromCubochoric(rot.asCubochoric()).asHomochoric())
c = Rotation.fromCubochoric(rot.asCubochoric())
assert np.allclose(rot.asHomochoric(),c.asHomochoric())
def test_Quaternion(self,default):
for rot in default:
assert np.allclose(rot.asCubochoric(),
Rotation.fromQuaternion(rot.asQuaternion()).asCubochoric())
c = Rotation.fromQuaternion(rot.asQuaternion())
assert np.allclose(rot.asCubochoric(),c.asCubochoric())