def test_aniversario_hoje(self):
dt1 = date.today() - timedelta(days=4)
hoje = date.today()
pessoa1 = Colaborador('Birth', hoje.day, hoje.month, 2000)
pessoa2 = Colaborador('not Birth', dt1.day, dt1.month, 2000)
assert pessoa1.aniversario_hoje() is True
assert pessoa2.aniversario_hoje() is False
class Departamento:
def __init__(self, nome_setor, responsavel={}):
self._nome_setor = nome_setor
self._responsavel = None
if responsavel:
self.informar_responsavel(responsavel['nome'], responsavel['dia'],
responsavel['mes'], responsavel['ano'])
self._colaboradores = []
@property
def nome(self):
return self._nome_setor
@nome.setter
def nome(self, value):
self._nome_setor = value
@property
def responsavel(self):
if self._responsavel is None:
return None
return str(self._responsavel)
@property
def colaboradores(self):
return self._colaboradores
def informar_responsavel(self, nome, dia, mes, ano):
self._responsavel = Colaborador(nome, dia, mes, ano)
def add_colaborador(self, nome, dia, mes, ano):
self._colaboradores.append(Colaborador(nome, dia, mes, ano))
def verificar_aniversariantes(self):
lista = []
for colaborador in self.colaboradores:
if colaborador.aniversario_hoje():
lista.append(
(colaborador.nome, colaborador.aniversario, self.nome))
if self.responsavel:
if self._responsavel.aniversario_hoje():
lista.append((self._responsavel.nome,
self._responsavel.aniversario, self.nome))
return lista
def __str__(self):
return self._nome_setor
def __repr__(self):
return 'Departamento = ' + self._nome_setor
