jan/meanas
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add test_poynting_planes

Browse Source
This commit is contained in:
Jan Petykiewicz 2019-07-19 00:19:32 -07:00
parent 950e70213a
commit fb3c88a78d
1 changed files with 25 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
fdfd_tools/test_fdtd.py
View File

@ -3,6 +3,7 @@ import numpy
from fdfd_tools import fdtd from fdfd_tools import fdtd
class BasicTests(): class BasicTests():
def test_initial_fields(self): def test_initial_fields(self):
# Make sure initial fields didn't change # Make sure initial fields didn't change
@ -49,7 +50,7 @@ class BasicTests():
self.assertTrue(numpy.allclose(u_estep.sum(), u0)) self.assertTrue(numpy.allclose(u_estep.sum(), u0))
def test_poynting(self): def test_poynting_divergence(self):
args = {'dxes': self.dxes, args = {'dxes': self.dxes,
'epsilon': self.epsilon} 'epsilon': self.epsilon}
@ -74,6 +75,29 @@ class BasicTests():
u_eprev = u_estep u_eprev = u_estep
def test_poynting_planes(self):
args = {'dxes': self.dxes,
'epsilon': self.epsilon}
u_eprev = None
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)
mx = numpy.roll(self.src_mask, (-1, -1), axis=(0, 1))
my = numpy.roll(self.src_mask, -1, axis=2)
mz = numpy.roll(self.src_mask, (+1, -1), axis=(0, 3))
px = numpy.roll(self.src_mask, -1, axis=0)
py = self.src_mask.copy()
pz = numpy.roll(self.src_mask, +1, axis=0)
s_h2e = -fdtd.poynting(e=self.es[ii], h=self.hs[ii]) * self.dt
planes = [s_h2e[px].sum(), -s_h2e[mx].sum(),
s_h2e[py].sum(), -s_h2e[my].sum(),
s_h2e[pz].sum(), -s_h2e[mz].sum()]
self.assertTrue(numpy.allclose(sum(planes), (u_estep - u_hstep)[self.src_mask[1]]))
class Basic2DNoDXOnlyVacuum(unittest.TestCase, BasicTests): class Basic2DNoDXOnlyVacuum(unittest.TestCase, BasicTests):
def setUp(self): def setUp(self):
shape = [3, 5, 5, 1] shape = [3, 5, 5, 1]