def test_fechas_columna(símismo):
fechas = ['1947-8-14', '1947-8-15']
bd = MicroDatos('prueba',
pd.DataFrame({
'a': [1, 2],
'f': fechas
}),
tiempo='f')
lista_fechas_igual(bd.obt_datos('a')['tiempo'].values, fechas)
def test_lugares_columna(símismo):
lugar = ['1', '2']
bd = MicroDatos('prueba',
pd.DataFrame({
'a': [1, 2],
'l': lugar
}),
lugar='l')
lugares_bd = bd.obt_datos('a').lugar.values.tolist()
símismo.assertListEqual(lugares_bd, lugar)
def test_obt_datos_código_vacío(símismo):
bd = MicroDatos('prueba',
pd.DataFrame({'a': [1, 2, 'jaja']}),
cód_vacío='jaja')
npt.assert_array_equal(bd.obt_datos('a')['a'], [1, 2, np.nan])
def test_lugares_texto(símismo):
bd = MicroDatos('prueba', pd.DataFrame({'a': [1, 2]}), lugar='708')
lugares_bd = bd.obt_datos('a').lugar.values.tolist()
símismo.assertListEqual(lugares_bd, ['708'] * 2)
def test_fecha_texto(símismo):
fecha = '1966-7-21'
bd = MicroDatos('prueba', pd.DataFrame({'a': [1, 2]}), tiempo=fecha)
lista_fechas_igual(bd.obt_datos('a')['tiempo'].values, [fecha] * 2)
def test_fechas_ft(símismo):
fecha = ft.date(1966, 1, 1)
bd = MicroDatos('prueba', pd.DataFrame({'a': [1, 2]}), tiempo=fecha)
lista_fechas_igual(bd.obt_datos('a')['tiempo'].values, [fecha] * 2)
def test_fechas_texto_año(símismo):
bd = MicroDatos('prueba', pd.DataFrame({'a': [1, 2]}), tiempo='1966')
lista_fechas_igual(
bd.obt_datos('a')['tiempo'].values, [ft.date(1966, 1, 1)] * 2)