fix tests

MegaMek · Sep 24, 2024 · c9f257a · c9f257a
1 parent ce80e92
commit c9f257a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/megamek/src/megamek/common/Entity.java b/megamek/src/megamek/common/Entity.java
@@ -4834,6 +4834,8 @@ public int getHeatCapacity() {
      * @return The formatted heat capacity
      */
     public String formatHeat() {
+        // This method might make sense as an abstract method with overrides in Mek and Aero,
+        // But since the implementation would be the same in both classes, this way avoids code duplication.
         int sinks;
         if (this instanceof Mek m) {
             sinks = m.getActiveSinks();

diff --git a/megamek/unittests/megamek/common/EntityTest.java b/megamek/unittests/megamek/common/EntityTest.java
@@ -122,7 +122,7 @@ void testFormatHeat() {
             f = new File("testresources/megamek/common/units/Sagittaire SGT-14D.mtf");
             mfp = new MekFileParser(f);
             e = mfp.getEntity();
-            expectedHeat = "28 [42]";
+            expectedHeat = "28, 42 with RHS";
             computedHeat = e.formatHeat();
             assertEquals(expectedHeat, computedHeat);
         } catch (Exception ex) {