Update unit test for swigged shuffle function.

@oxtopus please review