def test_repr(self):
for arg in self.valid_args:
u, beta = arg
stt = mm.STT(u, beta)
assert repr(stt) == ('STT(u={}, beta={}, '
'name=\'stt\')'.format(u, beta))
stt = mm.STT((1, 2, 3), 15, name='test_name')
assert repr(stt) == 'STT(u=(1, 2, 3), beta=15, name=\'test_name\')'
def test_repr(self):
for arg in self.valid_args:
u, beta = arg
stt = mm.STT(u, beta)
assert repr(stt) == "STT(u={}, beta={}, name=\"stt\")".format(
u, beta)
stt = mm.STT((1, 2, 3), 15, name="test_name")
assert repr(stt) == 'STT(u=(1, 2, 3), beta=15, name=\"test_name\")'
def test_init_valid_args(self):
for arg in self.valid_args:
u, beta = arg
stt = mm.STT(u, beta)
assert stt.u == u
assert isinstance(stt.u, tuple)
assert stt.beta == beta
assert isinstance(stt.beta, numbers.Real)
def setup(self):
gamma = 2.21e5
self.precession = mm.Precession(gamma)
alpha = 0.5
self.damping = mm.Damping(alpha)
u = (0, 0, 500)
beta = 0.2
self.stt = mm.STT(u=u, beta=beta)
self.terms = [self.precession, self.damping, self.stt]
self.invalid_terms = [
1, 2.5, 0, 'abc', [3, 7e-12], [self.precession, self.damping]
]
def test_repr_latex_(self):
for arg in self.valid_args:
u, beta = arg
stt = mm.STT(u, beta)
latex = stt._repr_latex_()
# Assert some characteristics of LaTeX string.
assert isinstance(latex, str)
assert latex[0] == latex[-1] == '$'
assert r'\beta' in latex
assert r'\mathbf{m}' in latex
assert r'\mathbf{u}' in latex
assert r'\times' in latex
assert r'\boldsymbol' in latex
assert r'\nabla' in latex
def test_script(self):
for arg in self.valid_args:
u, beta = arg
stt = mm.STT(u, beta)
with pytest.raises(NotImplementedError):
script = stt._script
def test_name(self):
for arg in self.valid_args:
u, beta = arg
stt = mm.STT(u, beta)
assert stt.name == 'stt'
def test_init_invalid_args(self):
for arg in self.invalid_args:
u, beta = arg
with pytest.raises(Exception):
stt = mm.STT(gamma)