[CC] Updating graph title for publication

Author: BonneelP

Methods/Machine/MachineUD/plot.py

@@ -71,7 +71,10 @@ def plot(
         raise ValueError("Comp_machine is not available for MachineUD")
     axes.set_xlabel("(m)")
     axes.set_ylabel("(m)")
-    axes.set_title("Machine")
+    if self.name in [None, ""]:
+        axes.set_title("Machine")
+    else:
+        axes.set_title(self.name)
 
     # Axis Setup
     axis("equal")

Tests/Plot/test_ICEM_2020.py

@@ -199,6 +199,7 @@ def test_MachineUD(self):
         """Figure 12: Check that you can plot a machine with 4 laminations
         """
         machine = MachineUD()
+        machine.name = "Machine with 4 laminations"
 
         # Main geometry parameter
         Rext = 170e-3  # Exterior radius of outter lamination
@@ -303,6 +304,7 @@ def test_SlotUD(self):
         # Enfore first point on rotor bore
         Rrotor = abs(0.205917893677990 - 0.107339745962156j)
         machine = MachineSRM()
+        machine.name = "User-Defined Slot"
         # Stator definintion
         machine.stator = LamSlotWind(
             Rint=Rrotor + 5e-3, Rext=Rrotor + 120e-3, is_internal=False, is_stator=True
@@ -354,6 +356,7 @@ def test_WindingUD(self):
         """
         plt.close("all")
         machine = MachineDFIM()
+        machine.name = "User Defined Winding"
         # Rotor definition
         machine.rotor = LamSlotWind(
             Rint=0.2,

Tests/Plot/test_Lam_Mag_surface_plot.py

@@ -51,27 +51,11 @@ def test_Lam_Mag_10_surface(self):
         )
         rotor.mat_type.mag = MatMagnetics(Wlam=0.5e-3)
 
-        stator = Lamination(
-            Rint=230e-3,
-            Rext=0.3,
-            is_internal=False,
-            is_stator=True,
-            L1=0.5,
-            Nrvd=0,
-            Wrvd=0.05,
-        )
-        stator.mat_type.mag = MatMagnetics(Wlam=0.5e-3)
-
         rotor.plot()
         fig = plt.gcf()
         self.assertEqual(len(fig.axes[0].patches), 10)
         fig.savefig(join(save_path, "test_Lam_Mag_10s_2-Rotor.png"))
 
-        stator.plot()
-        fig = plt.gcf()
-        self.assertEqual(len(fig.axes[0].patches), 2)
-        fig.savefig(join(save_path, "test_Lam_Mag_10s_3-Stator.png"))
-
         magnet2 = [MagnetType10(Lmag=0.5, Hmag=0.02, Wmag=0.04)]
         rotor.slot = SlotMFlat(Zs=8, W0=0.04, W0_is_rad=False, magnet=magnet2)
         rotor.plot()