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) {