96d48a923a
* Core: renamed `CollectionState.collect` arg from `event` to `prevent_sweep` and remove forced collection * Update TestDungeon.py --------- Co-authored-by: NewSoupVi <57900059+NewSoupVi@users.noreply.github.com>
59 lines
3.4 KiB
Python
59 lines
3.4 KiB
Python
from ...options import SeasonRandomization, Friendsanity, FriendsanityHeartSize
|
|
from ...test import SVTestBase
|
|
|
|
|
|
class TestFriendsanityDatingRules(SVTestBase):
|
|
options = {
|
|
SeasonRandomization.internal_name: SeasonRandomization.option_randomized_not_winter,
|
|
Friendsanity.internal_name: Friendsanity.option_all_with_marriage,
|
|
FriendsanityHeartSize.internal_name: 3
|
|
}
|
|
|
|
def test_earning_dating_heart_requires_dating(self):
|
|
self.collect_all_the_money()
|
|
self.multiworld.state.collect(self.create_item("Fall"), prevent_sweep=False)
|
|
self.multiworld.state.collect(self.create_item("Beach Bridge"), prevent_sweep=False)
|
|
self.multiworld.state.collect(self.create_item("Progressive House"), prevent_sweep=False)
|
|
for i in range(3):
|
|
self.multiworld.state.collect(self.create_item("Progressive Pickaxe"), prevent_sweep=False)
|
|
self.multiworld.state.collect(self.create_item("Progressive Weapon"), prevent_sweep=False)
|
|
self.multiworld.state.collect(self.create_item("Progressive Axe"), prevent_sweep=False)
|
|
self.multiworld.state.collect(self.create_item("Progressive Barn"), prevent_sweep=False)
|
|
for i in range(10):
|
|
self.multiworld.state.collect(self.create_item("Foraging Level"), prevent_sweep=False)
|
|
self.multiworld.state.collect(self.create_item("Farming Level"), prevent_sweep=False)
|
|
self.multiworld.state.collect(self.create_item("Mining Level"), prevent_sweep=False)
|
|
self.multiworld.state.collect(self.create_item("Combat Level"), prevent_sweep=False)
|
|
self.multiworld.state.collect(self.create_item("Progressive Mine Elevator"), prevent_sweep=False)
|
|
self.multiworld.state.collect(self.create_item("Progressive Mine Elevator"), prevent_sweep=False)
|
|
|
|
npc = "Abigail"
|
|
heart_name = f"{npc} <3"
|
|
step = 3
|
|
|
|
self.assert_can_reach_heart_up_to(npc, 3, step)
|
|
self.multiworld.state.collect(self.create_item(heart_name), prevent_sweep=False)
|
|
self.assert_can_reach_heart_up_to(npc, 6, step)
|
|
self.multiworld.state.collect(self.create_item(heart_name), prevent_sweep=False)
|
|
self.assert_can_reach_heart_up_to(npc, 8, step)
|
|
self.multiworld.state.collect(self.create_item(heart_name), prevent_sweep=False)
|
|
self.assert_can_reach_heart_up_to(npc, 10, step)
|
|
self.multiworld.state.collect(self.create_item(heart_name), prevent_sweep=False)
|
|
self.assert_can_reach_heart_up_to(npc, 14, step)
|
|
|
|
def assert_can_reach_heart_up_to(self, npc: str, max_reachable: int, step: int):
|
|
prefix = "Friendsanity: "
|
|
suffix = " <3"
|
|
for i in range(1, max_reachable + 1):
|
|
if i % step != 0 and i != 14:
|
|
continue
|
|
location = f"{prefix}{npc} {i}{suffix}"
|
|
can_reach = self.world.logic.region.can_reach_location(location)(self.multiworld.state)
|
|
self.assertTrue(can_reach, f"Should be able to earn relationship up to {i} hearts")
|
|
for i in range(max_reachable + 1, 14 + 1):
|
|
if i % step != 0 and i != 14:
|
|
continue
|
|
location = f"{prefix}{npc} {i}{suffix}"
|
|
can_reach = self.world.logic.region.can_reach_location(location)(self.multiworld.state)
|
|
self.assertFalse(can_reach, f"Should not be able to earn relationship up to {i} hearts")
|