Test of the augment_data function in dictization_functions.py