jan
/
fdfd_tools
2
2
Fork 1
Code Issues Pull Requests Releases Wiki Activity

add some more tests

Browse Source
fdtd_extras
Jan Petykiewicz 5 years ago
parent 935b2c9a80
commit a528effd89
1 changed files with 145 additions and 32 deletions
Show Stats Download Patch File Download Diff File

177
fdfd_tools/test_fdtd.py
Unescape Escape View File

@ -3,23 +3,94 @@ import numpy
from fdfd_tools import fdtd
class TestBasic2D(unittest.TestCase):
class BasicTests():
def test_initial_fields(self):
# Make sure initial fields didn't change
e0 = self.es[0]
h0 = self.hs[0]
mask = self.src_mask
self.assertEqual(e0[mask], self.j_mag / self.epsilon[mask])
self.assertFalse(e0[~mask].any())
self.assertFalse(h0.any())
def test_initial_energy(self):
e0 = self.es[0]
h0 = self.hs[0]
h1 = self.hs[1]
mask = self.src_mask[1]
u0 = self.j_mag * self.j_mag / self.epsilon[self.src_mask]
args = {'dxes': self.dxes,
'epsilon': self.epsilon}
# Make sure initial energy and E dot J are correct
energy0 = fdtd.energy_estep(h0=h0, e1=e0, h2=self.hs[1], **args)
e_dot_j_0 = fdtd.delta_energy_j(j0=(e0 - 0) * self.epsilon, e1=e0, dxes=self.dxes)
self.assertEqual(energy0[mask], u0)
self.assertFalse(energy0[~mask].any())
self.assertEqual(e_dot_j_0[mask], u0)
self.assertFalse(e_dot_j_0[~mask].any())
def test_energy_conservation(self):
e0 = self.es[0]
u0 = fdtd.delta_energy_j(j0=(e0 - 0) * self.epsilon, e1=e0).sum()
args = {'dxes': self.dxes,
'epsilon': self.epsilon}
for ii in range(1, 8):
with self.subTest(i=ii):
u_hstep = fdtd.energy_hstep(e0=self.es[ii-1], h1=self.hs[ii], e2=self.es[ii], **args)
u_estep = fdtd.energy_estep(h0=self.hs[ii], e1=self.es[ii], h2=self.hs[ii + 1], **args)
self.assertTrue(numpy.allclose(u_hstep.sum(), u0))
self.assertTrue(numpy.allclose(u_estep.sum(), u0))
def test_poynting(self):
args = {'dxes': self.dxes,
'epsilon': self.epsilon}
for ii in range(1, 8):
u_eprev = None
with self.subTest(i=ii):
u_hstep = fdtd.energy_hstep(e0=self.es[ii-1], h1=self.hs[ii], e2=self.es[ii], **args)
u_estep = fdtd.energy_estep(h0=self.hs[ii], e1=self.es[ii], h2=self.hs[ii + 1], **args)
du_half_h2e = u_estep - u_hstep
div_s_h2e = self.dt * fdtd.poynting_divergence(e=self.es[ii], h=self.hs[ii], dxes=self.dxes)
self.assertTrue(numpy.allclose(du_half_h2e, -div_s_h2e))
if u_eprev is None:
u_eprev = u_estep
continue
# previous half-step
du_half_e2h = u_hstep - u_eprev
div_s_e2h = self.dt * fdtd.poynting_divergence(e=self.es[ii], h=self.hs[ii-1], dxes=self.dxes)
self.assertTrue(numpy.allclose(du_half_e2h, -div_s_e2h))
u_eprev = u_estep
class Basic2DNoDXOnlyVacuum(unittest.TestCase, BasicTests):
def setUp(self):
shape = [3, 5, 5, 1]
dt = 0.5
epsilon = numpy.ones(shape, dtype=float)
j_mag = 32
dxes = None
src_mask = numpy.zeros_like(epsilon, dtype=bool)
src_mask[1, 2, 2, 0] = True
e = numpy.zeros_like(epsilon)
h = numpy.zeros_like(epsilon)
e[src_mask] = 32
e[src_mask] = j_mag / epsilon[src_mask]
es = [e]
hs = [h]
eh2h = fdtd.maxwell_h(dt=dt)
eh2e = fdtd.maxwell_e(dt=dt)
eh2h = fdtd.maxwell_h(dt=dt, dxes=dxes)
eh2e = fdtd.maxwell_e(dt=dt, dxes=dxes)
for _ in range(9):
e = e.copy()
h = h.copy()
@ -32,37 +103,79 @@ class TestBasic2D(unittest.TestCase):
self.hs = hs
self.dt = dt
self.epsilon = epsilon
self.dxes = dxes
self.src_mask = src_mask
def test_initial_fields(self):
# Make sure initial fields didn't change
e0 = self.es[0]
h0 = self.hs[0]
self.assertEqual(e0[1, 2, 2, 0], 32)
self.assertFalse(e0[~self.src_mask].any())
self.assertFalse(h0.any())
self.j_mag = j_mag
def test_initial_energy(self):
e0 = self.es[0]
h0 = self.hs[0]
h1 = self.hs[1]
mask = self.src_mask[1]
class Basic3DUniformDXOnlyVacuum(unittest.TestCase, BasicTests):
def setUp(self):
shape = [3, 5, 5, 5]
dt = 0.33
epsilon = numpy.ones(shape, dtype=float)
j_mag = 32
dxes = tuple(tuple(numpy.ones(s) for s in shape[1:]) for _ in range(2))
# Make sure initial energy and E dot J are correct
energy0 = fdtd.energy_estep(h0=h0, e1=e0, h2=self.hs[1])
e_dot_j_0 = fdtd.delta_energy_j(j0=e0 - 0, e1=e0)
self.assertEqual(energy0[mask], 32 * 32)
self.assertFalse(energy0[~mask].any())
self.assertEqual(e_dot_j_0[mask], 32 * 32)
self.assertFalse(e_dot_j_0[~mask].any())
src_mask = numpy.zeros_like(epsilon, dtype=bool)
src_mask[1, 2, 2, 0] = True
e = numpy.zeros_like(epsilon)
h = numpy.zeros_like(epsilon)
e[src_mask] = j_mag / epsilon[src_mask]
es = [e]
hs = [h]
eh2h = fdtd.maxwell_h(dt=dt, dxes=dxes)
eh2e = fdtd.maxwell_e(dt=dt, dxes=dxes)
for _ in range(9):
e = e.copy()
h = h.copy()
eh2h(e, h)
eh2e(e, h, epsilon)
es.append(e)
hs.append(h)
self.es = es
self.hs = hs
self.dt = dt
self.epsilon = epsilon
self.dxes = dxes
self.src_mask = src_mask
self.j_mag = j_mag
def test_energy_conservation(self):
for ii in range(1, 8):
with self.subTest(i=ii):
u_estep = fdtd.energy_estep(h0=self.hs[ii], e1=self.es[ii], h2=self.hs[ii + 1])
u_hstep = fdtd.energy_hstep(e0=self.es[ii-1], h1=self.hs[ii], e2=self.es[ii])
self.assertTrue(numpy.allclose(u_estep.sum(), 32 * 32))
self.assertTrue(numpy.allclose(u_hstep.sum(), 32 * 32))
class Basic3DUniformDX(unittest.TestCase, BasicTests):
def setUp(self):
shape = [3, 5, 5, 5]
dt = 0.33
epsilon = numpy.full(shape, 2, dtype=float)
j_mag = 32
dxes = tuple(tuple(numpy.ones(s) for s in shape[1:]) for _ in range(2))
src_mask = numpy.zeros_like(epsilon, dtype=bool)
src_mask[1, 2, 2, 0] = True
e = numpy.zeros_like(epsilon)
h = numpy.zeros_like(epsilon)
e[src_mask] = j_mag / epsilon[src_mask]
es = [e]
hs = [h]
eh2h = fdtd.maxwell_h(dt=dt, dxes=dxes)
eh2e = fdtd.maxwell_e(dt=dt, dxes=dxes)
for _ in range(9):
e = e.copy()
h = h.copy()
eh2h(e, h)
eh2e(e, h, epsilon)
es.append(e)
hs.append(h)
self.es = es
self.hs = hs
self.dt = dt
self.epsilon = epsilon
self.dxes = dxes
self.src_mask = src_mask
self.j_mag = j_mag

Write Preview
Loading…
Cancel
Save