How could I cover the tests in the case of a function that returns a dataframe that can change at any time? I am starting tests and I have no idea how to cover my code with unit tests, what do you advise me to do?
def update_with_formulario(date):
df_formulario = _get_formulario(date)
assert df_formulario is not None, 'Dataframe df_formulario is null'
df_csv = utilities.get_data_csv_with_pandas('df.csv')
assert df_csv is not None, 'Dataframe df_csv is null'
df = df_formulario.merge(df_csv, 'left')
utilities.write_csv_dataframe(f"{ROOT_WRITE_PATH}df.csv", df)
def _get_formulario(date):
global cnn
try:
repo = Repository()
cnn = repo.connect()
query = f"""SELECT DISTINCT fecha,
hora ,
coalesce(valor_copia, valor) AS value
FROM table
WHERE VARIABLE IN ('variable')
AND fecha_data = '{date} 00:00:00'"""
df_formulario = pd.read_sql_query(query, cnn)
return df_formulario
except:
logging.exception(exc_msg.EXC_PROCEDURE_EXCEPT_MSG)
finally:
if cnn:
cnn.close()
logging.info(exc_msg.CNN_CLOSED_MSG)
