Adjusted test_motion.py for 3D.

One test case still fails suggesting a bug in HMM motion correction.