-
Notifications
You must be signed in to change notification settings - Fork 589
/
test_ring.py
72 lines (54 loc) · 2.15 KB
/
test_ring.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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
# Python port of meep/examples/ring.ctl
# Calculating 2d ring-resonator modes, from the Meep tutorial.
import unittest
import meep as mp
class TestRing(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.temp_dir = mp.make_output_directory()
@classmethod
def tearDownClass(cls):
mp.delete_directory(cls.temp_dir)
def init(self):
n = 3.4
w = 1
r = 1
pad = 4
dpml = 2
sxy = 2 * (r + w + pad + dpml)
dielectric = mp.Medium(epsilon=n * n)
air = mp.Medium()
c1 = mp.Cylinder(r + w, material=dielectric)
c2 = mp.Cylinder(r, material=air)
fcen = 0.15
df = 0.1
src = mp.Source(mp.GaussianSource(fcen, fwidth=df), mp.Ez, mp.Vector3(r + 0.1))
self.sim = mp.Simulation(cell_size=mp.Vector3(sxy, sxy),
geometry=[c1, c2],
sources=[src],
resolution=10,
symmetries=[mp.Mirror(mp.Y)],
boundary_layers=[mp.PML(dpml)])
self.sim.use_output_directory(self.temp_dir)
self.h = mp.Harminv(mp.Ez, mp.Vector3(r + 0.1), fcen, df)
def test_harminv(self):
self.init()
self.sim.run(
mp.at_beginning(mp.output_epsilon),
mp.after_sources(self.h),
until_after_sources=300
)
m1 = self.h.modes[0]
self.assertAlmostEqual(m1.freq, 0.118101315147, places=4)
self.assertAlmostEqual(m1.decay, -0.000731513241623, places=4)
self.assertAlmostEqual(abs(m1.amp), 0.00341267634436, places=4)
self.assertAlmostEqual(m1.amp.real, -0.00304951667301, places=4)
self.assertAlmostEqual(m1.amp.imag, -0.00153192946717, places=3)
v = mp.Vector3(1, 1)
fp = self.sim.get_field_point(mp.Ez, v)
ep = self.sim.get_epsilon_point(v)
places = 5 if mp.is_single_precision() else 7
self.assertAlmostEqual(ep, 11.559999999999999, places=places)
self.assertAlmostEqual(fp, -0.08185972142450348, places=places)
if __name__ == '__main__':
unittest.main()