class TestHouse(unittest.TestCase):
def setUp(self):
self.house = House()
self.sim1 = self.house.move_in_sim('sim1', 20, 'w')
self.sim2 = self.house.move_in_sim('sim2', 45, 'm')
def test_move_in(self):
self.house.move_in_sim('sim3', 4, 'w')
self.assertEqual(self.house.family['sim3'].name, 'sim3')
self.assertEqual(self.house.family['sim3'].age, 4)
self.assertEqual(self.house.family['sim3'].sex, 'w')
self.assertEqual(len(self.house.family), 3)
def test_add_bathroom(self):
self.house.add_bathroom()
self.assertEqual(len(self.house.bathrooms), 2)
def test_add_bedroom(self):
self.house.add_bedroom(1)
self.assertEqual(len(self.house.bedrooms), 2)
def test_run(self):
self.house.run(3)
for s in self.house.family.values():
for v in s.needs.values():
self.assertEqual(v, 7)
self.assertEqual(self.house.time, 3)
odp = '1'
while odp != '2':
if odp != '1':
print('Please enter 1 or 2')
odp = input()
continue
print('Please name the person:')
name = input()
print('Enter the persons age:')
age = input()
while not age.isdigit():
print('Please enter a number:')
age = input()
print('Enter the person\'s gender:')
sex = input()
house.move_in_sim(name, age, sex)
print('Do you want to add another family member?\n1. yes\n2. no')
odp = input()
print('Time to play!')
print(
'What do you want to do now?\n1. Check stats\n2. Choose person to play\n3. Run the game\n4. Add new family member'
'\n5. Add rooms\n6. Check activity queues\n7. Exit the game')
odp = input()
while odp != '7':
if odp not in ('1', '2', '3', '4', '5', '6'):
print('Please enter number from the list:')
odp = input()
continue
if odp == '1':
house.show_needs()
class Test(unittest.TestCase):
def setUp(self):
self.house = House()
self.house.move_in_sim('sim1', 15, 'm')
self.house.move_in_sim('sim2', 60, 'w')
self.sim1 = self.house.family['sim1']
self.sim2 = self.house.family['sim2']
def test_empty_queue(self):
self.sim1.sleep()
self.sim1.toilet()
self.assertNotEqual(self.sim1.activity_queue, [])
self.sim1.empty_queue()
self.assertEqual(self.sim1.activity_queue, [])
def test_needs_down(self):
for need in self.sim1.needs.values():
self.assertEqual(need, 10)
self.sim1.needs_down()
for need in self.sim1.needs.values():
self.assertEqual(need, 9)
def test_pet(self):
self.sim1.get_pet('Lara')
self.assertEqual(self.sim1.house.pets['Lara'], self.sim1.pet)
self.assertEqual(self.sim1.pet.name, 'Lara')
self.sim1.feed_pet(self.sim1.pet)
self.assertEqual(self.sim1.activity_queue, [('feed', 'Lara')])
self.sim1.pet.hunger = 7
self.sim1.pet_feed('Lara')
self.assertEqual(self.sim1.pet.hunger, 10)
def test_shower(self):
self.sim1.shower()
self.assertEqual(self.sim1.activity_queue, ['shower'])
self.sim1.use_shower()
self.assertEqual(self.house.bathrooms[0].queue,
[(self.sim1, 'shower')])
self.house.add_bathroom()
self.sim2.use_shower()
self.assertEqual(self.house.bathrooms[1].queue,
[(self.sim2, 'shower')])
def test_talk(self):
self.sim1.needs['social'] = 6
self.sim2.needs['social'] = 6
self.sim1.interact_with(self.sim2.name)
self.assertEqual(self.sim1.activity_queue, [('talk', self.sim2.name)])
self.sim1.talk(self.sim2.name)
self.assertEqual(
(self.sim1.needs['social'], self.sim2.needs['social']), (10, 8))
def test_check_queue(self):
self.sim1.rest()
self.sim1.eat_snack()
self.sim1.get_pet('Lara')
self.sim1.pet.hunger = 7
self.sim1.feed_pet(self.sim1.pet)
self.sim1.check_queue()
self.assertEqual(self.house.livingroom.couch_queue, [self.sim1])
self.sim1.check_queue()
self.assertEqual(self.house.kitchen.fridge_queue, [self.sim1])
self.sim1.check_queue()
self.assertEqual(self.sim1.pet.hunger, 10)
