def test_impute_no_missing_values(self): """ After imputation, no change should occur""" imputed = complete_case(self.data_c) self.assertTrue(np.array_equal(imputed, self.data_c))
def test_impute_missing_values(self): """ After imputation, no NaN's should exist""" imputed = complete_case(self.data_m) self.assertTrue(np.shape(imputed) == (4, 5))
def test_return_type(self): """ Check return type, should return an np.ndarray""" imputed = complete_case(self.data_m) self.assertTrue(isinstance(imputed, np.ndarray))