DAMASK_EICMD/python/tests/test_Lattice.py

42 lines
1.5 KiB
Python
Raw Normal View History

import random
import pytest
import numpy as np
from damask import Symmetry
class TestSymmetry:
@pytest.mark.parametrize('invalid_symmetry',['fcc','bcc','hello'])
2020-05-23 00:12:13 +05:30
def test_invalid_symmetry(self,invalid_symmetry):
with pytest.raises(KeyError):
2020-05-25 19:24:22 +05:30
s = Symmetry(invalid_symmetry) # noqa
def test_equal(self):
symmetry = random.choice(Symmetry.lattices)
2020-05-24 12:36:11 +05:30
print(symmetry)
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'])
2020-05-23 00:12:13 +05:30
def test_invalid_argument(self,function):
2020-05-25 19:24:22 +05:30
s = Symmetry() # noqa
with pytest.raises(ValueError):
2020-06-25 01:04:51 +05:30
eval(f's.{function}(np.ones(4))')