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 8af1570..d63934d 100644 (file)
--- a/test/test_contribuinte.py
+++ b/test/test_contribuinte.py
@@ -11,6 +11,8 @@ 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)
@@ -20,6 +22,9 @@ class TestContribuinte(unittest.TestCase):
         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