test/test_municipios.py

   1 # -*- mode: python; encoding: utf-8; -*-
   2 import unittest
   3 import sys
   4 import os
   5
   6 myPath = os.path.dirname(os.path.abspath(__file__))
   7 sys.path.insert(0, myPath + '/../src')
   8
   9 from municipios import Municipios
  10
  11 class TestMunicipios(unittest.TestCase):
  12
  13     def setUp(self):
  14         self.municipios = Municipios()
  15
  16         self.municipios.carregar_estado('MG')
  17         self.belo_horizonte = self.municipios.get_municipio('4123')
  18
  19         self.municipios.carregar_estado('SP')
  20         self.sao_paulo = self.municipios.get_municipio('7107')
  21
  22     def test_VerifyCEP(self):
  23         self.assertTrue(self.municipios.verify_cep(self.belo_horizonte, '31230120'))
  24         self.assertFalse(self.municipios.verify_cep(self.belo_horizonte, '05020000'))
  25
  26         self.assertTrue(self.municipios.verify_cep(self.sao_paulo, '05020000'))
  27         self.assertFalse(self.municipios.verify_cep(self.sao_paulo, '31230120'))
  28
  29
  30 if __name__ == '__main__':
  31     unittest.main()
  32
  33 # vim:tabstop=4:expandtab:smartindent:fenc=utf8