Working on the get_data unittest