42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
import random
|
|
|
|
import pytest
|
|
import numpy as np
|
|
|
|
from damask import Symmetry
|
|
from damask import Lattice
|
|
|
|
class TestSymmetry:
|
|
|
|
@pytest.mark.parametrize('invalid_symmetry',['fcc','bcc','hello'])
|
|
def test_invalid_symmetry(self,invalid_symmetry):
|
|
with pytest.raises(KeyError):
|
|
s = Symmetry(invalid_symmetry)
|
|
|
|
def test_equal(self):
|
|
symmetry = random.choice(Symmetry.lattices)
|
|
assert Symmetry(symmetry) == Symmetry(symmetry)
|
|
|
|
def test_not_equal(self):
|
|
symmetries = random.sample(Symmetry.lattices,k=2)
|
|
assert Symmetry(symmetries[0]) != Symmetry(symmetries[1])
|
|
|
|
@pytest.mark.parametrize('lattice',Symmetry.lattices)
|
|
def test_inFZ(self,lattice):
|
|
assert Symmetry(lattice).inFZ(np.zeros(3))
|
|
|
|
@pytest.mark.parametrize('lattice',Symmetry.lattices)
|
|
def test_inDisorientationSST(self,lattice):
|
|
assert Symmetry(lattice).inDisorientationSST(np.zeros(3))
|
|
|
|
@pytest.mark.parametrize('lattice',Symmetry.lattices)
|
|
@pytest.mark.parametrize('proper',[True,False])
|
|
def test_inSST(self,lattice,proper):
|
|
assert Symmetry(lattice).inSST(np.zeros(3),proper)
|
|
|
|
@pytest.mark.parametrize('function',['inFZ','inDisorientationSST'])
|
|
def test_invalid_argument(self,function):
|
|
s = Symmetry()
|
|
with pytest.raises(ValueError):
|
|
eval('s.{}(np.ones(4))'.format(function))
|