class CidadeTest(unittest.TestCase): cidade_sp = { 'sigla_uf_nome_cidade': u'SP_SAO PAULO', 'area_km2': '1099', } def setUp(self): self.db = MongoDb() self.db.insert_or_update_cidade(self.cidade_sp) def tearDown(self): self.db._db.ufs.remove() def test_get(self): result = self.db.get_one_cidade(u'SP', u'São Paulo') for key, expected in self.cidade_sp.items(): self.assertEqual(expected, result[key]) def test_get_alt(self): result = self.db.get_one_cidade(u'SP', u'Outro lugar (São Paulo)') self.assertEqual('1099', result['area_km2']) def test_update(self): self.db.insert_or_update_cidade({ 'sigla_uf_nome_cidade': u'SP_SAO PAULO', 'area_km2': '2000', }) result = self.db.get_one_cidade(u'SP', u'São Paulo') self.assertEqual('2000', result['area_km2'])