Add tests to make sure Imputation is working as expected

[schae234]

I think there are beagle test files that do this.