Archipelago/worlds/stardew_valley/test/TestGeneration.py

1102 lines
54 KiB
Python

from typing import List
from BaseClasses import ItemClassification, Item
from . import SVTestBase, allsanity_options_without_mods, \
allsanity_options_with_mods, minimal_locations_maximal_items, minimal_locations_maximal_items_with_island, get_minsanity_options, default_options
from .. import items, location_table, options
from ..data.villagers_data import all_villagers_by_name, all_villagers_by_mod_by_name
from ..items import Group, item_table
from ..locations import LocationTags
from ..mods.mod_data import ModNames
from ..options import Friendsanity, SpecialOrderLocations, Shipsanity, Chefsanity, SeasonRandomization, Craftsanity, ExcludeGingerIsland, ToolProgression, \
FriendsanityHeartSize
from ..strings.region_names import Region
class TestBaseItemGeneration(SVTestBase):
options = {
Friendsanity.internal_name: Friendsanity.option_all_with_marriage,
SeasonRandomization.internal_name: SeasonRandomization.option_progressive,
SpecialOrderLocations.internal_name: SpecialOrderLocations.option_board_qi,
Shipsanity.internal_name: Shipsanity.option_everything,
Chefsanity.internal_name: Chefsanity.option_all,
Craftsanity.internal_name: Craftsanity.option_all,
}
def test_all_progression_items_are_added_to_the_pool(self):
all_created_items = [item.name for item in self.multiworld.itempool]
# Ignore all the stuff that the algorithm chooses one of, instead of all, to fulfill logical progression
items_to_ignore = [event.name for event in items.events]
items_to_ignore.extend(item.name for item in items.all_items if item.mod_name is not None)
items_to_ignore.extend(deprecated.name for deprecated in items.items_by_group[Group.DEPRECATED])
items_to_ignore.extend(season.name for season in items.items_by_group[Group.SEASON])
items_to_ignore.extend(weapon.name for weapon in items.items_by_group[Group.WEAPON])
items_to_ignore.extend(baby.name for baby in items.items_by_group[Group.BABY])
items_to_ignore.extend(resource_pack.name for resource_pack in items.items_by_group[Group.RESOURCE_PACK])
items_to_ignore.append("The Gateway Gazette")
progression_items = [item for item in items.all_items if item.classification is ItemClassification.progression and item.name not in items_to_ignore]
for progression_item in progression_items:
with self.subTest(f"{progression_item.name}"):
self.assertIn(progression_item.name, all_created_items)
def test_creates_as_many_item_as_non_event_locations(self):
non_event_locations = self.get_real_locations()
self.assertEqual(len(non_event_locations), len(self.multiworld.itempool))
def test_does_not_create_deprecated_items(self):
all_created_items = [item.name for item in self.multiworld.itempool]
for deprecated_item in items.items_by_group[items.Group.DEPRECATED]:
with self.subTest(f"{deprecated_item.name}"):
self.assertNotIn(deprecated_item.name, all_created_items)
def test_does_not_create_more_than_one_maximum_one_items(self):
all_created_items = [item.name for item in self.multiworld.itempool]
for maximum_one_item in items.items_by_group[items.Group.MAXIMUM_ONE]:
with self.subTest(f"{maximum_one_item.name}"):
self.assertLessEqual(all_created_items.count(maximum_one_item.name), 1)
def test_does_not_create_exactly_two_items(self):
all_created_items = [item.name for item in self.multiworld.itempool]
for exactly_two_item in items.items_by_group[items.Group.EXACTLY_TWO]:
with self.subTest(f"{exactly_two_item.name}"):
count = all_created_items.count(exactly_two_item.name)
self.assertTrue(count == 0 or count == 2)
class TestNoGingerIslandItemGeneration(SVTestBase):
options = {
Friendsanity.internal_name: Friendsanity.option_all_with_marriage,
SeasonRandomization.internal_name: SeasonRandomization.option_progressive,
ExcludeGingerIsland.internal_name: ExcludeGingerIsland.option_true,
Shipsanity.internal_name: Shipsanity.option_everything,
Chefsanity.internal_name: Chefsanity.option_all,
Craftsanity.internal_name: Craftsanity.option_all,
}
def test_all_progression_items_except_island_are_added_to_the_pool(self):
all_created_items = [item.name for item in self.multiworld.itempool]
# Ignore all the stuff that the algorithm chooses one of, instead of all, to fulfill logical progression
items_to_ignore = [event.name for event in items.events]
items_to_ignore.extend(item.name for item in items.all_items if item.mod_name is not None)
items_to_ignore.extend(deprecated.name for deprecated in items.items_by_group[Group.DEPRECATED])
items_to_ignore.extend(season.name for season in items.items_by_group[Group.SEASON])
items_to_ignore.extend(season.name for season in items.items_by_group[Group.WEAPON])
items_to_ignore.extend(baby.name for baby in items.items_by_group[Group.BABY])
items_to_ignore.append("The Gateway Gazette")
progression_items = [item for item in items.all_items if item.classification is ItemClassification.progression and item.name not in items_to_ignore]
for progression_item in progression_items:
with self.subTest(f"{progression_item.name}"):
if Group.GINGER_ISLAND in progression_item.groups:
self.assertNotIn(progression_item.name, all_created_items)
else:
self.assertIn(progression_item.name, all_created_items)
def test_creates_as_many_item_as_non_event_locations(self):
non_event_locations = self.get_real_locations()
self.assertEqual(len(non_event_locations), len(self.multiworld.itempool))
def test_does_not_create_deprecated_items(self):
all_created_items = [item.name for item in self.multiworld.itempool]
for deprecated_item in items.items_by_group[items.Group.DEPRECATED]:
with self.subTest(f"Deprecated item: {deprecated_item.name}"):
self.assertNotIn(deprecated_item.name, all_created_items)
def test_does_not_create_more_than_one_maximum_one_items(self):
all_created_items = [item.name for item in self.multiworld.itempool]
for maximum_one_item in items.items_by_group[items.Group.MAXIMUM_ONE]:
with self.subTest(f"{maximum_one_item.name}"):
self.assertLessEqual(all_created_items.count(maximum_one_item.name), 1)
def test_does_not_create_exactly_two_items(self):
all_created_items = [item.name for item in self.multiworld.itempool]
for exactly_two_item in items.items_by_group[items.Group.EXACTLY_TWO]:
with self.subTest(f"{exactly_two_item.name}"):
count = all_created_items.count(exactly_two_item.name)
self.assertTrue(count == 0 or count == 2)
class TestMonstersanityNone(SVTestBase):
options = {options.Monstersanity.internal_name: options.Monstersanity.option_none}
def test_when_generate_world_then_5_generic_weapons_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Weapon"), 5)
def test_when_generate_world_then_zero_specific_weapons_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Sword"), 0)
self.assertEqual(item_pool.count("Progressive Club"), 0)
self.assertEqual(item_pool.count("Progressive Dagger"), 0)
def test_when_generate_world_then_2_slingshots_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Slingshot"), 2)
def test_when_generate_world_then_3_shoes_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Footwear"), 3)
class TestMonstersanityGoals(SVTestBase):
options = {options.Monstersanity.internal_name: options.Monstersanity.option_goals}
def test_when_generate_world_then_no_generic_weapons_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Weapon"), 0)
def test_when_generate_world_then_5_specific_weapons_of_each_type_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Sword"), 5)
self.assertEqual(item_pool.count("Progressive Club"), 5)
self.assertEqual(item_pool.count("Progressive Dagger"), 5)
def test_when_generate_world_then_2_slingshots_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Slingshot"), 2)
def test_when_generate_world_then_4_shoes_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Footwear"), 4)
def test_when_generate_world_then_all_monster_checks_are_inaccessible(self):
for location in self.get_real_locations():
if LocationTags.MONSTERSANITY not in location_table[location.name].tags:
continue
with self.subTest(location.name):
self.assertFalse(location.can_reach(self.multiworld.state))
class TestMonstersanityOnePerCategory(SVTestBase):
options = {options.Monstersanity.internal_name: options.Monstersanity.option_one_per_category}
def test_when_generate_world_then_no_generic_weapons_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Weapon"), 0)
def test_when_generate_world_then_5_specific_weapons_of_each_type_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Sword"), 5)
self.assertEqual(item_pool.count("Progressive Club"), 5)
self.assertEqual(item_pool.count("Progressive Dagger"), 5)
def test_when_generate_world_then_2_slingshots_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Slingshot"), 2)
def test_when_generate_world_then_4_shoes_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Footwear"), 4)
def test_when_generate_world_then_all_monster_checks_are_inaccessible(self):
for location in self.get_real_locations():
if LocationTags.MONSTERSANITY not in location_table[location.name].tags:
continue
with self.subTest(location.name):
self.assertFalse(location.can_reach(self.multiworld.state))
class TestMonstersanityProgressive(SVTestBase):
options = {options.Monstersanity.internal_name: options.Monstersanity.option_progressive_goals}
def test_when_generate_world_then_no_generic_weapons_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Weapon"), 0)
def test_when_generate_world_then_5_specific_weapons_of_each_type_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Sword"), 5)
self.assertEqual(item_pool.count("Progressive Club"), 5)
self.assertEqual(item_pool.count("Progressive Dagger"), 5)
def test_when_generate_world_then_2_slingshots_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Slingshot"), 2)
def test_when_generate_world_then_4_shoes_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Footwear"), 4)
def test_when_generate_world_then_many_rings_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertIn("Hot Java Ring", item_pool)
self.assertIn("Wedding Ring", item_pool)
self.assertIn("Slime Charmer Ring", item_pool)
def test_when_generate_world_then_all_monster_checks_are_inaccessible(self):
for location in self.get_real_locations():
if LocationTags.MONSTERSANITY not in location_table[location.name].tags:
continue
with self.subTest(location.name):
self.assertFalse(location.can_reach(self.multiworld.state))
class TestMonstersanitySplit(SVTestBase):
options = {options.Monstersanity.internal_name: options.Monstersanity.option_split_goals}
def test_when_generate_world_then_no_generic_weapons_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Weapon"), 0)
def test_when_generate_world_then_5_specific_weapons_of_each_type_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Sword"), 5)
self.assertEqual(item_pool.count("Progressive Club"), 5)
self.assertEqual(item_pool.count("Progressive Dagger"), 5)
def test_when_generate_world_then_2_slingshots_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Slingshot"), 2)
def test_when_generate_world_then_4_shoes_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertEqual(item_pool.count("Progressive Footwear"), 4)
def test_when_generate_world_then_many_rings_in_the_pool(self):
item_pool = [item.name for item in self.multiworld.itempool]
self.assertIn("Hot Java Ring", item_pool)
self.assertIn("Wedding Ring", item_pool)
self.assertIn("Slime Charmer Ring", item_pool)
def test_when_generate_world_then_all_monster_checks_are_inaccessible(self):
for location in self.get_real_locations():
if LocationTags.MONSTERSANITY not in location_table[location.name].tags:
continue
with self.subTest(location.name):
self.assertFalse(location.can_reach(self.multiworld.state))
class TestProgressiveElevator(SVTestBase):
options = {
options.ElevatorProgression.internal_name: options.ElevatorProgression.option_progressive,
options.ToolProgression.internal_name: options.ToolProgression.option_progressive,
options.SkillProgression.internal_name: options.SkillProgression.option_progressive,
}
def test_given_elevator_to_floor_105_when_find_another_elevator_then_has_access_to_floor_120(self):
items_for_115 = self.generate_items_for_mine_115()
last_elevator = self.get_item_by_name("Progressive Mine Elevator")
self.collect(items_for_115)
floor_115 = self.multiworld.get_region("The Mines - Floor 115", self.player)
floor_120 = self.multiworld.get_region("The Mines - Floor 120", self.player)
self.assertTrue(floor_115.can_reach(self.multiworld.state))
self.assertFalse(floor_120.can_reach(self.multiworld.state))
self.collect(last_elevator)
self.assertTrue(floor_120.can_reach(self.multiworld.state))
def generate_items_for_mine_115(self) -> List[Item]:
pickaxes = [self.get_item_by_name("Progressive Pickaxe")] * 2
elevators = [self.get_item_by_name("Progressive Mine Elevator")] * 21
swords = [self.get_item_by_name("Progressive Sword")] * 3
combat_levels = [self.get_item_by_name("Combat Level")] * 4
mining_levels = [self.get_item_by_name("Mining Level")] * 4
return [*combat_levels, *mining_levels, *elevators, *pickaxes, *swords]
def generate_items_for_extra_mine_levels(self, weapon_name: str) -> List[Item]:
last_pickaxe = self.get_item_by_name("Progressive Pickaxe")
last_weapon = self.multiworld.create_item(weapon_name, self.player)
second_last_combat_level = self.get_item_by_name("Combat Level")
last_combat_level = self.get_item_by_name("Combat Level")
second_last_mining_level = self.get_item_by_name("Mining Level")
last_mining_level = self.get_item_by_name("Mining Level")
return [last_pickaxe, last_weapon, second_last_combat_level, last_combat_level, second_last_mining_level, last_mining_level]
class TestSkullCavernLogic(SVTestBase):
options = {
options.ElevatorProgression.internal_name: options.ElevatorProgression.option_vanilla,
ToolProgression.internal_name: ToolProgression.option_progressive,
options.SkillProgression.internal_name: options.SkillProgression.option_progressive,
}
def test_given_access_to_floor_115_when_find_more_tools_then_has_access_to_skull_cavern_25(self):
items_for_115 = self.generate_items_for_mine_115()
items_for_skull_50 = self.generate_items_for_skull_50()
items_for_skull_100 = self.generate_items_for_skull_100()
self.collect(items_for_115)
floor_115 = self.multiworld.get_region(Region.mines_floor_115, self.player)
skull_25 = self.multiworld.get_region(Region.skull_cavern_25, self.player)
skull_75 = self.multiworld.get_region(Region.skull_cavern_75, self.player)
self.assertTrue(floor_115.can_reach(self.multiworld.state))
self.assertFalse(skull_25.can_reach(self.multiworld.state))
self.assertFalse(skull_75.can_reach(self.multiworld.state))
self.remove(items_for_115)
self.collect(items_for_skull_50)
self.assertTrue(floor_115.can_reach(self.multiworld.state))
self.assertTrue(skull_25.can_reach(self.multiworld.state))
self.assertFalse(skull_75.can_reach(self.multiworld.state))
self.remove(items_for_skull_50)
self.collect(items_for_skull_100)
self.assertTrue(floor_115.can_reach(self.multiworld.state))
self.assertTrue(skull_25.can_reach(self.multiworld.state))
self.assertTrue(skull_75.can_reach(self.multiworld.state))
def generate_items_for_mine_115(self) -> List[Item]:
pickaxes = [self.get_item_by_name("Progressive Pickaxe")] * 2
swords = [self.get_item_by_name("Progressive Sword")] * 3
combat_levels = [self.get_item_by_name("Combat Level")] * 4
mining_levels = [self.get_item_by_name("Mining Level")] * 4
bus = self.get_item_by_name("Bus Repair")
skull_key = self.get_item_by_name("Skull Key")
return [*combat_levels, *mining_levels, *pickaxes, *swords, bus, skull_key]
def generate_items_for_skull_50(self) -> List[Item]:
pickaxes = [self.get_item_by_name("Progressive Pickaxe")] * 3
swords = [self.get_item_by_name("Progressive Sword")] * 4
combat_levels = [self.get_item_by_name("Combat Level")] * 6
mining_levels = [self.get_item_by_name("Mining Level")] * 6
bus = self.get_item_by_name("Bus Repair")
skull_key = self.get_item_by_name("Skull Key")
return [*combat_levels, *mining_levels, *pickaxes, *swords, bus, skull_key]
def generate_items_for_skull_100(self) -> List[Item]:
pickaxes = [self.get_item_by_name("Progressive Pickaxe")] * 4
swords = [self.get_item_by_name("Progressive Sword")] * 5
combat_levels = [self.get_item_by_name("Combat Level")] * 8
mining_levels = [self.get_item_by_name("Mining Level")] * 8
bus = self.get_item_by_name("Bus Repair")
skull_key = self.get_item_by_name("Skull Key")
return [*combat_levels, *mining_levels, *pickaxes, *swords, bus, skull_key]
class TestLocationGeneration(SVTestBase):
def test_all_location_created_are_in_location_table(self):
for location in self.get_real_locations():
self.assertIn(location.name, location_table)
class TestMinLocationAndMaxItem(SVTestBase):
options = minimal_locations_maximal_items()
# They do not pass and I don't know why.
skip_base_tests = True
def test_minimal_location_maximal_items_still_valid(self):
valid_locations = self.get_real_locations()
number_locations = len(valid_locations)
number_items = len([item for item in self.multiworld.itempool
if Group.RESOURCE_PACK not in item_table[item.name].groups and Group.TRAP not in item_table[item.name].groups])
self.assertGreaterEqual(number_locations, number_items)
print(f"Stardew Valley - Minimum Locations: {number_locations}, Maximum Items: {number_items} [ISLAND EXCLUDED]")
class TestMinLocationAndMaxItemWithIsland(SVTestBase):
options = minimal_locations_maximal_items_with_island()
def test_minimal_location_maximal_items_with_island_still_valid(self):
valid_locations = self.get_real_locations()
number_locations = len(valid_locations)
number_items = len([item for item in self.multiworld.itempool
if Group.RESOURCE_PACK not in item_table[item.name].groups and Group.TRAP not in item_table[item.name].groups])
self.assertGreaterEqual(number_locations, number_items)
print(f"Stardew Valley - Minimum Locations: {number_locations}, Maximum Items: {number_items} [ISLAND INCLUDED]")
class TestMinSanityHasAllExpectedLocations(SVTestBase):
options = get_minsanity_options()
def test_minsanity_has_fewer_than_locations(self):
expected_locations = 76
real_locations = self.get_real_locations()
number_locations = len(real_locations)
self.assertLessEqual(number_locations, expected_locations)
print(f"Stardew Valley - Minsanity Locations: {number_locations}")
if number_locations != expected_locations:
print(f"\tDisappeared Locations Detected!"
f"\n\tPlease update test_minsanity_has_fewer_than_locations"
f"\n\t\tExpected: {expected_locations}"
f"\n\t\tActual: {number_locations}")
class TestDefaultSettingsHasAllExpectedLocations(SVTestBase):
options = default_options()
def test_default_settings_has_exactly_locations(self):
expected_locations = 422
real_locations = self.get_real_locations()
number_locations = len(real_locations)
print(f"Stardew Valley - Default options locations: {number_locations}")
if number_locations != expected_locations:
print(f"\tNew locations detected!"
f"\n\tPlease update test_default_settings_has_exactly_locations"
f"\n\t\tExpected: {expected_locations}"
f"\n\t\tActual: {number_locations}")
class TestAllSanitySettingsHasAllExpectedLocations(SVTestBase):
options = allsanity_options_without_mods()
def test_allsanity_without_mods_has_at_least_locations(self):
expected_locations = 1956
real_locations = self.get_real_locations()
number_locations = len(real_locations)
self.assertGreaterEqual(number_locations, expected_locations)
print(f"Stardew Valley - Allsanity Locations without mods: {number_locations}")
if number_locations != expected_locations:
print(f"\tNew locations detected!"
f"\n\tPlease update test_allsanity_without_mods_has_at_least_locations"
f"\n\t\tExpected: {expected_locations}"
f"\n\t\tActual: {number_locations}")
class TestAllSanityWithModsSettingsHasAllExpectedLocations(SVTestBase):
options = allsanity_options_with_mods()
def test_allsanity_with_mods_has_at_least_locations(self):
expected_locations = 2804
real_locations = self.get_real_locations()
number_locations = len(real_locations)
self.assertGreaterEqual(number_locations, expected_locations)
print(f"\nStardew Valley - Allsanity Locations with all mods: {number_locations}")
if number_locations != expected_locations:
print(f"\tNew locations detected!"
f"\n\tPlease update test_allsanity_with_mods_has_at_least_locations"
f"\n\t\tExpected: {expected_locations}"
f"\n\t\tActual: {number_locations}")
class TestFriendsanityNone(SVTestBase):
options = {
Friendsanity.internal_name: Friendsanity.option_none,
}
@property
def run_default_tests(self) -> bool:
# None is default
return False
def test_friendsanity_none(self):
with self.subTest("No Items"):
self.check_no_friendsanity_items()
with self.subTest("No Locations"):
self.check_no_friendsanity_locations()
def check_no_friendsanity_items(self):
for item in self.multiworld.itempool:
self.assertFalse(item.name.endswith(" <3"))
def check_no_friendsanity_locations(self):
for location_name in self.get_real_location_names():
self.assertFalse(location_name.startswith("Friendsanity"))
class TestFriendsanityBachelors(SVTestBase):
options = {
Friendsanity.internal_name: Friendsanity.option_bachelors,
FriendsanityHeartSize.internal_name: 1,
}
bachelors = {"Harvey", "Elliott", "Sam", "Alex", "Shane", "Sebastian", "Emily", "Haley", "Leah", "Abigail", "Penny",
"Maru"}
def test_friendsanity_only_bachelors(self):
with self.subTest("Items are valid"):
self.check_only_bachelors_items()
with self.subTest("Locations are valid"):
self.check_only_bachelors_locations()
def check_only_bachelors_items(self):
suffix = " <3"
for item in self.multiworld.itempool:
if item.name.endswith(suffix):
villager_name = item.name[:item.name.index(suffix)]
self.assertIn(villager_name, self.bachelors)
def check_only_bachelors_locations(self):
prefix = "Friendsanity: "
suffix = " <3"
for location_name in self.get_real_location_names():
if location_name.startswith(prefix):
name_no_prefix = location_name[len(prefix):]
name_trimmed = name_no_prefix[:name_no_prefix.index(suffix)]
parts = name_trimmed.split(" ")
name = parts[0]
hearts = parts[1]
self.assertIn(name, self.bachelors)
self.assertLessEqual(int(hearts), 8)
class TestFriendsanityStartingNpcs(SVTestBase):
options = {
Friendsanity.internal_name: Friendsanity.option_starting_npcs,
FriendsanityHeartSize.internal_name: 1,
}
excluded_npcs = {"Leo", "Krobus", "Dwarf", "Sandy", "Kent"}
def test_friendsanity_only_starting_npcs(self):
with self.subTest("Items are valid"):
self.check_only_starting_npcs_items()
with self.subTest("Locations are valid"):
self.check_only_starting_npcs_locations()
def check_only_starting_npcs_items(self):
suffix = " <3"
for item in self.multiworld.itempool:
if item.name.endswith(suffix):
villager_name = item.name[:item.name.index(suffix)]
self.assertNotIn(villager_name, self.excluded_npcs)
def check_only_starting_npcs_locations(self):
prefix = "Friendsanity: "
suffix = " <3"
for location_name in self.get_real_location_names():
if location_name.startswith(prefix):
name_no_prefix = location_name[len(prefix):]
name_trimmed = name_no_prefix[:name_no_prefix.index(suffix)]
parts = name_trimmed.split(" ")
name = parts[0]
hearts = parts[1]
self.assertNotIn(name, self.excluded_npcs)
self.assertTrue(name in all_villagers_by_mod_by_name[ModNames.vanilla] or name == "Pet")
if name == "Pet":
self.assertLessEqual(int(hearts), 5)
elif all_villagers_by_name[name].bachelor:
self.assertLessEqual(int(hearts), 8)
else:
self.assertLessEqual(int(hearts), 10)
class TestFriendsanityAllNpcs(SVTestBase):
options = {
Friendsanity.internal_name: Friendsanity.option_all,
FriendsanityHeartSize.internal_name: 4,
}
def test_friendsanity_all_npcs(self):
with self.subTest("Items are valid"):
self.check_items_are_valid()
with self.subTest("Correct number of items"):
self.check_correct_number_of_items()
with self.subTest("Locations are valid"):
self.check_locations_are_valid()
def check_items_are_valid(self):
suffix = " <3"
for item in self.multiworld.itempool:
if item.name.endswith(suffix):
villager_name = item.name[:item.name.index(suffix)]
self.assertTrue(villager_name in all_villagers_by_mod_by_name[ModNames.vanilla] or villager_name == "Pet")
def check_correct_number_of_items(self):
suffix = " <3"
item_names = [item.name for item in self.multiworld.itempool]
for villager_name in all_villagers_by_mod_by_name[ModNames.vanilla]:
heart_item_name = f"{villager_name}{suffix}"
number_heart_items = item_names.count(heart_item_name)
if all_villagers_by_name[villager_name].bachelor:
self.assertEqual(number_heart_items, 2)
else:
self.assertEqual(number_heart_items, 3)
self.assertEqual(item_names.count("Pet <3"), 2)
def check_locations_are_valid(self):
prefix = "Friendsanity: "
suffix = " <3"
for location_name in self.get_real_location_names():
if not location_name.startswith(prefix):
continue
name_no_prefix = location_name[len(prefix):]
name_trimmed = name_no_prefix[:name_no_prefix.index(suffix)]
parts = name_trimmed.split(" ")
name = parts[0]
hearts = int(parts[1])
self.assertTrue(name in all_villagers_by_mod_by_name[ModNames.vanilla] or name == "Pet")
if name == "Pet":
self.assertTrue(hearts == 4 or hearts == 5)
elif all_villagers_by_name[name].bachelor:
self.assertTrue(hearts == 4 or hearts == 8 or hearts == 12 or hearts == 14)
else:
self.assertTrue(hearts == 4 or hearts == 8 or hearts == 10)
class TestFriendsanityAllNpcsExcludingGingerIsland(SVTestBase):
options = {
Friendsanity.internal_name: Friendsanity.option_all,
FriendsanityHeartSize.internal_name: 4,
ExcludeGingerIsland.internal_name: ExcludeGingerIsland.option_true
}
def test_friendsanity_all_npcs_exclude_island(self):
with self.subTest("Items"):
self.check_items()
with self.subTest("Locations"):
self.check_locations()
def check_items(self):
suffix = " <3"
for item in self.multiworld.itempool:
if item.name.endswith(suffix):
villager_name = item.name[:item.name.index(suffix)]
self.assertNotEqual(villager_name, "Leo")
self.assertTrue(villager_name in all_villagers_by_mod_by_name[ModNames.vanilla] or villager_name == "Pet")
def check_locations(self):
prefix = "Friendsanity: "
suffix = " <3"
for location_name in self.get_real_location_names():
if location_name.startswith(prefix):
name_no_prefix = location_name[len(prefix):]
name_trimmed = name_no_prefix[:name_no_prefix.index(suffix)]
parts = name_trimmed.split(" ")
name = parts[0]
hearts = parts[1]
self.assertNotEqual(name, "Leo")
self.assertTrue(name in all_villagers_by_mod_by_name[ModNames.vanilla] or name == "Pet")
if name == "Pet":
self.assertLessEqual(int(hearts), 5)
elif all_villagers_by_name[name].bachelor:
self.assertLessEqual(int(hearts), 8)
else:
self.assertLessEqual(int(hearts), 10)
class TestFriendsanityHeartSize3(SVTestBase):
options = {
Friendsanity.internal_name: Friendsanity.option_all_with_marriage,
FriendsanityHeartSize.internal_name: 3,
}
def test_friendsanity_all_npcs_with_marriage(self):
with self.subTest("Items are valid"):
self.check_items_are_valid()
with self.subTest("Correct number of items"):
self.check_correct_number_of_items()
with self.subTest("Locations are valid"):
self.check_locations_are_valid()
def check_items_are_valid(self):
suffix = " <3"
for item in self.multiworld.itempool:
if item.name.endswith(suffix):
villager_name = item.name[:item.name.index(suffix)]
self.assertTrue(villager_name in all_villagers_by_mod_by_name[ModNames.vanilla] or villager_name == "Pet")
def check_correct_number_of_items(self):
suffix = " <3"
item_names = [item.name for item in self.multiworld.itempool]
for villager_name in all_villagers_by_mod_by_name[ModNames.vanilla]:
heart_item_name = f"{villager_name}{suffix}"
number_heart_items = item_names.count(heart_item_name)
if all_villagers_by_name[villager_name].bachelor:
self.assertEqual(number_heart_items, 5)
else:
self.assertEqual(number_heart_items, 4)
self.assertEqual(item_names.count("Pet <3"), 2)
def check_locations_are_valid(self):
prefix = "Friendsanity: "
suffix = " <3"
for location_name in self.get_real_location_names():
if not location_name.startswith(prefix):
continue
name_no_prefix = location_name[len(prefix):]
name_trimmed = name_no_prefix[:name_no_prefix.index(suffix)]
parts = name_trimmed.split(" ")
name = parts[0]
hearts = int(parts[1])
self.assertTrue(name in all_villagers_by_mod_by_name[ModNames.vanilla] or name == "Pet")
if name == "Pet":
self.assertTrue(hearts == 3 or hearts == 5)
elif all_villagers_by_name[name].bachelor:
self.assertTrue(hearts == 3 or hearts == 6 or hearts == 9 or hearts == 12 or hearts == 14)
else:
self.assertTrue(hearts == 3 or hearts == 6 or hearts == 9 or hearts == 10)
class TestFriendsanityHeartSize5(SVTestBase):
options = {
Friendsanity.internal_name: Friendsanity.option_all_with_marriage,
FriendsanityHeartSize.internal_name: 5,
}
def test_friendsanity_all_npcs_with_marriage(self):
with self.subTest("Items are valid"):
self.check_items_are_valid()
with self.subTest("Correct number of items"):
self.check_correct_number_of_items()
with self.subTest("Locations are valid"):
self.check_locations_are_valid()
def check_items_are_valid(self):
suffix = " <3"
for item in self.multiworld.itempool:
if item.name.endswith(suffix):
villager_name = item.name[:item.name.index(suffix)]
self.assertTrue(villager_name in all_villagers_by_mod_by_name[ModNames.vanilla] or villager_name == "Pet")
def check_correct_number_of_items(self):
suffix = " <3"
item_names = [item.name for item in self.multiworld.itempool]
for villager_name in all_villagers_by_mod_by_name[ModNames.vanilla]:
heart_item_name = f"{villager_name}{suffix}"
number_heart_items = item_names.count(heart_item_name)
if all_villagers_by_name[villager_name].bachelor:
self.assertEqual(number_heart_items, 3)
else:
self.assertEqual(number_heart_items, 2)
self.assertEqual(item_names.count("Pet <3"), 1)
def check_locations_are_valid(self):
prefix = "Friendsanity: "
suffix = " <3"
for location_name in self.get_real_location_names():
if not location_name.startswith(prefix):
continue
name_no_prefix = location_name[len(prefix):]
name_trimmed = name_no_prefix[:name_no_prefix.index(suffix)]
parts = name_trimmed.split(" ")
name = parts[0]
hearts = int(parts[1])
self.assertTrue(name in all_villagers_by_mod_by_name[ModNames.vanilla] or name == "Pet")
if name == "Pet":
self.assertTrue(hearts == 5)
elif all_villagers_by_name[name].bachelor:
self.assertTrue(hearts == 5 or hearts == 10 or hearts == 14)
else:
self.assertTrue(hearts == 5 or hearts == 10)
class TestShipsanityNone(SVTestBase):
options = {
Shipsanity.internal_name: Shipsanity.option_none
}
def test_no_shipsanity_locations(self):
for location in self.get_real_locations():
with self.subTest(location.name):
self.assertFalse("Shipsanity" in location.name)
self.assertNotIn(LocationTags.SHIPSANITY, location_table[location.name].tags)
class TestShipsanityCrops(SVTestBase):
options = {
Shipsanity.internal_name: Shipsanity.option_crops,
SpecialOrderLocations.internal_name: SpecialOrderLocations.option_board_qi
}
def test_only_crop_shipsanity_locations(self):
for location in self.get_real_locations():
if LocationTags.SHIPSANITY in location_table[location.name].tags:
with self.subTest(location.name):
self.assertIn(LocationTags.SHIPSANITY_CROP, location_table[location.name].tags)
def test_include_island_crop_shipsanity_locations(self):
location_names = [location.name for location in self.multiworld.get_locations(self.player)]
self.assertIn("Shipsanity: Banana", location_names)
self.assertIn("Shipsanity: Mango", location_names)
self.assertIn("Shipsanity: Pineapple", location_names)
self.assertIn("Shipsanity: Taro Root", location_names)
self.assertIn("Shipsanity: Ginger", location_names)
self.assertIn("Shipsanity: Magma Cap", location_names)
self.assertIn("Shipsanity: Qi Fruit", location_names)
class TestShipsanityCropsExcludeIsland(SVTestBase):
options = {
Shipsanity.internal_name: Shipsanity.option_crops,
ExcludeGingerIsland.internal_name: ExcludeGingerIsland.option_true
}
def test_only_crop_shipsanity_locations(self):
for location in self.get_real_locations():
if LocationTags.SHIPSANITY in location_table[location.name].tags:
with self.subTest(location.name):
self.assertIn(LocationTags.SHIPSANITY_CROP, location_table[location.name].tags)
def test_only_mainland_crop_shipsanity_locations(self):
location_names = [location.name for location in self.multiworld.get_locations(self.player)]
self.assertNotIn("Shipsanity: Banana", location_names)
self.assertNotIn("Shipsanity: Mango", location_names)
self.assertNotIn("Shipsanity: Pineapple", location_names)
self.assertNotIn("Shipsanity: Taro Root", location_names)
self.assertNotIn("Shipsanity: Ginger", location_names)
self.assertNotIn("Shipsanity: Magma Cap", location_names)
self.assertNotIn("Shipsanity: Qi Fruit", location_names)
class TestShipsanityCropsNoQiCropWithoutSpecialOrders(SVTestBase):
options = {
Shipsanity.internal_name: Shipsanity.option_crops,
SpecialOrderLocations.internal_name: SpecialOrderLocations.option_board_only
}
def test_only_crop_shipsanity_locations(self):
for location in self.get_real_locations():
if LocationTags.SHIPSANITY in location_table[location.name].tags:
with self.subTest(location.name):
self.assertIn(LocationTags.SHIPSANITY_CROP, location_table[location.name].tags)
def test_island_crops_without_qi_fruit_shipsanity_locations(self):
location_names = [location.name for location in self.multiworld.get_locations(self.player)]
self.assertIn("Shipsanity: Banana", location_names)
self.assertIn("Shipsanity: Mango", location_names)
self.assertIn("Shipsanity: Pineapple", location_names)
self.assertIn("Shipsanity: Taro Root", location_names)
self.assertIn("Shipsanity: Ginger", location_names)
self.assertIn("Shipsanity: Magma Cap", location_names)
self.assertNotIn("Shipsanity: Qi Fruit", location_names)
class TestShipsanityFish(SVTestBase):
options = {
Shipsanity.internal_name: Shipsanity.option_fish,
SpecialOrderLocations.internal_name: SpecialOrderLocations.option_board_qi
}
def test_only_fish_shipsanity_locations(self):
for location in self.get_real_locations():
if LocationTags.SHIPSANITY in location_table[location.name].tags:
with self.subTest(location.name):
self.assertIn(LocationTags.SHIPSANITY_FISH, location_table[location.name].tags)
def test_include_island_fish_shipsanity_locations(self):
location_names = [location.name for location in self.multiworld.get_locations(self.player)]
self.assertIn("Shipsanity: Blue Discus", location_names)
self.assertIn("Shipsanity: Lionfish", location_names)
self.assertIn("Shipsanity: Stingray", location_names)
self.assertIn("Shipsanity: Glacierfish Jr.", location_names)
self.assertIn("Shipsanity: Legend II", location_names)
self.assertIn("Shipsanity: Ms. Angler", location_names)
self.assertIn("Shipsanity: Radioactive Carp", location_names)
self.assertIn("Shipsanity: Son of Crimsonfish", location_names)
class TestShipsanityFishExcludeIsland(SVTestBase):
options = {
Shipsanity.internal_name: Shipsanity.option_fish,
ExcludeGingerIsland.internal_name: ExcludeGingerIsland.option_true
}
def test_only_fish_shipsanity_locations(self):
for location in self.get_real_locations():
if LocationTags.SHIPSANITY in location_table[location.name].tags:
with self.subTest(location.name):
self.assertIn(LocationTags.SHIPSANITY_FISH, location_table[location.name].tags)
def test_exclude_island_fish_shipsanity_locations(self):
location_names = [location.name for location in self.multiworld.get_locations(self.player)]
self.assertNotIn("Shipsanity: Blue Discus", location_names)
self.assertNotIn("Shipsanity: Lionfish", location_names)
self.assertNotIn("Shipsanity: Stingray", location_names)
self.assertNotIn("Shipsanity: Glacierfish Jr.", location_names)
self.assertNotIn("Shipsanity: Legend II", location_names)
self.assertNotIn("Shipsanity: Ms. Angler", location_names)
self.assertNotIn("Shipsanity: Radioactive Carp", location_names)
self.assertNotIn("Shipsanity: Son of Crimsonfish", location_names)
class TestShipsanityFishExcludeQiOrders(SVTestBase):
options = {
Shipsanity.internal_name: Shipsanity.option_fish,
SpecialOrderLocations.internal_name: SpecialOrderLocations.option_board_only
}
def test_only_fish_shipsanity_locations(self):
for location in self.get_real_locations():
if LocationTags.SHIPSANITY in location_table[location.name].tags:
with self.subTest(location.name):
self.assertIn(LocationTags.SHIPSANITY_FISH, location_table[location.name].tags)
def test_include_island_fish_no_extended_family_shipsanity_locations(self):
location_names = [location.name for location in self.multiworld.get_locations(self.player)]
self.assertIn("Shipsanity: Blue Discus", location_names)
self.assertIn("Shipsanity: Lionfish", location_names)
self.assertIn("Shipsanity: Stingray", location_names)
self.assertNotIn("Shipsanity: Glacierfish Jr.", location_names)
self.assertNotIn("Shipsanity: Legend II", location_names)
self.assertNotIn("Shipsanity: Ms. Angler", location_names)
self.assertNotIn("Shipsanity: Radioactive Carp", location_names)
self.assertNotIn("Shipsanity: Son of Crimsonfish", location_names)
class TestShipsanityFullShipment(SVTestBase):
options = {
Shipsanity.internal_name: Shipsanity.option_full_shipment,
SpecialOrderLocations.internal_name: SpecialOrderLocations.option_board_qi
}
def test_only_full_shipment_shipsanity_locations(self):
for location in self.get_real_locations():
if LocationTags.SHIPSANITY in location_table[location.name].tags:
with self.subTest(location.name):
self.assertIn(LocationTags.SHIPSANITY_FULL_SHIPMENT, location_table[location.name].tags)
self.assertNotIn(LocationTags.SHIPSANITY_FISH, location_table[location.name].tags)
def test_include_island_items_shipsanity_locations(self):
location_names = [location.name for location in self.multiworld.get_locations(self.player)]
self.assertIn("Shipsanity: Cinder Shard", location_names)
self.assertIn("Shipsanity: Bone Fragment", location_names)
self.assertIn("Shipsanity: Radioactive Ore", location_names)
self.assertIn("Shipsanity: Radioactive Bar", location_names)
self.assertIn("Shipsanity: Banana", location_names)
self.assertIn("Shipsanity: Mango", location_names)
self.assertIn("Shipsanity: Pineapple", location_names)
self.assertIn("Shipsanity: Taro Root", location_names)
self.assertIn("Shipsanity: Ginger", location_names)
self.assertIn("Shipsanity: Magma Cap", location_names)
class TestShipsanityFullShipmentExcludeIsland(SVTestBase):
options = {
Shipsanity.internal_name: Shipsanity.option_full_shipment,
ExcludeGingerIsland.internal_name: ExcludeGingerIsland.option_true
}
def test_only_full_shipment_shipsanity_locations(self):
for location in self.get_real_locations():
if LocationTags.SHIPSANITY in location_table[location.name].tags:
with self.subTest(location.name):
self.assertIn(LocationTags.SHIPSANITY_FULL_SHIPMENT, location_table[location.name].tags)
self.assertNotIn(LocationTags.SHIPSANITY_FISH, location_table[location.name].tags)
def test_exclude_island_items_shipsanity_locations(self):
location_names = [location.name for location in self.multiworld.get_locations(self.player)]
self.assertNotIn("Shipsanity: Cinder Shard", location_names)
self.assertNotIn("Shipsanity: Radioactive Ore", location_names)
self.assertNotIn("Shipsanity: Radioactive Bar", location_names)
self.assertNotIn("Shipsanity: Banana", location_names)
self.assertNotIn("Shipsanity: Mango", location_names)
self.assertNotIn("Shipsanity: Pineapple", location_names)
self.assertNotIn("Shipsanity: Taro Root", location_names)
self.assertNotIn("Shipsanity: Ginger", location_names)
self.assertNotIn("Shipsanity: Magma Cap", location_names)
class TestShipsanityFullShipmentExcludeQiBoard(SVTestBase):
options = {
Shipsanity.internal_name: Shipsanity.option_full_shipment,
SpecialOrderLocations.internal_name: SpecialOrderLocations.option_disabled
}
def test_only_full_shipment_shipsanity_locations(self):
for location in self.get_real_locations():
if LocationTags.SHIPSANITY in location_table[location.name].tags:
with self.subTest(location.name):
self.assertIn(LocationTags.SHIPSANITY_FULL_SHIPMENT, location_table[location.name].tags)
self.assertNotIn(LocationTags.SHIPSANITY_FISH, location_table[location.name].tags)
def test_exclude_qi_board_items_shipsanity_locations(self):
location_names = [location.name for location in self.multiworld.get_locations(self.player)]
self.assertIn("Shipsanity: Cinder Shard", location_names)
self.assertIn("Shipsanity: Bone Fragment", location_names)
self.assertNotIn("Shipsanity: Radioactive Ore", location_names)
self.assertNotIn("Shipsanity: Radioactive Bar", location_names)
self.assertIn("Shipsanity: Banana", location_names)
self.assertIn("Shipsanity: Mango", location_names)
self.assertIn("Shipsanity: Pineapple", location_names)
self.assertIn("Shipsanity: Taro Root", location_names)
self.assertIn("Shipsanity: Ginger", location_names)
self.assertIn("Shipsanity: Magma Cap", location_names)
class TestShipsanityFullShipmentWithFish(SVTestBase):
options = {
Shipsanity.internal_name: Shipsanity.option_full_shipment_with_fish,
SpecialOrderLocations.internal_name: SpecialOrderLocations.option_board_qi
}
def test_only_full_shipment_and_fish_shipsanity_locations(self):
for location in self.get_real_locations():
if LocationTags.SHIPSANITY in location_table[location.name].tags:
with self.subTest(location.name):
self.assertTrue(LocationTags.SHIPSANITY_FULL_SHIPMENT in location_table[location.name].tags or
LocationTags.SHIPSANITY_FISH in location_table[location.name].tags)
def test_include_island_items_shipsanity_locations(self):
location_names = [location.name for location in self.multiworld.get_locations(self.player)]
self.assertIn("Shipsanity: Cinder Shard", location_names)
self.assertIn("Shipsanity: Bone Fragment", location_names)
self.assertIn("Shipsanity: Radioactive Ore", location_names)
self.assertIn("Shipsanity: Radioactive Bar", location_names)
self.assertIn("Shipsanity: Banana", location_names)
self.assertIn("Shipsanity: Mango", location_names)
self.assertIn("Shipsanity: Pineapple", location_names)
self.assertIn("Shipsanity: Taro Root", location_names)
self.assertIn("Shipsanity: Ginger", location_names)
self.assertIn("Shipsanity: Magma Cap", location_names)
self.assertIn("Shipsanity: Blue Discus", location_names)
self.assertIn("Shipsanity: Lionfish", location_names)
self.assertIn("Shipsanity: Stingray", location_names)
self.assertIn("Shipsanity: Glacierfish Jr.", location_names)
self.assertIn("Shipsanity: Legend II", location_names)
self.assertIn("Shipsanity: Ms. Angler", location_names)
self.assertIn("Shipsanity: Radioactive Carp", location_names)
self.assertIn("Shipsanity: Son of Crimsonfish", location_names)
class TestShipsanityFullShipmentWithFishExcludeIsland(SVTestBase):
options = {
Shipsanity.internal_name: Shipsanity.option_full_shipment_with_fish,
ExcludeGingerIsland.internal_name: ExcludeGingerIsland.option_true
}
def test_only_full_shipment_and_fish_shipsanity_locations(self):
for location in self.get_real_locations():
if LocationTags.SHIPSANITY in location_table[location.name].tags:
with self.subTest(location.name):
self.assertTrue(LocationTags.SHIPSANITY_FULL_SHIPMENT in location_table[location.name].tags or
LocationTags.SHIPSANITY_FISH in location_table[location.name].tags)
def test_exclude_island_items_shipsanity_locations(self):
location_names = [location.name for location in self.multiworld.get_locations(self.player)]
self.assertNotIn("Shipsanity: Cinder Shard", location_names)
self.assertNotIn("Shipsanity: Radioactive Ore", location_names)
self.assertNotIn("Shipsanity: Radioactive Bar", location_names)
self.assertNotIn("Shipsanity: Banana", location_names)
self.assertNotIn("Shipsanity: Mango", location_names)
self.assertNotIn("Shipsanity: Pineapple", location_names)
self.assertNotIn("Shipsanity: Taro Root", location_names)
self.assertNotIn("Shipsanity: Ginger", location_names)
self.assertNotIn("Shipsanity: Magma Cap", location_names)
self.assertNotIn("Shipsanity: Blue Discus", location_names)
self.assertNotIn("Shipsanity: Lionfish", location_names)
self.assertNotIn("Shipsanity: Stingray", location_names)
self.assertNotIn("Shipsanity: Glacierfish Jr.", location_names)
self.assertNotIn("Shipsanity: Legend II", location_names)
self.assertNotIn("Shipsanity: Ms. Angler", location_names)
self.assertNotIn("Shipsanity: Radioactive Carp", location_names)
self.assertNotIn("Shipsanity: Son of Crimsonfish", location_names)
class TestShipsanityFullShipmentWithFishExcludeQiBoard(SVTestBase):
options = {
Shipsanity.internal_name: Shipsanity.option_full_shipment_with_fish,
SpecialOrderLocations.internal_name: SpecialOrderLocations.option_board_only
}
def test_only_full_shipment_and_fish_shipsanity_locations(self):
for location in self.get_real_locations():
if LocationTags.SHIPSANITY in location_table[location.name].tags:
with self.subTest(location.name):
self.assertTrue(LocationTags.SHIPSANITY_FULL_SHIPMENT in location_table[location.name].tags or
LocationTags.SHIPSANITY_FISH in location_table[location.name].tags)
def test_exclude_qi_board_items_shipsanity_locations(self):
location_names = [location.name for location in self.multiworld.get_locations(self.player)]
self.assertIn("Shipsanity: Cinder Shard", location_names)
self.assertIn("Shipsanity: Bone Fragment", location_names)
self.assertNotIn("Shipsanity: Radioactive Ore", location_names)
self.assertNotIn("Shipsanity: Radioactive Bar", location_names)
self.assertIn("Shipsanity: Banana", location_names)
self.assertIn("Shipsanity: Mango", location_names)
self.assertIn("Shipsanity: Pineapple", location_names)
self.assertIn("Shipsanity: Taro Root", location_names)
self.assertIn("Shipsanity: Ginger", location_names)
self.assertIn("Shipsanity: Magma Cap", location_names)
self.assertIn("Shipsanity: Blue Discus", location_names)
self.assertIn("Shipsanity: Lionfish", location_names)
self.assertIn("Shipsanity: Stingray", location_names)
self.assertNotIn("Shipsanity: Glacierfish Jr.", location_names)
self.assertNotIn("Shipsanity: Legend II", location_names)
self.assertNotIn("Shipsanity: Ms. Angler", location_names)
self.assertNotIn("Shipsanity: Radioactive Carp", location_names)
self.assertNotIn("Shipsanity: Son of Crimsonfish", location_names)