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)
sim.toilet()
elif odp2 == '10':
sim.shower()
elif odp2 == '11':
sim.sleep()
elif odp2 == '12':
sim.rest()
elif odp2 == '13':
sim.watch_tv()
elif odp == '3':
print('How many steps do you want to run the game for?')
n = input()
while not n.isdigit():
print('Please enter a number:')
n = input()
house.run(int(n))
house.show_needs()
for sim in house.family.values():
print('{} current activity: {}'.format(sim.name,
sim.current_activity))
elif odp == '4':
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)
