Testes para abstração de datas

test_data_abstraction.py

import unittest

class TestData(unittest.TestCase):

	def test_data_definida_deve_retornar_valores_validos(self):
		data = Data(11,10,2016)

		self.assertEqual(data.dia,11)
		self.assertEqual(data.mes,10)
		self.assertEqual(data.ano,2016)

	def test_data_definida_deve_retornar_valor_formatado_builtin(self):
		data = Data(10,10,2015)

		formatado = data.nativo()

		self.assertEqual(formatado.year,2015)
		self.assertEqual(formatado.month,10)
		self.assertEqual(formatado.day,10)

	def test_data_criada_sem_parametros_deve_retornar_data_atual(self):

		hoje = Data()

		from datetime import date

		today = date.today()

		self.assertEqual(today.day,hoje.dia)
		self.assertEqual(today.month,hoje.mes)
		self.assertEqual(today.year,hoje.ano)

	def test_comparacao_de_datas_deve_retornar_verdadeiro(self):
		data1 = Data(10,10,2016).nativo()
		data2 = Data(10,11,2016).nativo()

		self.assertTrue(data2 > data1)

if __name__ == "__main__":
	unittest.main(verbosity=2)