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)
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'
while odp != '2':
if odp != '1':
print('Please enter 1 or 2')
odp = input()
continue
