83 lines
3.4 KiB
Python
83 lines
3.4 KiB
Python
|
from ...locations import LocationTags, location_table
|
||
|
|
from ...options import BuildingProgression, Shipsanity
|
||
|
|
from ...test import SVTestBase
|
||
|
|
|
||
|
|
|
||
|
|
class TestShipsanityNone(SVTestBase):
|
||
|
|
options = {
|
||
|
|
Shipsanity.internal_name: Shipsanity.option_none
|
||
|
|
}
|
||
|
|
|
||
|
|
def test_no_shipsanity_locations(self):
|
||
|
|
for location in self.get_real_locations():
|
||
|
|
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
|
||
|
|
}
|
||
|
|
|
||
|
|
def test_only_crop_shipsanity_locations(self):
|
||
|
|
for location in self.get_real_locations():
|
||
|
|
if LocationTags.SHIPSANITY in location_table[location.name].tags:
|
||
|
|
self.assertIn(LocationTags.SHIPSANITY_CROP, location_table[location.name].tags)
|
||
|
|
|
||
|
|
|
||
|
|
class TestShipsanityFish(SVTestBase):
|
||
|
|
options = {
|
||
|
|
Shipsanity.internal_name: Shipsanity.option_fish
|
||
|
|
}
|
||
|
|
|
||
|
|
def test_only_fish_shipsanity_locations(self):
|
||
|
|
for location in self.get_real_locations():
|
||
|
|
if LocationTags.SHIPSANITY in location_table[location.name].tags:
|
||
|
|
self.assertIn(LocationTags.SHIPSANITY_FISH, location_table[location.name].tags)
|
||
|
|
|
||
|
|
|
||
|
|
class TestShipsanityFullShipment(SVTestBase):
|
||
|
|
options = {
|
||
|
|
Shipsanity.internal_name: Shipsanity.option_full_shipment
|
||
|
|
}
|
||
|
|
|
||
|
|
def test_only_full_shipment_shipsanity_locations(self):
|
||
|
|
for location in self.get_real_locations():
|
||
|
|
if LocationTags.SHIPSANITY in location_table[location.name].tags:
|
||
|
|
self.assertIn(LocationTags.SHIPSANITY_FULL_SHIPMENT, location_table[location.name].tags)
|
||
|
|
self.assertNotIn(LocationTags.SHIPSANITY_FISH, location_table[location.name].tags)
|
||
|
|
|
||
|
|
|
||
|
|
class TestShipsanityFullShipmentWithFish(SVTestBase):
|
||
|
|
options = {
|
||
|
|
Shipsanity.internal_name: Shipsanity.option_full_shipment_with_fish
|
||
|
|
}
|
||
|
|
|
||
|
|
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:
|
||
|
|
self.assertTrue(LocationTags.SHIPSANITY_FULL_SHIPMENT in location_table[location.name].tags or
|
||
|
|
LocationTags.SHIPSANITY_FISH in location_table[location.name].tags)
|
||
|
|
|
||
|
|
|
||
|
|
class TestShipsanityEverything(SVTestBase):
|
||
|
|
options = {
|
||
|
|
Shipsanity.internal_name: Shipsanity.option_everything,
|
||
|
|
BuildingProgression.internal_name: BuildingProgression.option_progressive
|
||
|
|
}
|
||
|
|
|
||
|
|
def test_all_shipsanity_locations_require_shipping_bin(self):
|
||
|
|
bin_name = "Shipping Bin"
|
||
|
|
self.collect_all_except(bin_name)
|
||
|
|
shipsanity_locations = [location for location in self.get_real_locations() if
|
||
|
|
LocationTags.SHIPSANITY in location_table[location.name].tags]
|
||
|
|
bin_item = self.create_item(bin_name)
|
||
|
|
for location in shipsanity_locations:
|
||
|
|
with self.subTest(location.name):
|
||
|
|
self.remove(bin_item)
|
||
|
|
self.assertFalse(self.world.logic.region.can_reach_location(location.name)(self.multiworld.state))
|
||
|
|
self.multiworld.state.collect(bin_item, event=False)
|
||
|
|
shipsanity_rule = self.world.logic.region.can_reach_location(location.name)
|
||
|
|
self.assert_rule_true(shipsanity_rule, self.multiworld.state)
|
||
|
|
self.remove(bin_item)
|