[OC2] Fix "Moon 1-5" never appearing in level pool (#1366)

This commit is contained in:
toasterparty
2023-01-04 06:21:52 -08:00
committed by GitHub
parent fe2b431821
commit 92d1ed60c6
3 changed files with 34 additions and 10 deletions

View File

@@ -147,3 +147,26 @@ class Overcooked2Test(unittest.TestCase):
for item in ITEMS_TO_EXCLUDE_IF_NO_DLC:
self.assertIn(item, item_table.keys())
def testLevelCounts(self):
for dlc in Overcooked2Dlc:
level_id_range = range(dlc.start_level_id, dlc.end_level_id)
for level_id in dlc.excluded_levels():
self.assertIn(level_id, level_id_range, f"Excluded level {dlc.name} - {level_id} out of range")
for level_id in dlc.horde_levels():
self.assertIn(level_id, level_id_range, f"Horde level {dlc.name} - {level_id} out of range")
for level_id in dlc.prep_levels():
self.assertIn(level_id, level_id_range, f"Prep level {dlc.name} - {level_id} out of range")
for level_id in level_id_range:
self.assertIn((dlc, level_id), level_id_to_shortname, "A valid level is not represented in level directory")
count = 0
for (dlc_key, _) in level_id_to_shortname:
if dlc == dlc_key:
count += 1
self.assertEqual(count, len(level_id_range), f"Number of levels in {dlc.name} has discrepancy between level_id range and directory")