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)