COSMIT: PEP8 in test_imputation