Added ImageGrid test
Pep8 fixes in axes_grid test