tests for plot_data_in_grid