def test_impute_missing_values(self): """ After imputation, no NaN's should exist""" imputed = impy.buck_iterative(self.data_m) self.assertFalse(np.isnan(imputed).any())
def test_return_type(self): """ Check return type, should return an np.ndarray""" imputed = impy.buck_iterative(self.data_m) self.assertTrue(isinstance(imputed, np.ndarray))
def test_impute_missing_values(): """ After imputation, no NaN's should exist""" imputed = impy.buck_iterative(data_m) assert not np.isnan(imputed).any()
def test_buck_iter(buck_test_data): imputed = impy.buck_iterative(buck_test_data) return_na_check(imputed)
def test_return_type(): """ Check return type, should return an np.ndarray""" imputed = impy.buck_iterative(data_m) assert isinstance(imputed, np.ndarray)