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))