This look good, welcome to dipy. I think you want to use vstack instead of hstack here, that should fix the tests.