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)
from House import House
house = House()
print("Hello!\nType in your house address:")
house.address = str(input())
print(
'For now you have one bathroom, one bedroom, one kitchen and one living room. Do you want to add some more rooms?'
'\n1. Bathroom\n2. Bedroom\n3. no')
odp = input()
while odp != '3':
if odp == '1':
house.add_bathroom()
print('Your house now have {} bathrooms.'.format(
(len(house.bathrooms))))
elif odp == '2':
print('How big bed do you want? For 1 person or 2?')
n = input()
while n not in ('1', '2'):
print('Please enter 1 or 2:')
n = input()
house.add_bedroom(int(n))
print('Your house now have {} bedrooms.'.format((len(house.bedrooms))))
else:
print('Please, type in 1, 2 or 3.')
odp = input()
continue
print('Do you want to add any more room?\n1. Bathroom\n2. Bedroom\n3. no')
odp = input()
print('Now it\'s time to add family members!')
odp = '1'
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)
