def test_fill_missing_with_feature_means_feature_all_empty(self): dataset = DataSet([[2, np.NaN, np.NaN], [7, np.NaN, 10], [5, np.NaN, np.NaN]]) dataset.fill_missing_with_feature_means() assert_that(dataset, equals_dataset([[2, 0, 10], [7, 0, 10], [5, 0, 10]]))
def test_fill_missing_with_feature_means(self): dataset = DataSet([[2, np.NaN, np.NaN], [np.NaN, 6, 10], [5, 4, np.NaN]]) dataset.fill_missing_with_feature_means() assert_that(dataset, equals_dataset([[2, 5, 10], [3.5, 6, 10], [5, 4, 10]]))