testing functionality of Lattice separately

This commit is contained in:
Martin Diehl 2020-05-22 18:45:27 +02:00
parent b33de48528
commit 11eda510aa
1 changed files with 41 additions and 0 deletions

View File

@ -0,0 +1,41 @@
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(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(self,function):
s = Symmetry()
with pytest.raises(ValueError):
eval('s.{}(np.ones(4))'.format(function))