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