updated unit tests for test_eme.py

Author: jewettaijfc

tests/test_components/test_eme.py

@@ -282,22 +282,22 @@ def test_eme_monitor():
         center=(1, 2, 3), size=(2, 2, 2), freqs=[300e12], num_modes=2, name="eme_coeffs"
     )
 
-
-def test_eme_simulation():
+@pytest.mark.parametrize("transpose", [True, False])
+def test_eme_simulation(transpose):
     sim = make_eme_sim()
-    _ = sim.plot(x=0, ax=AX)
-    _ = sim.plot(y=0, ax=AX)
-    _ = sim.plot(z=0, ax=AX)
-    _ = sim.plot_grid(x=0, ax=AX)
-    _ = sim.plot_grid(y=0, ax=AX)
-    _ = sim.plot_grid(z=0, ax=AX)
-    _ = sim.plot_eps(x=0, ax=AX)
-    _ = sim.plot_eps(y=0, ax=AX)
-    _ = sim.plot_eps(z=0, ax=AX)
+    _ = sim.plot(x=0, ax=AX, transpose=transpose)
+    _ = sim.plot(y=0, ax=AX, transpose=transpose)
+    _ = sim.plot(z=0, ax=AX, transpose=transpose)
+    _ = sim.plot_grid(x=0, ax=AX, transpose=transpose)
+    _ = sim.plot_grid(y=0, ax=AX, transpose=transpose)
+    _ = sim.plot_grid(z=0, ax=AX, transpose=transpose)
+    _ = sim.plot_eps(x=0, ax=AX, transpose=transpose)
+    _ = sim.plot_eps(y=0, ax=AX, transpose=transpose)
+    _ = sim.plot_eps(z=0, ax=AX, transpose=transpose)
     sim2 = sim.updated_copy(axis=1)
-    _ = sim2.plot(x=0, ax=AX)
-    _ = sim2.plot(y=0, ax=AX)
-    _ = sim2.plot(z=0, ax=AX)
+    _ = sim2.plot(x=0, ax=AX, transpose=transpose)
+    _ = sim2.plot(y=0, ax=AX, transpose=transpose)
+    _ = sim2.plot(z=0, ax=AX, transpose=transpose)
 
     # need at least one freq
     with pytest.raises(pd.ValidationError):

tidy3d/components/mode/mode_solver.py

@@ -344,6 +344,8 @@ def _get_solver_grid(
             represent the region where custom medium data is needed for proper subpixel.
         truncate_symmetry : bool = True
             Truncate to symmetry quadrant if symmetry present.
+        transpose : bool = False
+            Swap the coordinates in the plane. (This overrides the default lexicographic axis order).
 
         Returns
         -------
@@ -2106,6 +2108,8 @@ def plot(
         ----------
         ax : matplotlib.axes._subplots.Axes = None
             Matplotlib axes to plot on, if not specified, one is created.
+        transpose : bool = False
+            Swap horizontal and vertical axes. (This overrides the default lexicographic axis order).
 
         Returns
         -------
@@ -2160,6 +2164,8 @@ def plot_eps(
             Defaults to the structure default alpha.
         ax : matplotlib.axes._subplots.Axes = None
             Matplotlib axes to plot on, if not specified, one is created.
+        transpose : bool = False
+            Swap horizontal and vertical axes. (This overrides the default lexicographic axis order).
 
         Returns
         -------
@@ -2223,6 +2229,8 @@ def plot_structures_eps(
             If ``True``, it is plotteed in white (suitable for black backgrounds).
         ax : matplotlib.axes._subplots.Axes = None
             Matplotlib axes to plot on, if not specified, one is created.
+        transpose : bool = False
+            Swap horizontal and vertical axes. (This overrides the default lexicographic axis order).
 
         Returns
         -------
@@ -2270,6 +2278,8 @@ def plot_grid(
         ----------
         ax : matplotlib.axes._subplots.Axes = None
             Matplotlib axes to plot on, if not specified, one is created.
+        transpose : bool = False
+            Swap horizontal and vertical axes. (This overrides the default lexicographic axis order).
         **kwargs
             Optional keyword arguments passed to the matplotlib ``LineCollection``.
             For details on accepted values, refer to
@@ -2419,6 +2429,8 @@ def plot_pml(
         ----------
         ax : matplotlib.axes._subplots.Axes = None
             Matplotlib axes to plot on, if not specified, one is created.
+        transpose : bool = False
+            Swap horizontal and vertical axes. (This overrides the default lexicographic axis order).
 
         Returns
         -------

tidy3d/components/tcad/simulation/heat.py

@@ -98,6 +98,8 @@ def plot_heat_conductivity(
             The x range if plotting on xy or xz planes, y range if plotting on yz plane.
         vlim : Tuple[float, float] = None
             The z range if plotting on xz or yz planes, y plane if plotting on xy plane.
+        transpose : bool = False
+            Swap horizontal and vertical axes. (This overrides the default lexicographic axis order).
 
         Returns
         -------

tidy3d/components/tcad/simulation/heat_charge.py

@@ -993,6 +993,8 @@ def plot_property(
             The x range if plotting on xy or xz planes, y range if plotting on yz plane.
         vlim : Tuple[float, float] = None
             The z range if plotting on xz or yz planes, y plane if plotting on xy plane.
+        transpose : bool = False
+            Swap horizontal and vertical axes. (This overrides the default lexicographic axis order).
 
         Returns
         -------
@@ -1097,6 +1099,8 @@ def plot_heat_conductivity(
             The x range if plotting on xy or xz planes, y range if plotting on yz plane.
         vlim : Tuple[float, float] = None
             The z range if plotting on xz or yz planes, y plane if plotting on xy plane.
+        transpose : bool = False
+            Swap horizontal and vertical axes. (This overrides the default lexicographic axis order).
 
         Returns
         -------
@@ -1156,6 +1160,8 @@ def plot_boundaries(
             Options are ["heat_conductivity", "electric_conductivity"]
         ax : matplotlib.axes._subplots.Axes = None
             Matplotlib axes to plot on, if not specified, one is created.
+        transpose : bool = False
+            Swap horizontal and vertical axes. (This overrides the default lexicographic axis order).
 
         Returns
         -------
@@ -1466,6 +1472,8 @@ def _construct_heat_charge_boundaries(
             target plane.
         boundary_spec : List[HeatBoundarySpec]
             list of boundary conditions associated with structures.
+        transpose : bool = False
+            Swap the boundary-box coordinates in the plane. (This overrides the default lexicographic axis order).
 
         Returns
         -------
@@ -1550,6 +1558,8 @@ def plot_sources(
             Opacity of the sources, If ``None`` uses Tidy3d default.
         ax : matplotlib.axes._subplots.Axes = None
             Matplotlib axes to plot on, if not specified, one is created.
+        transpose : bool = False
+            Swap horizontal and vertical axes. (This overrides the default lexicographic axis order).
 
         Returns
         -------