def test_raise_error_si_get_items_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): db = OmipDB() db.get_items(date_ini = ini , date_fin = fin)
def test_get_items_entre_dos_fechas(self): db = OmipDB() db.save_item(dt.datetime(2017, 10, 1), "Q1-18", 10.44) db.save_item(dt.datetime(2017, 10, 1), "Q2-18", 20.44) db.save_item(dt.datetime(2017, 10, 2), "Q1-18", 10.55) db.save_item(dt.datetime(2017, 10, 2), "Q2-18", 20.55) db.save_item(dt.datetime(2017, 10, 3), "Q1-18", 11.43) ini = dt.datetime(2017, 10, 1) # 01/10/2017 fin = dt.datetime(2017, 10, 2) # 02/10/2017 values = db.get_items(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_items_sin_indicar_fecha_muestra_todos_los_values(self): db = OmipDB() db.save_item(dt.datetime(2017, 10, 1), "Q1-18", 10.44) db.save_item(dt.datetime(2017, 10, 1), "Q2-18", 20.44) db.save_item(dt.datetime(2017, 10, 2), "Q1-18", 10.55) db.save_item(dt.datetime(2017, 10, 2), "Q2-18", 20.55) db.save_item(dt.datetime(2017, 10, 3), "Q1-18", 11.43) values = db.get_items() 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)