def setUpClass(cls):
dic_datos = {
'lugar': [708, 708, 7],
'fecha': ['2000', '2002', '2000'],
'var_completo': [1, 2, 2],
'var_incompleto': [np.nan, 10, 20],
'temp': [1, 1, 1]
}
cls.bds = {
'pandas': fnt.FuentePandas(pd.DataFrame(dic_datos), 'Pandas'),
'csv': fnt.FuenteCSV(arch_datos1, 'CSV'),
'dic': fnt.FuenteDic(dic_datos, 'Dic'),
'xarray': fnt.FuenteBaseXarray(xr.Dataset(
{
'var_completo': ('n', dic_datos['var_completo']),
'var_incompleto': ('n', dic_datos['var_incompleto']),
'fecha': ('n', dic_datos['fecha']),
'lugar': ('n', dic_datos['lugar'])
}
), 'Xarray', fechas='fecha', lugares='lugar'
)
}
def test_lugares_texto(símismo):
fuente = fnt.FuentePandas(pd.DataFrame({'a': [1, 2]}), 'prueba', lugares='708')
lugares_fuente = fuente.obt_vals('a')[_('lugar')].values.tolist()
símismo.assertEqual(lugares_fuente, ['708', '708'])
def test_lugares_columna(símismo):
lugar = ['1', '2']
fuente = fnt.FuentePandas(pd.DataFrame({'a': [1, 2], 'l': lugar}), 'prueba', lugares='l')
lugares_fuente = fuente.obt_vals('a').lugar.values.tolist()
símismo.assertListEqual(lugares_fuente, lugar)
def test_fechas_error(símismo):
with símismo.assertRaises(ValueError):
fnt.FuentePandas(pd.DataFrame({'a': [1, 2]}), 'prueba', fechas='No soy una columna').obt_vals('a')
def test_fecha_texto(símismo):
fecha = '1966-7-21'
fuente = fnt.FuentePandas(pd.DataFrame({'a': [1, 2]}), 'prueba', fechas=fecha)
lista_fechas_igual(fuente.obt_vals('a')[_('fecha')].values, [fecha] * 2)
def test_fechas_ft(símismo):
fecha = ft.date(1966, 1, 1)
fuente = fnt.FuentePandas(pd.DataFrame({'a': [1, 2]}), 'prueba', fechas=fecha)
lista_fechas_igual(fuente.obt_vals('a')[_('fecha')].values, [fecha] * 2)
def test_fechas_texto_año(símismo):
fuente = fnt.FuentePandas(pd.DataFrame({'a': [1, 2]}), 'prueba', fechas='1966')
lista_fechas_igual(fuente.obt_vals('a')[_('fecha')].values, [ft.date(1966, 1, 1)] * 2)
def test_fechas_columna(símismo):
fechas = ['1947-8-14', '1947-8-15']
fuente = fnt.FuentePandas(pd.DataFrame({'a': [1, 2], 'f': fechas}), 'prueba', fechas='f')
lista_fechas_igual(fuente.obt_vals('a')[_('fecha')].values, fechas)