def test_raise_error_si_get_values_con_fecha_superior_a_dia_hoy(self): ini = dt.datetime(2017, 10, 2) # 01/10/2017 manana = dt.datetime.today() + dt.timedelta(days=1) with self.assertRaises(ValueError): omip_data = OmipData(get_file_path("omip_data_1.csv")) omip_data.get_values(ini, manana)
def test_load_datos_iniciales_vacio(self): omip_data = OmipData(get_file_path("omip_data_vacio1.csv")) values = omip_data.get_values() self.assertEqual(len(values), 0) omip_data = OmipData(get_file_path("omip_data_vacio2.csv")) values = omip_data.get_values() self.assertEqual(len(values), 0)
def test_raise_error_si_get_values_con_fechas_fin_menor_que_fecha_ini( self): ini = dt.datetime(2017, 9, 28) # 28/09/2017 fin = dt.datetime(2017, 9, 1) # 01/09/2017 with self.assertRaises(ValueError): omip_data = OmipData(get_file_path("omip_data_1.csv")) omip_data.get_values(date_ini=ini, date_fin=fin)
def test_get_values_si_no_hay_datos_para_productos_solicitados(self): product_list = ("Q4-19") ini = dt.datetime(2017, 10, 2) # 01/10/2017 fin = dt.datetime(2017, 10, 4) # 03/10/2017 omip_data = OmipData(get_file_path("omip_data_1_reducido.csv")) values = omip_data.get_values(product_list, ini, fin) self.assertEqual(len(values), 0) # No hay datos para el producto solicitado
def test_get_values_si_no_hay_datos(self): product_list = ("Q1-18", "Q2-18", "Q3-18", "Q4-18") ini = dt.datetime(2017, 10, 2) # 01/10/2017 fin = dt.datetime(2017, 10, 4) # 03/10/2017 omip_data = OmipData(get_file_path("omip_data_vacio1.csv")) values_db = omip_data.get_values(product_list, ini, fin) self.assertEqual(values_db.items(), ()) # No hay datos para el producto solicitado
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_get_values_sin_indicar_fecha_muestra_todos_los_values(self): omip_data = OmipData(get_file_path("omip_data_1_reducido.csv")) values = omip_data.get_values() self.assertEqual(len(values), 5) #print("values: ", values) self.assertEqual(values[dt.datetime(2017, 10, 1), "Q1-18"], 10.44) self.assertEqual(values[dt.datetime(2017, 10, 1), "Q2-18"], 20.44) self.assertEqual(values[dt.datetime(2017, 10, 2), "Q1-18"], 10.55) self.assertEqual(values[dt.datetime(2017, 10, 2), "Q2-18"], 20.55) self.assertEqual(values[dt.datetime(2017, 10, 3), "Q1-18"], 11.43)
def test_get_values_entre_dos_fechas(self): omip_data = OmipData(get_file_path("omip_data_1_reducido.csv")) ini = dt.datetime(2017, 10, 1) # 01/10/2017 fin = dt.datetime(2017, 10, 2) # 02/10/2017 values = omip_data.get_values(date_ini=ini, date_fin=fin) self.assertEqual(len(values), 4) self.assertEqual(values[dt.datetime(2017, 10, 1), "Q1-18"], 10.44) self.assertEqual(values[dt.datetime(2017, 10, 1), "Q2-18"], 20.44) self.assertEqual(values[dt.datetime(2017, 10, 2), "Q1-18"], 10.55) self.assertEqual(values[dt.datetime(2017, 10, 2), "Q2-18"], 20.55)
def test_get_values_entre_dos_fechas_for_product_list(self): products = ("Q1-18", "Q2-18") ini = dt.datetime(2017, 10, 1) # 01/10/2017 fin = dt.datetime(2017, 10, 3) # 03/10/2017 omip_data = OmipData(get_file_path("omip_data_4_reducido.csv")) values = omip_data.get_values(products, ini, fin) self.assertEqual(len(values), 6) self.assertEqual(values[dt.datetime(2017, 10, 1), "Q1-18"], 10.44) self.assertEqual(values[dt.datetime(2017, 10, 1), "Q2-18"], 20.44) self.assertEqual(values[dt.datetime(2017, 10, 2), "Q1-18"], 10.55) self.assertEqual(values[dt.datetime(2017, 10, 2), "Q2-18"], 20.55) self.assertEqual(values[dt.datetime(2017, 10, 3), "Q1-18"], 53.80) self.assertEqual(values[dt.datetime(2017, 10, 3), "Q2-18"], 44.05)
def test_load_datos_iniciales_from_file(self): omip_data = OmipData(get_file_path("omip_data_1.csv")) values = omip_data.get_values() self.assertEqual(len(values), 14)