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

Use 2-point sources to test fdfd poynting

Browse Source
This commit is contained in:
Jan Petykiewicz 2019-11-04 20:30:45 -08:00
parent 0bb4105e7a
commit 68f857e83e
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
meanas/test/test_fdfd.py
View File

@ -10,8 +10,10 @@ from .utils import assert_close, assert_fields_close, PRNG
def test_poynting_planes(sim): def test_poynting_planes(sim):
mask = (sim.j != 0).any(axis=0) mask = (sim.j != 0).any(axis=0)
if mask.sum() > 1: if mask.sum() != 2:
pytest.skip(f'test_poynting_planes can only test single point sources, got {mask.sum()}') pytest.skip(f'test_poynting_planes will only test 2-point sources, got {mask.sum()}')
points = numpy.where(mask)
mask[points[0][0], points[1][0], points[2][0]] = 0
mx = numpy.roll(mask, -1, axis=0) mx = numpy.roll(mask, -1, axis=0)
my = numpy.roll(mask, -1, axis=1) my = numpy.roll(mask, -1, axis=1)