Co-authored-by: Alchav <59858495+Alchav@users.noreply.github.com> Co-authored-by: Aaron Wagener <mmmcheese158@gmail.com> Co-authored-by: Doug Hoskisson <beauxq@yahoo.com> Co-authored-by: Fabian Dill <Berserker66@users.noreply.github.com>
		
			
				
	
	
		
			246 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			246 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
from typing import List, Tuple
 | 
						|
from . import KDL3TestBase
 | 
						|
from ..Room import KDL3Room
 | 
						|
 | 
						|
 | 
						|
class TestCopyAbilityShuffle(KDL3TestBase):
 | 
						|
    options = {
 | 
						|
        "open_world": False,
 | 
						|
        "goal_speed": "normal",
 | 
						|
        "total_heart_stars": 30,
 | 
						|
        "heart_stars_required": 50,
 | 
						|
        "filler_percentage": 0,
 | 
						|
        "copy_ability_randomization": "enabled",
 | 
						|
    }
 | 
						|
 | 
						|
    def test_goal(self):
 | 
						|
        self.assertBeatable(False)
 | 
						|
        heart_stars = self.get_items_by_name("Heart Star")
 | 
						|
        self.collect(heart_stars[0:14])
 | 
						|
        self.assertEqual(self.count("Heart Star"), 14, str(self.multiworld.seed))
 | 
						|
        self.assertBeatable(False)
 | 
						|
        self.collect(heart_stars[14:15])
 | 
						|
        self.assertEqual(self.count("Heart Star"), 15, str(self.multiworld.seed))
 | 
						|
        self.assertBeatable(False)
 | 
						|
        self.collect_by_name(["Burning", "Cutter", "Kine"])
 | 
						|
        self.assertBeatable(True)
 | 
						|
        self.remove([self.get_item_by_name("Love-Love Rod")])
 | 
						|
        self.collect(heart_stars)
 | 
						|
        self.assertEqual(self.count("Heart Star"), 30, str(self.multiworld.seed))
 | 
						|
        self.assertBeatable(True)
 | 
						|
 | 
						|
    def test_kine(self):
 | 
						|
        self.collect_by_name(["Cutter", "Burning", "Heart Star"])
 | 
						|
        self.assertBeatable(False)
 | 
						|
 | 
						|
    def test_cutter(self):
 | 
						|
        self.collect_by_name(["Kine", "Burning", "Heart Star"])
 | 
						|
        self.assertBeatable(False)
 | 
						|
 | 
						|
    def test_burning(self):
 | 
						|
        self.collect_by_name(["Cutter", "Kine", "Heart Star"])
 | 
						|
        self.assertBeatable(False)
 | 
						|
 | 
						|
    def test_cutter_and_burning_reachable(self):
 | 
						|
        rooms = self.multiworld.worlds[1].rooms
 | 
						|
        copy_abilities = self.multiworld.worlds[1].copy_abilities
 | 
						|
        sand_canyon_5 = self.multiworld.get_region("Sand Canyon 5 - 9", 1)
 | 
						|
        assert isinstance(sand_canyon_5, KDL3Room)
 | 
						|
        valid_rooms = [room for room in rooms if (room.level < sand_canyon_5.level)
 | 
						|
                       or (room.level == sand_canyon_5.level and room.stage < sand_canyon_5.stage)]
 | 
						|
        for room in valid_rooms:
 | 
						|
            if any(copy_abilities[enemy] == "Cutter Ability" for enemy in room.enemies):
 | 
						|
                break
 | 
						|
        else:
 | 
						|
            self.fail("Could not reach Cutter Ability before Sand Canyon 5!")
 | 
						|
        iceberg_4 = self.multiworld.get_region("Iceberg 4 - 7", 1)
 | 
						|
        assert isinstance(iceberg_4, KDL3Room)
 | 
						|
        valid_rooms = [room for room in rooms if (room.level < iceberg_4.level)
 | 
						|
                       or (room.level == iceberg_4.level and room.stage < iceberg_4.stage)]
 | 
						|
        for room in valid_rooms:
 | 
						|
            if any(copy_abilities[enemy] == "Burning Ability" for enemy in room.enemies):
 | 
						|
                break
 | 
						|
        else:
 | 
						|
            self.fail("Could not reach Burning Ability before Iceberg 4!")
 | 
						|
 | 
						|
    def test_valid_abilities_for_ROB(self):
 | 
						|
        # there exists a subset of 4-7 abilities that will allow us access to ROB heart star on default settings
 | 
						|
        self.collect_by_name(["Heart Star", "Kine", "Coo"])  # we will guaranteed need Coo, Kine, and Heart Stars to reach
 | 
						|
        # first we need to identify our bukiset requirements
 | 
						|
        groups = [
 | 
						|
            ({"Parasol Ability", "Cutter Ability"}, {'Bukiset (Parasol)', 'Bukiset (Cutter)'}),
 | 
						|
            ({"Spark Ability", "Clean Ability"}, {'Bukiset (Spark)', 'Bukiset (Clean)'}),
 | 
						|
            ({"Ice Ability", "Needle Ability"}, {'Bukiset (Ice)', 'Bukiset (Needle)'}),
 | 
						|
            ({"Stone Ability", "Burning Ability"}, {'Bukiset (Stone)', 'Bukiset (Burning)'}),
 | 
						|
        ]
 | 
						|
        copy_abilities = self.multiworld.worlds[1].copy_abilities
 | 
						|
        required_abilities: List[Tuple[str]] = []
 | 
						|
        for abilities, bukisets in groups:
 | 
						|
            potential_abilities: List[str] = list()
 | 
						|
            for bukiset in bukisets:
 | 
						|
                if copy_abilities[bukiset] in abilities:
 | 
						|
                    potential_abilities.append(copy_abilities[bukiset])
 | 
						|
            required_abilities.append(tuple(potential_abilities))
 | 
						|
        collected_abilities = list()
 | 
						|
        for group in required_abilities:
 | 
						|
            self.assertFalse(len(group) == 0, str(self.multiworld.seed))
 | 
						|
            collected_abilities.append(group[0])
 | 
						|
        self.collect_by_name([ability.replace(" Ability", "") for ability in collected_abilities])
 | 
						|
        if "Parasol Ability" not in collected_abilities or "Stone Ability" not in collected_abilities:
 | 
						|
            # required for non-Bukiset related portions
 | 
						|
            self.collect_by_name(["Parasol", "Stone"])
 | 
						|
 | 
						|
        if "Cutter Ability" not in collected_abilities:
 | 
						|
            # we can't actually reach 3-6 without Cutter
 | 
						|
            self.assertFalse(self.can_reach_location("Sand Canyon 6 - Professor Hector & R.O.B"), str(self.multiworld.seed))
 | 
						|
            self.collect_by_name(["Cutter"])
 | 
						|
 | 
						|
        self.assertTrue(self.can_reach_location("Sand Canyon 6 - Professor Hector & R.O.B"),
 | 
						|
                        ''.join(str(self.multiworld.seed)).join(collected_abilities))
 | 
						|
 | 
						|
 | 
						|
class TestAnimalShuffle(KDL3TestBase):
 | 
						|
    options = {
 | 
						|
        "open_world": False,
 | 
						|
        "goal_speed": "normal",
 | 
						|
        "total_heart_stars": 30,
 | 
						|
        "heart_stars_required": 50,
 | 
						|
        "filler_percentage": 0,
 | 
						|
        "animal_randomization": "full",
 | 
						|
    }
 | 
						|
 | 
						|
    def test_goal(self):
 | 
						|
        self.assertBeatable(False)
 | 
						|
        heart_stars = self.get_items_by_name("Heart Star")
 | 
						|
        self.collect(heart_stars[0:14])
 | 
						|
        self.assertEqual(self.count("Heart Star"), 14, str(self.multiworld.seed))
 | 
						|
        self.assertBeatable(False)
 | 
						|
        self.collect(heart_stars[14:15])
 | 
						|
        self.assertEqual(self.count("Heart Star"), 15, str(self.multiworld.seed))
 | 
						|
        self.assertBeatable(False)
 | 
						|
        self.collect_by_name(["Burning", "Cutter", "Kine"])
 | 
						|
        self.assertBeatable(True)
 | 
						|
        self.remove([self.get_item_by_name("Love-Love Rod")])
 | 
						|
        self.collect(heart_stars)
 | 
						|
        self.assertEqual(self.count("Heart Star"), 30, str(self.multiworld.seed))
 | 
						|
        self.assertBeatable(True)
 | 
						|
 | 
						|
    def test_kine(self):
 | 
						|
        self.collect_by_name(["Cutter", "Burning", "Heart Star"])
 | 
						|
        self.assertBeatable(False)
 | 
						|
 | 
						|
    def test_cutter(self):
 | 
						|
        self.collect_by_name(["Kine", "Burning", "Heart Star"])
 | 
						|
        self.assertBeatable(False)
 | 
						|
 | 
						|
    def test_burning(self):
 | 
						|
        self.collect_by_name(["Cutter", "Kine", "Heart Star"])
 | 
						|
        self.assertBeatable(False)
 | 
						|
 | 
						|
    def test_locked_animals(self):
 | 
						|
        self.assertTrue(self.multiworld.get_location("Ripple Field 5 - Animal 2", 1).item.name == "Pitch Spawn")
 | 
						|
        self.assertTrue(self.multiworld.get_location("Iceberg 4 - Animal 1", 1).item.name == "ChuChu Spawn")
 | 
						|
        self.assertTrue(self.multiworld.get_location("Sand Canyon 6 - Animal 1", 1).item.name in {"Kine Spawn", "Coo Spawn"})
 | 
						|
 | 
						|
 | 
						|
class TestAllShuffle(KDL3TestBase):
 | 
						|
    options = {
 | 
						|
        "open_world": False,
 | 
						|
        "goal_speed": "normal",
 | 
						|
        "total_heart_stars": 30,
 | 
						|
        "heart_stars_required": 50,
 | 
						|
        "filler_percentage": 0,
 | 
						|
        "animal_randomization": "full",
 | 
						|
        "copy_ability_randomization": "enabled",
 | 
						|
    }
 | 
						|
 | 
						|
    def test_goal(self):
 | 
						|
        self.assertBeatable(False)
 | 
						|
        heart_stars = self.get_items_by_name("Heart Star")
 | 
						|
        self.collect(heart_stars[0:14])
 | 
						|
        self.assertEqual(self.count("Heart Star"), 14, str(self.multiworld.seed))
 | 
						|
        self.assertBeatable(False)
 | 
						|
        self.collect(heart_stars[14:15])
 | 
						|
        self.assertEqual(self.count("Heart Star"), 15, str(self.multiworld.seed))
 | 
						|
        self.assertBeatable(False)
 | 
						|
        self.collect_by_name(["Burning", "Cutter", "Kine"])
 | 
						|
        self.assertBeatable(True)
 | 
						|
        self.remove([self.get_item_by_name("Love-Love Rod")])
 | 
						|
        self.collect(heart_stars)
 | 
						|
        self.assertEqual(self.count("Heart Star"), 30, str(self.multiworld.seed))
 | 
						|
        self.assertBeatable(True)
 | 
						|
 | 
						|
    def test_kine(self):
 | 
						|
        self.collect_by_name(["Cutter", "Burning", "Heart Star"])
 | 
						|
        self.assertBeatable(False)
 | 
						|
 | 
						|
    def test_cutter(self):
 | 
						|
        self.collect_by_name(["Kine", "Burning", "Heart Star"])
 | 
						|
        self.assertBeatable(False)
 | 
						|
 | 
						|
    def test_burning(self):
 | 
						|
        self.collect_by_name(["Cutter", "Kine", "Heart Star"])
 | 
						|
        self.assertBeatable(False)
 | 
						|
 | 
						|
    def test_locked_animals(self):
 | 
						|
        self.assertTrue(self.multiworld.get_location("Ripple Field 5 - Animal 2", 1).item.name == "Pitch Spawn")
 | 
						|
        self.assertTrue(self.multiworld.get_location("Iceberg 4 - Animal 1", 1).item.name == "ChuChu Spawn")
 | 
						|
        self.assertTrue(self.multiworld.get_location("Sand Canyon 6 - Animal 1", 1).item.name in {"Kine Spawn", "Coo Spawn"})
 | 
						|
 | 
						|
    def test_cutter_and_burning_reachable(self):
 | 
						|
        rooms = self.multiworld.worlds[1].rooms
 | 
						|
        copy_abilities = self.multiworld.worlds[1].copy_abilities
 | 
						|
        sand_canyon_5 = self.multiworld.get_region("Sand Canyon 5 - 9", 1)
 | 
						|
        assert isinstance(sand_canyon_5, KDL3Room)
 | 
						|
        valid_rooms = [room for room in rooms if (room.level < sand_canyon_5.level)
 | 
						|
                       or (room.level == sand_canyon_5.level and room.stage < sand_canyon_5.stage)]
 | 
						|
        for room in valid_rooms:
 | 
						|
            if any(copy_abilities[enemy] == "Cutter Ability" for enemy in room.enemies):
 | 
						|
                break
 | 
						|
        else:
 | 
						|
            self.fail("Could not reach Cutter Ability before Sand Canyon 5!")
 | 
						|
        iceberg_4 = self.multiworld.get_region("Iceberg 4 - 7", 1)
 | 
						|
        assert isinstance(iceberg_4, KDL3Room)
 | 
						|
        valid_rooms = [room for room in rooms if (room.level < iceberg_4.level)
 | 
						|
                       or (room.level == iceberg_4.level and room.stage < iceberg_4.stage)]
 | 
						|
        for room in valid_rooms:
 | 
						|
            if any(copy_abilities[enemy] == "Burning Ability" for enemy in room.enemies):
 | 
						|
                break
 | 
						|
        else:
 | 
						|
            self.fail("Could not reach Burning Ability before Iceberg 4!")
 | 
						|
 | 
						|
    def test_valid_abilities_for_ROB(self):
 | 
						|
        # there exists a subset of 4-7 abilities that will allow us access to ROB heart star on default settings
 | 
						|
        self.collect_by_name(["Heart Star", "Kine", "Coo"])  # we will guaranteed need Coo, Kine, and Heart Stars to reach
 | 
						|
        # first we need to identify our bukiset requirements
 | 
						|
        groups = [
 | 
						|
            ({"Parasol Ability", "Cutter Ability"}, {'Bukiset (Parasol)', 'Bukiset (Cutter)'}),
 | 
						|
            ({"Spark Ability", "Clean Ability"}, {'Bukiset (Spark)', 'Bukiset (Clean)'}),
 | 
						|
            ({"Ice Ability", "Needle Ability"}, {'Bukiset (Ice)', 'Bukiset (Needle)'}),
 | 
						|
            ({"Stone Ability", "Burning Ability"}, {'Bukiset (Stone)', 'Bukiset (Burning)'}),
 | 
						|
        ]
 | 
						|
        copy_abilities = self.multiworld.worlds[1].copy_abilities
 | 
						|
        required_abilities: List[Tuple[str]] = []
 | 
						|
        for abilities, bukisets in groups:
 | 
						|
            potential_abilities: List[str] = list()
 | 
						|
            for bukiset in bukisets:
 | 
						|
                if copy_abilities[bukiset] in abilities:
 | 
						|
                    potential_abilities.append(copy_abilities[bukiset])
 | 
						|
            required_abilities.append(tuple(potential_abilities))
 | 
						|
        collected_abilities = list()
 | 
						|
        for group in required_abilities:
 | 
						|
            self.assertFalse(len(group) == 0, str(self.multiworld.seed))
 | 
						|
            collected_abilities.append(group[0])
 | 
						|
        self.collect_by_name([ability.replace(" Ability", "") for ability in collected_abilities])
 | 
						|
        if "Parasol Ability" not in collected_abilities or "Stone Ability" not in collected_abilities:
 | 
						|
            # required for non-Bukiset related portions
 | 
						|
            self.collect_by_name(["Parasol", "Stone"])
 | 
						|
 | 
						|
        if "Cutter Ability" not in collected_abilities:
 | 
						|
            # we can't actually reach 3-6 without Cutter
 | 
						|
            self.assertFalse(self.can_reach_location("Sand Canyon 6 - Professor Hector & R.O.B"), str(self.multiworld.seed))
 | 
						|
            self.collect_by_name(["Cutter"])
 | 
						|
 | 
						|
        self.assertTrue(self.can_reach_location("Sand Canyon 6 - Professor Hector & R.O.B"),
 | 
						|
                        ''.join(str(self.multiworld.seed)).join(collected_abilities))
 |