Add tests for the shuffle function. Currently other tests are broken, but this can be used for testing shuffle in the meantime.