Skip balancer sizes that are too small to fit required splitters

factorio_sat/calculate_optimal.py

@@ -133,7 +133,8 @@ def next_size(self, store: NetworkSolutionStore, underground_length: int) -> Opt
             if existence:
                 return None
 
-            if existence is None:
+            interior_area = (width - 2) * height
+            if interior_area >= len(store.network) * 2 and existence is None:
                 return size
             width += 1
 
@@ -151,6 +152,10 @@ def next_size(self, store: NetworkSolutionStore, underground_length: int) -> Opt
                 if height < min_height or height > 2 * min_height:
                     continue
 
+                interior_area = width * height
+                if interior_area < len(store.network) * 2:
+                    continue
+
                 size = underground_length, width + 2, height
                 existence = store.does_balancer_exist(size)
                 if existence:

test/test_end_to_end.py

@@ -117,7 +117,7 @@ def tokenise(lines: List[str]) -> List[Token]:
     for row, line in line_iter:
         if not line.startswith(TEST_EDGE_START):
             if len(line) > 0 and line[0] in TEST_EDGES:
-                raise RuntimeError(f'Invalid line start character "{line[0]}" at line {row+1}')
+                raise RuntimeError(f'Invalid line start character "{line[0]}" at line {row + 1}')
 
             match = block_start.match(line)
             if match is not None:
@@ -143,7 +143,7 @@ def tokenise(lines: List[str]) -> List[Token]:
                 if not line.startswith(TEST_EDGE_MIDDLE):
                     break
             if not line.startswith(TEST_EDGE_STOP):
-                raise RuntimeError(f'Invalid test instance format at line {row+1}')
+                raise RuntimeError(f'Invalid test instance format at line {row + 1}')
             tokenised.append(Token('test', row + 1, stringifier.decode(test_case)))
     return tokenised