added/updated unit tests for test_source.py

Author: jewettaijfc

tests/test_components/test_source.py

@@ -148,19 +148,20 @@ def test_dipole():
         _ = td.PointDipole(size=(1, 1, 1), source_time=g, center=(1, 2, 3), polarization="Ex")
 
 
-def test_FieldSource():
+@pytest.mark.parametrize("swap_axes", [True, False])
+def test_FieldSource(swap_axes):
     g = td.GaussianPulse(freq0=1e12, fwidth=0.1e12)
     mode_spec = td.ModeSpec(num_modes=2)
 
     # test we can make planewave
-    _ = td.PlaneWave(size=(0, td.inf, td.inf), source_time=g, pol_angle=np.pi / 2, direction="+")
-    # s.plot(y=0)
-    # plt.close()
+    s = td.PlaneWave(size=(0, td.inf, td.inf), source_time=g, pol_angle=np.pi / 2, direction="+")
+    s.plot(y=0, swap_axes=swap_axes)
+    plt.close()
 
     # test we can make gaussian beam
-    _ = td.GaussianBeam(size=(0, 1, 1), source_time=g, pol_angle=np.pi / 2, direction="+")
-    # s.plot(y=0)
-    # plt.close()
+    s = td.GaussianBeam(size=(0, 1, 1), source_time=g, pol_angle=np.pi / 2, direction="+")
+    s.plot(y=0, swap_axes=swap_axes)
+    plt.close()
 
     # test we can make an astigmatic gaussian beam
     _ = td.AstigmaticGaussianBeam(
@@ -173,11 +174,11 @@ def test_FieldSource():
     )
 
     # test we can make mode source
-    _ = td.ModeSource(
+    s = td.ModeSource(
         size=(0, 1, 1), direction="+", source_time=g, mode_spec=mode_spec, mode_index=0
     )
-    # s.plot(y=0)
-    # plt.close()
+    s.plot(y=0, swap_axes=swap_axes)
+    plt.close()
 
     # test that non-planar geometry crashes plane wave and gaussian beams
     with pytest.raises(pydantic.ValidationError):
@@ -203,10 +204,25 @@ def test_FieldSource():
     with pytest.raises(pydantic.ValidationError):
         _ = td.TFSF(size=(1, 1, 0), direction="+", source_time=g, injection_axis=2)
 
-    # s.plot(z=0)
+    # s.plot(z=0, swap_axes=swap_axes)
     # plt.close()
 
 
+@pytest.mark.parametrize("swap_axes", [True, False])
+def test_current_source(swap_axes):
+    L: float = 5.0
+    source = td.UniformCurrentSource(
+        center=(0, -L / 3, 0),
+        size=(L, 0, L / 2),
+        polarization="Ex",
+        source_time=td.GaussianPulse(
+            freq0=100e14,
+            fwidth=10e14,
+        ),
+    )
+    source.plot(z=0, swap_axes=swap_axes)
+
+
 def test_pol_arrow():
     g = td.GaussianPulse(freq0=1e12, fwidth=0.1e12)