66 lines
2.7 KiB
Python
66 lines
2.7 KiB
Python
import os
|
|
from itertools import permutations
|
|
|
|
import pytest
|
|
import numpy as np
|
|
|
|
import damask
|
|
from damask import Rotation
|
|
from damask import Orientation
|
|
from damask import Lattice
|
|
|
|
n = 1000
|
|
|
|
@pytest.fixture
|
|
def default():
|
|
"""A set of n random rotations."""
|
|
return [Rotation.from_random() for r in range(n)]
|
|
|
|
@pytest.fixture
|
|
def reference_dir(reference_dir_base):
|
|
"""Directory containing reference results."""
|
|
return os.path.join(reference_dir_base,'Rotation')
|
|
|
|
|
|
class TestOrientation:
|
|
|
|
@pytest.mark.parametrize('color',[{'label':'red', 'RGB':[1,0,0],'direction':[0,0,1]},
|
|
{'label':'green','RGB':[0,1,0],'direction':[0,1,1]},
|
|
{'label':'blue', 'RGB':[0,0,1],'direction':[1,1,1]}])
|
|
@pytest.mark.parametrize('lattice',['fcc','bcc'])
|
|
def test_IPF_cubic(self,default,color,lattice):
|
|
cube = damask.Orientation(damask.Rotation(),lattice)
|
|
for direction in set(permutations(np.array(color['direction']))):
|
|
assert np.allclose(cube.IPFcolor(direction),np.array(color['RGB']))
|
|
|
|
@pytest.mark.parametrize('lattice',Lattice.lattices)
|
|
def test_IPF(self,lattice):
|
|
direction = np.random.random(3)*2.0-1
|
|
for rot in [Rotation.from_random() for r in range(n//100)]:
|
|
R = damask.Orientation(rot,lattice)
|
|
color = R.IPFcolor(direction)
|
|
for equivalent in R.equivalentOrientations():
|
|
assert np.allclose(color,R.IPFcolor(direction))
|
|
|
|
@pytest.mark.parametrize('model',['Bain','KS','GT','GT_prime','NW','Pitsch'])
|
|
@pytest.mark.parametrize('lattice',['fcc','bcc'])
|
|
def test_relationship_forward_backward(self,model,lattice):
|
|
ori = Orientation(Rotation.from_random(),lattice)
|
|
for i,r in enumerate(ori.relatedOrientations(model)):
|
|
ori2 = r.relatedOrientations(model)[i]
|
|
misorientation = ori.rotation.misorientation(ori2.rotation)
|
|
assert misorientation.asAxisAngle(degrees=True)[3]<1.0e-5
|
|
|
|
@pytest.mark.parametrize('model',['Bain','KS','GT','GT_prime','NW','Pitsch'])
|
|
@pytest.mark.parametrize('lattice',['fcc','bcc'])
|
|
def test_relationship_reference(self,update,reference_dir,model,lattice):
|
|
reference = os.path.join(reference_dir,'{}_{}.txt'.format(lattice,model))
|
|
ori = Orientation(Rotation(),lattice)
|
|
eu = np.array([o.rotation.asEulers(degrees=True) for o in ori.relatedOrientations(model)])
|
|
if update:
|
|
coords = np.array([(1,i+1) for i,x in enumerate(eu)])
|
|
table = damask.Table(eu,{'Eulers':(3,)})
|
|
table.add('pos',coords)
|
|
table.to_ASCII(reference)
|
|
assert np.allclose(eu,damask.Table.from_ASCII(reference).get('Eulers'))
|