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

set fdfd sources in a way that catches incorrect j.e calculation

Browse Source
This commit is contained in:
Jan Petykiewicz 2024-01-17 23:30:08 -08:00
parent c3b6fd94a6
commit 107c0fcd7e
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
meanas/test/test_fdfd.py
View File

@ -26,6 +26,8 @@ def test_poynting_planes(sim: 'FDResult') -> None:
# for dxa in dxg: # for dxa in dxg:
# if not (dxa == sim.dxes[0][0][0]).all(): # if not (dxa == sim.dxes[0][0][0]).all():
# pytest.skip('test_poynting_planes skips nonuniform dxes') # pytest.skip('test_poynting_planes skips nonuniform dxes')
# pick only the second point
points = numpy.where(mask) points = numpy.where(mask)
mask[points[0][0], points[1][0], points[2][0]] = 0 mask[points[0][0], points[1][0], points[2][0]] = 0
@ -50,7 +52,7 @@ def test_poynting_planes(sim: 'FDResult') -> None:
) )
src_energy = -(e_dot_j.real * dv)[:, mask] / 2 src_energy = -(e_dot_j.real * dv)[:, mask] / 2
assert_close(sum(planes), src_energy.sum()) assert_close(sum(planes), src_energy.sum(), rtol=1e-6) # TODO improve energy calculation accuracy?
##################################### #####################################
@ -92,8 +94,8 @@ def j_distribution(
if request.param == 'center': if request.param == 'center':
j[center_mask] = j_mag j[center_mask] = j_mag
elif request.param == 'diag': elif request.param == 'diag':
j[numpy.roll(center_mask, [1, 1, 1], axis=(1, 2, 3))] = j_mag j[numpy.roll(center_mask, [1, 1, 1], axis=(1, 2, 3))] = (1 + 1j) * j_mag
j[numpy.roll(center_mask, [-1, -1, -1], axis=(1, 2, 3))] = -1j * j_mag j[numpy.roll(center_mask, [-1, -1, -1], axis=(1, 2, 3))] = (1 - 1j) * j_mag
yield j yield j