Archipelago/worlds/stardew_valley/test/TestItems.py

141 lines
6.9 KiB
Python

import sys
import random
import sys
from BaseClasses import MultiWorld, get_seed
from . import setup_solo_multiworld, SVTestCase, allsanity_options_without_mods, get_minsanity_options
from .. import StardewValleyWorld
from ..items import Group, item_table
from ..options import Friendsanity, SeasonRandomization, Museumsanity, Shipsanity, Goal
from ..strings.wallet_item_names import Wallet
all_seasons = ["Spring", "Summer", "Fall", "Winter"]
all_farms = ["Standard Farm", "Riverland Farm", "Forest Farm", "Hill-top Farm", "Wilderness Farm", "Four Corners Farm", "Beach Farm"]
class TestItems(SVTestCase):
def test_can_create_item_of_resource_pack(self):
item_name = "Resource Pack: 500 Money"
multi_world = MultiWorld(1)
multi_world.game[1] = "Stardew Valley"
multi_world.player_name = {1: "Tester"}
world = StardewValleyWorld(multi_world, 1)
item = world.create_item(item_name)
assert item.name == item_name
def test_items_table_footprint_is_between_717000_and_737000(self):
item_with_lowest_id = min((item for item in item_table.values() if item.code is not None), key=lambda x: x.code)
item_with_highest_id = max((item for item in item_table.values() if item.code is not None),
key=lambda x: x.code)
assert item_with_lowest_id.code >= 717000
assert item_with_highest_id.code < 737000
def test_babies_come_in_all_shapes_and_sizes(self):
baby_permutations = set()
options = {Friendsanity.internal_name: Friendsanity.option_bachelors}
for attempt_number in range(50):
if len(baby_permutations) >= 4:
print(f"Already got all 4 baby permutations, breaking early [{attempt_number} generations]")
break
seed = get_seed()
multiworld = setup_solo_multiworld(options, seed=seed, _cache={})
baby_items = [item for item in multiworld.get_items() if "Baby" in item.name]
self.assertEqual(len(baby_items), 2)
baby_permutations.add(f"{baby_items[0]} - {baby_items[1]}")
self.assertEqual(len(baby_permutations), 4)
def test_correct_number_of_stardrops(self):
allsanity_options = allsanity_options_without_mods()
multiworld = setup_solo_multiworld(allsanity_options)
stardrop_items = [item for item in multiworld.get_items() if "Stardrop" in item.name]
self.assertEqual(len(stardrop_items), 7)
def test_no_duplicate_rings(self):
allsanity_options = allsanity_options_without_mods()
multiworld = setup_solo_multiworld(allsanity_options)
ring_items = [item.name for item in multiworld.get_items() if Group.RING in item_table[item.name].groups]
self.assertEqual(len(ring_items), len(set(ring_items)))
def test_can_start_in_any_season(self):
starting_seasons_rolled = set()
options = {SeasonRandomization.internal_name: SeasonRandomization.option_randomized}
for attempt_number in range(50):
if len(starting_seasons_rolled) >= 4:
print(f"Already got all 4 starting seasons, breaking early [{attempt_number} generations]")
break
seed = get_seed()
multiworld = setup_solo_multiworld(options, seed=seed, _cache={})
starting_season_items = [item for item in multiworld.precollected_items[1] if item.name in all_seasons]
season_items = [item for item in multiworld.get_items() if item.name in all_seasons]
self.assertEqual(len(starting_season_items), 1)
self.assertEqual(len(season_items), 3)
starting_seasons_rolled.add(f"{starting_season_items[0]}")
self.assertEqual(len(starting_seasons_rolled), 4)
def test_can_start_on_any_farm(self):
starting_farms_rolled = set()
for attempt_number in range(60):
if len(starting_farms_rolled) >= 7:
print(f"Already got all 7 farm types, breaking early [{attempt_number} generations]")
break
seed = random.randrange(sys.maxsize)
multiworld = setup_solo_multiworld(seed=seed, _cache={})
starting_farm = multiworld.worlds[1].fill_slot_data()["farm_type"]
starting_farms_rolled.add(starting_farm)
self.assertEqual(len(starting_farms_rolled), 7)
class TestMetalDetectors(SVTestCase):
def test_minsanity_1_metal_detector(self):
options = get_minsanity_options()
multiworld = setup_solo_multiworld(options)
items = [item.name for item in multiworld.get_items() if item.name == Wallet.metal_detector]
self.assertEquals(len(items), 1)
def test_museumsanity_2_metal_detector(self):
options = get_minsanity_options().copy()
options[Museumsanity.internal_name] = Museumsanity.option_all
multiworld = setup_solo_multiworld(options)
items = [item.name for item in multiworld.get_items() if item.name == Wallet.metal_detector]
self.assertEquals(len(items), 2)
def test_shipsanity_full_shipment_1_metal_detector(self):
options = get_minsanity_options().copy()
options[Shipsanity.internal_name] = Shipsanity.option_full_shipment
multiworld = setup_solo_multiworld(options)
items = [item.name for item in multiworld.get_items() if item.name == Wallet.metal_detector]
self.assertEquals(len(items), 1)
def test_shipsanity_everything_2_metal_detector(self):
options = get_minsanity_options().copy()
options[Shipsanity.internal_name] = Shipsanity.option_everything
multiworld = setup_solo_multiworld(options)
items = [item.name for item in multiworld.get_items() if item.name == Wallet.metal_detector]
self.assertEquals(len(items), 2)
def test_complete_collection_2_metal_detector(self):
options = get_minsanity_options().copy()
options[Goal.internal_name] = Goal.option_complete_collection
multiworld = setup_solo_multiworld(options)
items = [item.name for item in multiworld.get_items() if item.name == Wallet.metal_detector]
self.assertEquals(len(items), 2)
def test_perfection_2_metal_detector(self):
options = get_minsanity_options().copy()
options[Goal.internal_name] = Goal.option_perfection
multiworld = setup_solo_multiworld(options)
items = [item.name for item in multiworld.get_items() if item.name == Wallet.metal_detector]
self.assertEquals(len(items), 2)
def test_maxsanity_4_metal_detector(self):
options = get_minsanity_options().copy()
options[Museumsanity.internal_name] = Museumsanity.option_all
options[Shipsanity.internal_name] = Shipsanity.option_everything
options[Goal.internal_name] = Goal.option_perfection
multiworld = setup_solo_multiworld(options)
items = [item.name for item in multiworld.get_items() if item.name == Wallet.metal_detector]
self.assertEquals(len(items), 4)