Fix tests

Author: BurnySc2

test/generate_pickle_files_bot.py

@@ -76,27 +76,63 @@ async def on_start_async(self):
 def main():
 
     maps = [
+        "16-BitLE.SC2Map",
+        "AbiogenesisLE",
+        "AbyssalReefLE",
+        "AcidPlantLE",
+        "AcolyteLE",
         "Acropolis",
         "Artana",
-        "CrystalCavern",
-        "DigitalFrontier",
-        "OldSunshine",
-        "Treachery",
-        "Triton",
+        "AscensiontoAiurLE",
         "AutomatonLE",
+        "BackwaterLE",
+        "Bandwidth",
+        "BattleontheBoardwalkLE",
+        "BelShirVestigeLE",
+        "BlackpinkLE",
+        "BloodBoilLE",
         "BlueshiftLE",
+        "CactusValleyLE",
+        "CatalystLE",
         "CeruleanFallLE",
+        "CrystalCavern",
+        "CyberForestLE",
         "DarknessSanctuaryLE",
+        "DefendersLandingLE",
+        "DigitalFrontier",
+        "DreamcatcherLE",
+        "EastwatchLE",
+        "Ephemeron",
+        "FractureLE",
+        "FrostLE",
+        "HonorgroundsLE",
+        "InterloperLE",
+        "KairosJunctionLE",
         "KairosJunctionLE",
+        "KingsCoveLE",
+        "LostandFoundLE",
+        "MechDepotLE",
+        "NeonVioletSquareLE",
+        "NewkirkPrecinctTE",
+        "NewRepugnancyLE",
+        "OdysseyLE",
+        "OldSunshine",
+        "PaladinoTerminalLE",
+        "ParaSiteLE",
         "ParaSiteLE",
         "PortAleksanderLE",
-        "Bandwidth",
-        "Ephemeron",
         "PrimusQ9",
+        "ProximaStationLE",
+        "RedshiftLE",
         "Reminiscence",
         "Sanglune",
+        "SequencerLE",
+        # "StasisLE", Commented out because it has uneven number of expansions
         "TheTimelessVoid",
+        "Treachery",
+        "Triton",
         "Urzagol",
+        "YearZeroLE",
     ]
 
     for map in maps:
@@ -106,7 +142,6 @@ def main():
                 sc2.maps.get(map),
                 [Bot(Race.Terran, ExporterBot()), Computer(Race.Zerg, Difficulty.Easy)],
                 realtime=False,
-                save_replay_as="Example.SC2Replay",
             )
         except ProtocolError:
             # ProtocolError appears after a leave game request

test/test_pickled_data.py

@@ -565,10 +565,10 @@ def test_position_pointlike(self, bot: BotAI, x1, y1, x2, y2, x3, y3):
             dist_furthest_point = pos1._distance_squared(furthest_point) ** 0.5
 
             # Distances between pos1-pos2 and pos1-pos3 might be the same, so the sorting might still be different, that's why I use a set here
-            assert pos1.closest(points) in {p for p in points2 if pos1.distance_to(p) == dist_closest_point}
-            assert pos1.distance_to_closest(points) == pos1._distance_squared(closest_point) ** 0.5
-            assert pos1.furthest(points) in {p for p in points2 if pos1.distance_to(p) == dist_furthest_point}
-            assert pos1.distance_to_furthest(points) == pos1._distance_squared(furthest_point) ** 0.5
+            assert pos1.closest(points) in {p for p in points2 if abs(pos1.distance_to(p) - dist_closest_point) < epsilon}
+            assert abs(pos1.distance_to_closest(points) - pos1._distance_squared(closest_point) ** 0.5) < epsilon
+            assert pos1.furthest(points) in {p for p in points2 if abs(pos1.distance_to(p) - dist_furthest_point) < epsilon}
+            assert abs(pos1.distance_to_furthest(points) - pos1._distance_squared(furthest_point) ** 0.5) < epsilon
             assert pos1.offset(pos2) == Point2((pos1.x + pos2.x, pos1.y + pos2.y))
             if pos1 != pos2:
                 assert pos1.unit_axes_towards(pos2) != Point2((0, 0))