def test_cant_meet_guest_requirements(self):
luke_black = Relationship(['Luke', 'Black'], preset_couple_stats['Luke,Black']['stats'])
kota_onsen = Relationship(['Kota', 'Onsen'], preset_couple_stats['Kota,Onsen']['stats'])
rel_list = RelationshipList()
rel_list.add_relationship(luke_black)
rel_list.add_relationship(kota_onsen)
current_maxes = {'Luke,Black': 2, 'Kota,Onsen': 2}
available_guests = ['Luke', 'Black', 'Onsen']
choices, couples = rel_list.get_available_couple_routes(available_guests, available_guests, current_maxes)
self.assertEqual(choices, [("Luke and Black's scene #0", 'Luke_Black_0')])
self.assertEqual(couples, [('Luke', 'Black')])
def test_create_relationship_list(self):
luke_black = Relationship(['Luke', 'Black'], stats=preset_couple_stats['Luke,Black']['stats'])
kota_onsen = Relationship(['Kota', 'Onsen'], stats=preset_couple_stats['Kota,Onsen']['stats'])
self.assertEqual(luke_black.status, 'unset')
rel_list = RelationshipList()
rel_list.add_relationship(luke_black)
self.assertEqual(rel_list.get_relationship_status('Luke,Black'), 'unset')
self.assertEqual(rel_list.get_relationship_status('Kota,Onsen'), 'missing')
rel_list.add_relationship(kota_onsen)
self.assertEqual(rel_list.get_relationship_status('Luke,Black'), 'unset')
self.assertEqual(rel_list.get_relationship_status('Kota,Onsen'), 'unset')
def test_start_relationship_modifies_guest_list(self):
luke = Guest('Luke', preset_guest_stats['Luke'])
black = Guest('Black', preset_guest_stats['Black'])
luke_black = Relationship(['Luke', 'Black'], stats=preset_couple_stats['Luke,Black']['stats'])
guest_list = GuestList()
guest_list.add_guest(luke)
guest_list.add_guest(black)
rel_list = RelationshipList()
rel_list.add_relationship(luke_black)
self.assertEqual(rel_list.get_relationship_progress('Luke'), 0)
self.assertEqual(guest_list.guests['Luke'].romance_status, 'single')
rel_list.advance_relationship(['Luke', 'Black'], guest_list=guest_list)
self.assertEqual(rel_list.get_relationship_progress('Luke'), 1)
self.assertEqual(guest_list.guests['Luke'].romance_status, 'dating')
self.assertEqual(guest_list.guests['Luke'].romance_partner, 'Black')
self.assertEqual(rel_list.get_relationship_status('Luke,Black'), 'started')
def test_stopped_by_current_maxes(self):
luke_black = Relationship(['Luke', 'Black'], preset_couple_stats['Luke,Black']['stats'])
kota_onsen = Relationship(['Kota', 'Onsen'], preset_couple_stats['Kota,Onsen']['stats'])
rel_list = RelationshipList()
rel_list.add_relationship(luke_black)
rel_list.add_relationship(kota_onsen)
rel_list.advance_relationship('Luke,Black', max_progress=3)
current_maxes = {'Luke,Black': 1, 'Kota,Onsen': 2}
available_guests = ['Luke', 'Black', 'Kota', 'Onsen']
choices, couples = rel_list.get_available_couple_routes(available_guests, available_guests, current_maxes)
self.assertEqual(choices, [("Kota and Onsen's scene #0", 'Kota_Onsen_0')])
self.assertEqual(couples, [('Kota', 'Onsen')])
def test_get_team_relationship_bonus(self):
dict_copy = preset_couple_stats['Luke,Black']
route_max = preset_maximum_relationship_length['Luke,Black']
luke_black = Relationship(['Luke', 'Black'], preset_couple_stats['Luke,Black']['stats'])
rel_list = RelationshipList()
rel_list.add_relationship(luke_black)
self.assertEqual(rel_list.sum_bonus().get_value('Tech'), 0)
self.assertEqual(rel_list.sum_bonus().get_value('Tech'), 0)
self.assertEqual(rel_list.get_relationships_with_guests('Luke,Black,Kota,Onsen').sum_bonus().get_value('Tech'), 0)
self.assertEqual(rel_list.get_relationship_status('Luke,Black'), 'unset')
# test again after completing the relationship
rel_list.advance_relationship('Luke,Black', amount=route_max, max_progress=route_max)
self.assertEqual(rel_list.get_relationship_status('Luke,Black'), 'complete')
self.assertEqual(rel_list.relationships[0].get_bonus_stats().get_value('Tech'), 1)
self.assertEqual(rel_list.sum_bonus(), Stats(preset_couple_stats['Luke,Black']['stats']))
self.assertEqual(rel_list.get_relationships_with_guests('Luke,Black,Kota,Onsen').sum_bonus().get_value('Tech'),
preset_couple_stats['Luke,Black']['stats']['Tech'])
# make sure dict is not being modified
self.assertEqual(dict_copy['stats']['Tech'], preset_couple_stats['Luke,Black']['stats']['Tech'])
def test_start_relationship_in_list(self):
luke_black = Relationship(['Luke', 'Black'], preset_couple_stats['Luke,Black']['stats'])
kota_onsen = Relationship(['Kota', 'Onsen'], preset_couple_stats['Kota,Onsen']['stats'])
rel_list = RelationshipList()
self.assertEqual(len(rel_list.get_relationships_that_can_start('Luke,Black,Onsen,Kota')), 0)
rel_list.add_relationship(luke_black)
rel_list.add_relationship(kota_onsen)
self.assertEqual(rel_list.get_relationship_status('Luke,Black'), 'unset')
self.assertEqual(len(rel_list.get_relationships_that_can_start('Luke,Black,Onsen,Kota')), 2)
self.assertEqual(rel_list.get_relationship_status('Onsen,Kota'), 'unset')
rel_list.advance_relationship('Luke,Onsen')
self.assertEqual(len(rel_list.get_relationships_that_can_start('Luke,Black,Onsen,Kota')), 2)
rel_list.advance_relationship('Kota,Onsen')
self.assertEqual(rel_list.get_relationship_status('Onsen,Kota'), 'started')
self.assertEqual(len(rel_list.get_relationships_that_can_start('Luke,Black,Onsen,Kota')), 1)
def test_advance_relationship(self):
luke_black = Relationship(['Luke', 'Black'], preset_couple_stats['Luke,Black']['stats'])
kota_onsen = Relationship(['Kota', 'Onsen'], preset_couple_stats['Kota,Onsen']['stats'])
rel_list = RelationshipList()
rel_list.add_relationship(luke_black)
rel_list.add_relationship(kota_onsen)
current_maxes = {'Luke,Black': 2, 'Kota,Onsen': 2}
available_guests = ['Luke', 'Black', 'Onsen', 'Kota']
rel_list.advance_relationship('Kota,Onsen')
choices, couples = rel_list.get_available_couple_routes(available_guests, available_guests, current_maxes)
self.assertEqual(choices, [("Luke and Black's scene #0", 'Luke_Black_0'), ("Kota and Onsen's scene #1", 'Kota_Onsen_1')])
self.assertEqual(couples, [('Luke', 'Black'), ('Kota', 'Onsen')])
rel_list.advance_relationship('Kota,Onsen', max_progress=2)
choices, couples = rel_list.get_available_couple_routes(available_guests, available_guests, current_maxes)
self.assertEqual(rel_list.get_relationship_status('Kota,Onsen'), 'complete')
self.assertEqual(choices, [("Luke and Black's scene #0", 'Luke_Black_0')])
self.assertEqual(couples, [('Luke', 'Black')])
from model.guest import GuestList
from model.inventory import Inventory, FileList
from model.couple import RelationshipList
from model.reward import RewardList
from model.daily_event import DailyEventList
from model.wardrobe import Wardrobe
player_background = ""
current_chapter = ""
guests = GuestList()
inventory = Inventory()
relationships = RelationshipList()
rewards = RewardList()
daily_events = DailyEventList()
wardrobe = Wardrobe()
UI_permissions = {'ledger': False,
'items': False,
'guests': False,
'rd': False,
'exploration': False,
'fragments': False,
'files': False,
'file_profile': False,
'file_mementos': False,
'file_tech': False,
'file_contracts': False,
'file_artifacts': False}
# permanent variables