Fix shapes of test files
Merge pull request #4 from ChrisBeaumont/master

Give stokes axis a single channel for test files