def test_continente_str():
canada = Pais('Canada', 'CAN', [(2016, 31995000), (2015, 31676000)])
usa = Pais('United States', 'USA', [(2016, 31995001), (2015, 31675000)])
mexico = Pais('Mexico', 'MEX', [(2016, 127540423), (2015, 112336538)])
paises = [canada, usa, mexico]
america_norte = Continente('América del Norte', 'AN', paises)
assert str(
america_norte) == 'América del Norte (Canada, Mexico, United States)'
def test_continente_poblacion():
canada = Pais('Canada', 'CAN', [(2016, 31995000), (2015, 31676000)])
usa = Pais('United States', 'USA', [(2016, 31995001), (2015, 31675000)])
mexico = Pais('Mexico', 'MEX', [(2016, 127540423), (2015, 112336538)])
paises = [canada, usa, mexico]
america_norte = Continente('América del Norte', 'AN', paises)
assert america_norte.poblacion() == 191530424
assert america_norte.poblacion(2015) == 175687538
def eliminar_pais_controller(self):
"""Controlador para eliminar un país"""
listado = self.traer_paises()
id = self.vista.eliminar_pais(listado)
id = int(id)
pais = Pais()
pais.id = id
pais.delete()
self.vista.confirmar_eliminar_pais()
self.pais_controller()
def make_dict(paises_prueba):
""" Crea un diccionaro de los continentes en los que
se encuentran los paises de la lista ``paises prueba``.
:param paises_prueba: lista con los nombres de los
paises a buscar en los archivos .json
:return dict_cont: diccionario con los contientes y
las respectivas clases ``Continente``
"""
# Cargamos los archivos json donde se obtendrán la población y códigos
with open('population_json.json') as f2:
population_json = json.load(f2)
with open('country-and-continent-codes-list_json.json') as f3:
country_continent_json = json.load(f3)
dict_cont = {} # Diccionario
# Obtenemos la información para cada país
for pd in country_continent_json:
name = pd['Country_Name']
cont = pd['Continent_Name']
if name in paises_prueba:
actual = (Pais(name, pd['Three_Letter_Country_Code'], []))
for pj in population_json:
if pj['Country Code'] == actual.codigo:
actual.asignapoblacion(pj['Year'], int(pj['Value']))
if cont in dict_cont.keys():
dict_cont[cont].paises.append(actual)
else:
dict_cont[cont] = Continente(cont, pd['Continent_Code'],
[actual])
return dict_cont
def editar_pais_controller(self):
"""Controlador para editar un país"""
listado = self.traer_paises()
(id, nombre, abbr) = self.vista.editar_pais(listado)
pais = Pais()
pais.id = int(id)
pais.pais = nombre
pais.abbr = abbr
pais.update()
self.vista.confirmar_editar_pais()
self.pais_controller()
def partida_guardada(self):
'''
Como es una partida ya creada no llama al generador de conexiones aereas.
Llama al metodo que carga los datos de las conexiones en los grafos correspondientes
Lee el archivo 'partida_guardada.csv' y crea objetos de la clase Pais con cada linea del archivo.
'''
self.cargar_conexiones()
self.mundo.lista_paises = ListaLigada()
for datos_pais in importar_datos('partida_guardada.csv'):
if datos_pais[0] == 'Pais':
pass
else:
nuevo_pais = Pais(*datos_pais)
self.mundo.lista_paises.append(nuevo_pais)
def crear_pais_controller(self):
"""Controlador para creación de nuevo país"""
(pais_nombre, pais_apellido) = self.vista.crear_pais()
pais = Pais()
pais.nombre = pais_nombre
pais.apellido = pais_apellido
pais.create()
self.vista.confirmar_creacion()
self.pais_controller()
def partida_nueva(self):
'''
LLama a la funcion que genera las conexiones aereas.
Llama al metodo que carga los datos de las conexiones en los grafos correspondientes
Lee el archivo 'population.csv' y crea objetos de la clase Pais con cada linea del archivo.
'''
generate_connections()
self.cargar_conexiones()
self.mundo.lista_paises = ListaLigada()
for datos_pais in importar_datos('population.csv'):
if datos_pais[0] == 'Pais':
pass
else:
nuevo_pais = Pais(*datos_pais)
self.mundo.lista_paises.append(nuevo_pais)
def crear_pais_controller(self):
"""Controlador para creación de nuevo país"""
if verifica.inicio():
(pais_nombre, pais_abbr) = self.vista.crear_pais()
pais = Pais()
pais.pais = pais_nombre
pais.abbr = pais_abbr
pais.create()
self.vista.confirmar_creacion()
self.pais_controller()
def it_add_a_pais_object(self):
pais = Pais("brasil","brasilia", "34032",[])
pais2 = Pais("Argentina", "argentina", "2323",[])
pais.inserirPais(pais2)
(pais2 in pais.listapaises) |should| equal_to(True)
def test_pais_str():
canada = Pais('Canada', 'CAN', [(2016, 31995000), (2015, 31676000)])
assert str(canada) == 'Canada tiene una población de 31995000 habitantes'
def it_vetores_a_pais_object(self):
pais = Pais("brasil","brasilia", "34032",[])
pais2 = Pais("Argentina", "argentina", "2323",[])
pais3 = Pais("Uruguai", "conclave", "2334",[])
pais4 = Pais("Chile", "chile", "22333",[])
pais.inserirPais(pais3)
pais.inserirPais(pais4)
pais2.inserirPais(pais3)
pais2.inserirPais(pais4)
pais.comparar_pais(pais2)
def traer_paises(self):
"""Trae una lista de todos los países"""
pais = Pais()
listado = pais.read_all()
return listado
def it_creates_a_pais_object(self):
pais = Pais("brasil","brasilia", "34032",[])
pais.nome |should| equal_to("brasil")
pais.capital |should| equal_to("brasilia")
pais.dimensao |should| equal_to("34032")
pais.listapaises |should| equal_to([])
def it_verificar_paises_iguais_object(self):
pais = Pais("brasil","brasilia", "34032",[])
pais2 = Pais("Argentina", "argentina", "2323",[])
pais3 = Pais("Uruguai", "conclave", "2334",[])
pais4 = Pais("Chile", "chile", "22333",[])
pais5 = Pais("Chile", "chile", "22333",[])
pais.inserirPais(pais3)
pais.inserirPais(pais4)
pais2.inserirPais(pais5)
pais2.inserirPais(pais4)
lista = []
lista.append(pais4)
pais.verificar_paises_iguais(pais2) |should| equal_to(lista)
def test_pais_poblacion():
canada = Pais('Canada', 'CAN', [(2016, 31995000), (2015, 31676000)])
assert canada.poblacion(2016) == 31995000
assert canada.poblacion(2015) == 31676000
assert canada.poblacion() == 31995000
def test_pais_asignaploblacion():
canada = Pais('Canada', 'CAN', [(2016, 31995000), (2015, 31676000)])
canada.asignapoblacion(2017, 32312000)
assert canada.poblacion(2017) == 32312000
assert canada.poblacion(2016) == 31995000
assert canada.poblacion() == 32312000
def test_pais_es_maspoblado():
canada = Pais('Canada', 'CAN', [(2016, 31995000), (2015, 31676000)])
usa = Pais('United States', 'USA', [(2016, 31995001), (2015, 31675000)])
assert canada.es_masgrande(usa, 2015) is True
assert canada.es_masgrande(usa) is False
def it_comparar_a_pais_object(self):
pais = Pais("brasil","brasilia", "34032",[])
pais2 = Pais("brasil", "brasilia","340323",[])
pais.comparar_pais(pais2) |should| equal_to(True)
def test_pais():
canada = Pais('Canada', 'CAN', [(2016, 31995000), (2015, 31676000)])
assert canada.nombre == 'Canada'
assert canada.codigo == 'CAN'