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)