test/test_contribuinte.py

   1 # -*- mode: python; encoding: utf-8; -*-
   2 import unittest
   3 import sys
   4 import os
   5 import dirs
   6 dirs.default_irpf_dir = dirs.IRPFTestDir()
   7
   8 from contribuinte import Contribuinte
   9
  10 class TestContribuinte(unittest.TestCase):
  11
  12     def setUp(self):
  13         self.cpf = '34278161174'
  14         self.cpf_normalizado = '342.781.611-74'
  15
  16         self.cpf_invalido = '34278161175'
  17
  18         self.contribuinte = Contribuinte(self.cpf)
  19
  20     def test_ValidarCPF(self):
  21         msg = 'Validação de CPF falhou!'
  22         self.assertTrue(self.contribuinte._validate_cpf(self.cpf), msg)
  23         self.assertFalse(self.contribuinte._validate_cpf(self.cpf_invalido), msg)
  24
  25     def test_NormalizeCPF(self):
  26         msg = 'Normalização falhou'
  27         self.assertEqual(self.contribuinte._normalize_cpf(self.cpf),
  28                 self.cpf_normalizado, msg)
  29
  30 # vim:tabstop=4:expandtab:smartindent:fenc=utf8