-
Notifications
You must be signed in to change notification settings - Fork 35
/
test_obj_Dipole.py
41 lines (32 loc) · 997 Bytes
/
test_obj_Dipole.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
import numpy as np
import magpylib as magpy
def test_Dipole_basicB():
""" Basic dipole class test
"""
src = magpy.misc.Dipole(moment=(1,2,3),position=(1,2,3))
sens = magpy.Sensor()
B = src.getB(sens)
Btest = np.array([0.00303828,0.00607656,0.00911485])
assert np.allclose(B, Btest)
def test_Dipole_basicH():
""" Basic dipole class test
"""
src = magpy.misc.Dipole(moment=(1,2,3),position=(1,2,3))
sens = magpy.Sensor()
H = src.getH(sens)
Htest = np.array([0.00241779, 0.00483558, 0.00725336])
assert np.allclose(H, Htest)
def test_Dipole_zero_position():
""" Basic dipole class test
"""
src = magpy.misc.Dipole(moment=(1,2,3))
sens = magpy.Sensor()
np.seterr(all='ignore')
B = magpy.getB(src,sens)
np.seterr(all='print')
assert all(np.isnan(B))
def test_repr():
""" test __repr__
"""
dip = magpy.misc.Dipole(moment=(1,2,3))
assert dip.__repr__()[:6] == 'Dipole', 'Dipole repr failed'