Updated axes_grid test for more parameters
Fixed axes_grid cax based on cbar_edge