Adding a test for verifying the shape of imputed matrices