def test_add_sub(self):
container1 = mm.Energy(terms=[self.exchange, self.dmi])
container2 = mm.Energy(terms=[self.zeeman, self.demag])
container3 = mm.Energy(terms=[self.demag])
assert container1 != container2
res = container1 + container2
check_container(res)
assert len(res) == 4
assert container1 + container2 == container2 + container1
res = container2 - container3
check_container(res)
assert len(res) == 1
assert self.demag not in res
with pytest.raises(ValueError):
res -= self.demag
assert res + self.demag == self.demag + res
with pytest.raises(TypeError):
res = container1 + 5
with pytest.raises(TypeError):
res = container1 - 5
def test_getattr(self):
container = mm.Energy(terms=self.terms)
check_container(container)
assert isinstance(container.exchange, mm.Exchange)
assert hasattr(container.exchange, 'A')
assert isinstance(container.zeeman, mm.Zeeman)
assert hasattr(container.zeeman, 'H')
assert isinstance(container.uniaxialanisotropy, mm.UniaxialAnisotropy)
assert hasattr(container.uniaxialanisotropy, 'K')
assert hasattr(container.uniaxialanisotropy, 'u')
assert isinstance(container.demag, mm.Demag)
assert isinstance(container.cubicanisotropy, mm.CubicAnisotropy)
assert hasattr(container.cubicanisotropy, 'K')
assert hasattr(container.cubicanisotropy, 'u1')
assert hasattr(container.cubicanisotropy, 'u2')
assert isinstance(container.dmi, mm.DMI)
assert hasattr(container.dmi, 'D')
assert hasattr(container.dmi, 'crystalclass')
# Try to get non-existing attribute.
container -= self.exchange
check_container(container)
with pytest.raises(AttributeError):
exchange = container.exchange
def test_init_invalid_args(self):
container = mm.Energy()
for term in self.invalid_terms:
with pytest.raises(TypeError):
container += term
check_container(container)
assert len(container) == 0
def energy(self, value):
empty_container = mm.Energy()
if value == 0:
self._energy = empty_container
elif isinstance(value, (mm.EnergyTerm, mm.Energy)):
self._energy = empty_container + value # checks by + operator
else:
msg = f'Cannot set energy equation with {type(value)}.'
raise TypeError(msg)
def test_repr(self):
container = mm.Energy(terms=self.terms)
check_container(container)
assert isinstance(repr(container), str)
assert 'Exchange' in repr(container)
container -= container.exchange
assert 'Exchange' not in repr(container)
def test_init(self):
# Init with terms list.
container = mm.Energy(terms=self.terms)
check_container(container)
assert len(container) == 6
# Empty terms list.
container = mm.Energy()
check_container(container)
assert len(container) == 0
# Add terms one by one.
for i, term in enumerate(self.terms):
container += term
check_container(container)
assert len(container) == i + 1
assert isinstance(container, mm.Energy)
# Create container as a sum of terms.
container = (self.exchange + self.zeeman + self.uniaxialanisotropy +
self.demag + self.dmi + self.cubicanisotropy)
check_container(container)
assert len(container) == 6
def test_repr_latex(self):
container = mm.Energy()
check_container(container)
latexstr = container._repr_latex_()
assert latexstr == '$0$'