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
|
from __future__ import division
import unittest
import meep as mp
import numpy as np
class Test3rdHarm1d(unittest.TestCase):
def setUp(self):
self.sz = 100
fcen = 1 / 3.0
df = fcen / 20.0
self.amp = 1.0
self.k = 1e-2
self.dpml = 1.0
dimensions = 1
cell = mp.Vector3(0, 0, self.sz)
default_material = mp.Medium(index=1, chi3=self.k)
pml_layers = mp.PML(self.dpml)
sources = mp.Source(mp.GaussianSource(fcen, fwidth=df), component=mp.Ex,
center=mp.Vector3(0, 0, (-0.5 * self.sz) + self.dpml), amplitude=self.amp)
nfreq = 400
fmin = fcen / 2.0
fmax = fcen * 4
self.sim = mp.Simulation(cell_size=cell,
geometry=[],
sources=[sources],
boundary_layers=[pml_layers],
default_material=default_material,
resolution=20,
dimensions=dimensions)
fr = mp.FluxRegion(mp.Vector3(0, 0, (0.5 * self.sz) - self.dpml - 0.5))
self.trans = self.sim.add_flux(0.5 * (fmin + fmax), fmax - fmin, nfreq, fr)
self.trans1 = self.sim.add_flux(fcen, 0, 1, fr)
self.trans3 = self.sim.add_flux(3 * fcen, 0, 1, fr)
def test_3rd_harm_1d(self):
expected_harmonics = [0.01, 1.0, 221.89548712071553, 1.752960413399477]
self.sim.run(
until_after_sources=mp.stop_when_fields_decayed(
50, mp.Ex, mp.Vector3(0, 0, (0.5 * self.sz) - self.dpml - 0.5), 1e-6
)
)
harmonics = [self.k, self.amp, mp.get_fluxes(self.trans1)[0], mp.get_fluxes(self.trans3)[0]]
np.testing.assert_allclose(expected_harmonics, harmonics)
if __name__ == '__main__':
unittest.main()
|