def test_save_data_to_csv(self):
omip_data = OmipData(get_file_path("omip_data_1.csv"))
filepath = get_file_path("omip_data_exportado_1.csv")
remove_file(filepath)
omip_data.save_data_to_csv(filepath=filepath)
csv_data = get_file(filepath)
self.assertEqual(len(csv_data.read_lines()), 15)
def test_update_data(self):
# Initial data
omip_data = OmipData(get_file_path("omip_data_vacio1.csv"))
ini = dt.datetime(2017, 10, 4) # 04/10/2017
fin = dt.datetime.now() # Ahora
values_antes = omip_data.get_values(date_ini=ini, date_fin=fin)
# Update data
omip_data.update() # Update todos los productos
values_despues = omip_data.get_values(date_ini=ini, date_fin=fin)
# Check data
self.assertEqual(len(values_despues) > len(values_antes), True)
# Save data updated
filepath = get_file_path("omip_data_exportado_1.csv")
remove_file(filepath)
omip_data.save_data_to_csv(filepath=filepath)
def test_using_omip_to_download_data(self):
bd_file = "omip_data_1.csv" # tiene datos del 01/10/2017 a 02/10/2017
omip_data = OmipData(get_file_path(bd_file))
print_stats("BD cargada desde archivo \"{}\"".format(bd_file),
omip_data)
fecha_fin = omip_data.get_stats("fecha_fin")
fecha_fin = fecha_fin + dt.timedelta(days=1) # 03/10/2017
today = dt.datetime.now()
#omip_data.update(date_ini = fecha_fin, date_fin = today)
omip_data.update() # Actualiza todos los productos
formater_fecha = FormaterFecha(formato="aaaa-mm-dd")
bd_file_to_save = "omip_data_" + formater_fecha.formatea(
today) + ".csv"
filepath = get_file_path(bd_file_to_save)
remove_file(filepath)
omip_data.save_data_to_csv(filepath=filepath)
print_stats(
"BD actualizada desde servidor y datos guardados en archivo \"{}\""
.format(bd_file_to_save), omip_data)
def test_using_omip_to_download_data_de_year(self):
bd_file = "omip_data_vacio3.csv" # archivo sin datos, sólo cabecera "fecha;producto;valor"
omip_data = OmipData(get_file_path(bd_file))
print_stats("BD cargada desde archivo \"{}\"".format(bd_file),
omip_data)
#fecha_fin = omip_data.get_stats()["fecha_fin"]
#fecha_fin = fecha_fin + dt.timedelta(days=0) # 02/01/2018
today = dt.datetime.now()
#omip_data.update(date_ini = fecha_fin, date_fin = today)
omip_data.update() # Actualiza todos los productos
#omip_data.update_multithreading() # Actualiza todos los productos
formater_fecha = FormaterFecha(formato="aaaa-mm-dd")
bd_file_to_save = "omip_data_" + formater_fecha.formatea(
today) + ".csv"
filepath = get_file_path(bd_file_to_save)
remove_file(filepath)
omip_data.save_data_to_csv(filepath=filepath)
print_stats(
"BD actualizada desde servidor y datos guardados en archivo \"{}\""
.format(bd_file_to_save), omip_data)