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)