def test_obter_sala(self):
sala.criar_sala(11,30)
sala.criar_sala(3,58)
s =sala.obter_sala(11)
self.assertEqual(11,s[0])
def menu_obter():
cod_sala = int(input('Obter sala pelo código de sala: '))
o = sala.obter_sala(cod_sala)
if o == None:
print('Sala não encontrada! ')
else:
imprimir_sala(o)
def teste_alterar_sala(self):
sala.criar_sala(11,30)
sala.criar_sala(3,58)
s = sala.obter_sala(11)
salas = sala.listar_salas()
s[1] = 60
self.assertEqual(11,s[0])
self.assertEqual(60,s[1])
def menu_alterar():
print('\n Alterar sala')
cod_sala = int(input('Cod. Sala: '))
s = sala.listar_salas()
if (s == False):
print("Sala não encontrada")
else:
capacidade = int(input('Digite a nova capacidade da sala: '))
s = sala.obter_sala(cod_sala)
s[1] = capacidade
sala.alterar_sala(cod_sala, capacidade)
return capacidade
def test_excluir_sala(self):
sala.criar_sala(11,30)
sala.criar_sala(3,58)
sala.excluir_sala(3)
s = sala.obter_sala(3)
self.assertIsNone(s)