Test: Normalização de CPF

[cascardo/irpf-gui.git] / test / test_contribuinte.py

diff --git a/test/test_contribuinte.py b/test/test_contribuinte.py
index 6048350..d63934d 100644 (file)
--- a/test/test_contribuinte.py
+++ b/test/test_contribuinte.py
@@ -11,12 +11,20 @@ class TestContribuinte(unittest.TestCase):
 
     def setUp(self):
         self.cpf = '34278161174'
+        self.cpf_normalizado = '342.781.611-74'
+
         self.cpf_invalido = '34278161175'
 
         self.contribuinte = Contribuinte(self.cpf)
 
     def test_ValidarCPF(self):
-        self.assertTrue(self.contribuinte._validate_cpf(self.cpf))
-        self.assertFalse(self.contribuinte._validate_cpf(self.cpf_invalido))
+        msg = 'Validação de CPF falhou!'
+        self.assertTrue(self.contribuinte._validate_cpf(self.cpf), msg)
+        self.assertFalse(self.contribuinte._validate_cpf(self.cpf_invalido), msg)
+
+    def test_NormalizeCPF(self):
+        msg = 'Normalização falhou'
+        self.assertEqual(self.contribuinte._normalize_cpf(self.cpf),
+                self.cpf_normalizado, msg)
 
 # vim:tabstop=4:expandtab:smartindent:fenc=utf8