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())
Example #2
0
 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())
Example #4
0
 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())