Nuke the axis=0 section.

Also, perhaps add short comments at the start of each test to describe the purpose of the test.