test: pin generated floors for fixed seeds

src/generators/generator.py

@@ -318,13 +318,13 @@ def add_special_rooms(self, dead_ends: list) -> None:
 
     def place_super_secret_room(self) -> None:
         """
-        Places a super secret room on the flor.
+        Places a super secret room on the floor.
         The super secret room should be as close as possible to the boss room
         but should not be adjacent to the boss room.
-        There should only have one adjacent room.
+        It should only have one adjacent room.
         """
         floor = self._floor
-        distance_to_boss = 25
+        distance_to_boss = my_globals.MAX_SUPER_SECRET_ROOM_DISTANCE_SQUARED
         candidate = None
         boss_room = floor.get_boss_room()
         for room in floor.get_rooms():

src/utils/globals.py

@@ -63,6 +63,9 @@ class Color(Enum):
 FLOOR_WIDTH: int = 9
 FLOOR_HEIGHT: int = 8
 MAX_ROOMS: int = 15
+# The super secret room is only placed closer than this to the boss room
+# (squared distance, 25 means a distance below 5 rooms)
+MAX_SUPER_SECRET_ROOM_DISTANCE_SQUARED: int = 25
 # Distance between start and boss room
 X_OFFSET = (FLOOR_WIDTH * ROOM_WIDTH) / 2
 Y_OFFSET = 0

src/utils/util_functions.py

@@ -4,7 +4,6 @@
 
 import os
 import random
-import math
 from typing import Tuple
 
 import utils.globals as my_globals
@@ -37,14 +36,13 @@ def get_picture_for_room_type(room_type: RoomType):
 def calculate_room_amount(stage_id: int, rng: random.Random) -> int:
     """
     Calculates the room amount.
+    The generator may place a few more rooms than this,
+    because it adds all neighbours of a room in one step.
     @param stage_id: id for the floor
     @param rng: random number generator to use
     :return: room amount
     """
-    return min(
-        my_globals.MAX_ROOMS,
-        int(rng.randint(0, 1) + 5 + math.floor(stage_id * 10) / 3.0),
-    )
+    return min(my_globals.MAX_ROOMS, rng.randint(0, 1) + 5 + stage_id * 10 // 3)
 
 
 def place_room(rng: random.Random) -> bool: