Grinch-AP/test/general/test_locations.py

import unittest
from collections import Counter
from worlds.AutoWorld import AutoWorldRegister, call_all
from . import setup_solo_multiworld


class TestBase(unittest.TestCase):
    def test_create_duplicate_locations(self):
        """Tests that no two Locations share a name or ID."""
        for game_name, world_type in AutoWorldRegister.world_types.items():
            multiworld = setup_solo_multiworld(world_type)
            locations = Counter(location.name for location in multiworld.get_locations())
            if locations:
                self.assertEqual(locations.most_common(1)[0][1], 1,
                                 f"{world_type.game} has duplicate of location name {locations.most_common(1)}")

            locations = Counter(location.address for location in multiworld.get_locations()
                                if type(location.address) is int)
            if locations:
                self.assertEqual(locations.most_common(1)[0][1], 1,
                                 f"{world_type.game} has duplicate of location ID {locations.most_common(1)}")

    def test_locations_in_datapackage(self):
        """Tests that created locations not filled before fill starts exist in the datapackage."""
        for game_name, world_type in AutoWorldRegister.world_types.items():
            with self.subTest("Game", game_name=game_name):
                multiworld = setup_solo_multiworld(world_type)
                locations = multiworld.get_unfilled_locations()  # do unfilled locations to avoid Events
                for location in locations:
                    self.assertIn(location.name, world_type.location_name_to_id)
                    self.assertEqual(location.address, world_type.location_name_to_id[location.name])

    def test_location_creation_steps(self):
        """Tests that Regions and Locations aren't created after `create_items`."""
        gen_steps = ("generate_early", "create_regions", "create_items")
        for game_name, world_type in AutoWorldRegister.world_types.items():
            with self.subTest("Game", game_name=game_name):
                multiworld = setup_solo_multiworld(world_type, gen_steps)
                region_count = len(multiworld.get_regions())
                location_count = len(multiworld.get_locations())

                call_all(multiworld, "set_rules")
                self.assertEqual(region_count, len(multiworld.get_regions()),
                                 f"{game_name} modified region count during rule creation")
                self.assertEqual(location_count, len(multiworld.get_locations()),
                                 f"{game_name} modified locations count during rule creation")

                call_all(multiworld, "connect_entrances")
                self.assertEqual(region_count, len(multiworld.get_regions()),
                                 f"{game_name} modified region count during rule creation")
                self.assertEqual(location_count, len(multiworld.get_locations()),
                                 f"{game_name} modified locations count during rule creation")

                call_all(multiworld, "generate_basic")
                self.assertEqual(region_count, len(multiworld.get_regions()),
                                 f"{game_name} modified region count during generate_basic")
                self.assertGreaterEqual(location_count, len(multiworld.get_locations()),
                                        f"{game_name} modified locations count during generate_basic")

                call_all(multiworld, "pre_fill")
                self.assertEqual(region_count, len(multiworld.get_regions()),
                                 f"{game_name} modified region count during pre_fill")
                self.assertGreaterEqual(location_count, len(multiworld.get_locations()),
                                        f"{game_name} modified locations count during pre_fill")
    
    def test_location_group(self):
        """Test that all location name groups contain valid locations and don't share names."""
        for game_name, world_type in AutoWorldRegister.world_types.items():
            with self.subTest(game_name, game_name=game_name):
                for group_name, locations in world_type.location_name_groups.items():
                    with self.subTest(group_name, group_name=group_name):
                        for location in locations:
                            self.assertIn(location, world_type.location_name_to_id)
                        self.assertNotIn(group_name, world_type.location_name_to_id)