Add unit tests for _validate_plot_axes