(Aegis) assert the existence of specific yeti-booster

ahyangyi · Nov 3, 2023 · 678ed4e · 678ed4e
1 parent ccb890c
commit 678ed4e
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/industry/lib/validator/yeti.py b/industry/lib/validator/yeti.py
@@ -1,5 +1,6 @@
 from industry.lib.economy import WorkerYard
 from industry.industries import worker_yard
+from industry.cargos import passengers, mail, tired_workers
 
 
 def check_yeti(economy):
@@ -13,3 +14,9 @@ def check_yeti(economy):
             assert len(conf.boosters) == 2
         else:
             assert len(conf.boosters) == 3
+            if economy.parameters["WORKFORCE"] == "YETI_PASSENGERS":
+                assert conf.boosters[2] == passengers
+            if economy.parameters["WORKFORCE"] == "YETI_MAIL":
+                assert conf.boosters[2] == mail
+            if economy.parameters["WORKFORCE"] == "YETI_TIRED":
+                assert conf.boosters[2] == tired_workers