updated unit tests for test_heat.py and test_heat_charge.py

Author: jewettaijfc

tests/test_components/test_heat.py

@@ -335,11 +335,12 @@ def make_heat_sim(include_custom_source: bool = True):
     return heat_sim
 
 
-def test_heat_sim():
+@pytest.mark.parametrize("swap_axes", [True, False])
+def test_heat_sim(swap_axes):
     bc_temp, bc_flux, bc_conv = make_heat_bcs()
     heat_sim = make_heat_sim()
 
-    _ = heat_sim.plot(x=0)
+    _ = heat_sim.plot(x=0, swap_axes=swap_axes)
 
     # wrong names given
     for pl in [
@@ -373,14 +374,14 @@ def test_heat_sim():
     with pytest.raises(pd.ValidationError):
         heat_sim.updated_copy(monitors=[temp_mnt, temp_mnt])
 
-    _ = heat_sim.plot(x=0)
+    _ = heat_sim.plot(x=0, swap_axes=swap_axes)
     plt.close()
 
-    _ = heat_sim.plot_heat_conductivity(y=0)
+    _ = heat_sim.plot_heat_conductivity(y=0, swap_axes=swap_axes)
     plt.close()
 
     heat_sim_sym = heat_sim.updated_copy(symmetry=(0, 1, 1))
-    _ = heat_sim_sym.plot_heat_conductivity(z=0, colorbar="source")
+    _ = heat_sim_sym.plot_heat_conductivity(z=0, colorbar="source", swap_axes=swap_axes)
     plt.close()
 
     # no negative symmetry

tests/test_components/test_heat_charge.py

@@ -852,7 +852,8 @@ def test_heat_charge_sources(structures):
         _ = td.HeatSource(structures=["solid_structure"], rate="100")
 
 
-def test_heat_charge_simulation(simulation_data):
+@pytest.mark.parametrize("swap_axes", [True, False])
+def test_heat_charge_simulation(swap_axes, simulation_data):
     """Tests 'HeatChargeSimulation' and 'ConductionSimulation' objects."""
     heat_sim_data, cond_sim_data, voltage_capacitance_sim_data, current_voltage_simulation_data = (
         simulation_data
@@ -876,6 +877,18 @@ def test_heat_charge_simulation(simulation_data):
         "Current-Voltage simulation should be created successfully."
     )
 
+    _ = heat_sim.plot_heat_conductivity(x=0, swap_axes=swap_axes)
+    plt.close()
+    _ = heat_sim.plot_property(x=0, property="heat_conductivity", swap_axes=swap_axes)
+    plt.close()
+    _ = cond_sim.plot_property(x=0, property="electric_conductivity", swap_axes=swap_axes)
+    plt.close()
+    for sim in (heat_sim, cond_sim, voltage_capacitance_sim, current_voltage_sim):
+        _ = sim.plot_boundaries(x=0, swap_axes=swap_axes)
+        plt.close()
+        _ = sim.plot_sources(x=0, swap_axes=swap_axes)
+        plt.close()
+
 
 def test_sim_data_plotting(simulation_data):
     """Tests whether simulation data can be plotted and appropriate errors are raised."""
@@ -1433,7 +1446,8 @@ def test_plotting_functions(simulation_data):
         heat_sim_data.plot_field("test", invalid_param=0)
 
 
-def test_bandgap_monitor():
+@pytest.mark.parametrize("swap_axes", [True, False])
+def test_bandgap_monitor(swap_axes):
     """Test energy bandgap monitor ploting function."""
     # create a triangle grid
     tri_grid_points = td.PointDataArray(
@@ -1573,8 +1587,8 @@ def test_bandgap_monitor():
 
     # test check for the voltage value in the list of arguments
 
-    tri_single_voltage_data.plot(x=0.0)
-    tri_multi_voltage_data.plot(x=0.0, voltage=1.0)
+    tri_single_voltage_data.plot(x=0.0, swap_axes=swap_axes)
+    tri_multi_voltage_data.plot(x=0.0, voltage=1.0, swap_axes=swap_axes)
 
     with pytest.raises(DataError):
         tri_multi_voltage_data.plot(x=0.0)