def test_fill_mean_object(self): df = pd.DataFrame({"a": [1, 2, 3], "b": ["1", np.NaN, 4]}) MissingValues.fill_with_mean(df) self.assertTrue(np.isnan(df.b.values[1]).all())
def test_fill_mean_object(self): df = pd.DataFrame({'a': [1, 2, 3], 'b': ['1', np.NaN, 4]}) MissingValues.fill_with_mean(df) self.assertTrue(np.isnan(df.b.values[1]).all())
def test_fill_mean(self): df = pd.DataFrame({"a": [1, 2, 3], "b": [1, np.NaN, 4]}) MissingValues.fill_with_mean(df) self.assertTrue((df.b.values == [1, 2.5, 4]).all())
def test_fill_mean(self): df = pd.DataFrame({'a': [1, 2, 3], 'b': [1, np.NaN, 4]}) MissingValues.fill_with_mean(df) self.assertTrue((df.b.values == [1, 2.5, 4]).all())