Add custom stages

1 files changed, 1 insertions, 18 deletions

diff --git a/MapTests/StageTests.swift b/MapTests/StageTests.swift
index 05812fe..97502a8 100644
--- a/MapTests/StageTests.swift
+++ b/MapTests/StageTests.swift
@@ -34,8 +34,7 @@ struct StageTests {
     #expect(StageType.properties.contains(.market))
     #expect(StageType.properties.contains(.efficiency))
 
-    #expect(StageType.custom.count == 1)
-    #expect(StageType.custom.contains(.behavior))
+    #expect(StageType.customStages.count >= 0)
   }
 
   @Test func testStageTypeIdentifiable() async throws {
@@ -99,21 +98,5 @@ struct StageTests {
     #expect(title == "Behavior")
   }
 
-  @Test func testAllStageTypesHaveTitles() async throws {
-    for stageType in StageType.allCases {
-      let title = Stage.title(stageType)
-      #expect(!title.isEmpty)
-    }
-  }
-
-  @Test func testAllStageTypesHaveStages() async throws {
-    for stageType in StageType.allCases {
-      let stage = Stage.stages(stageType)
-      #expect(!stage.i.isEmpty)
-      #expect(!stage.ii.isEmpty)
-      #expect(!stage.iii.isEmpty)
-      #expect(!stage.iv.isEmpty)
-    }
-  }
 
 }