def test_impute_two(self):
index = ["bla", "blue", "blo"]
values = [np.nan, np.nan, 0.4]
values_expected = [0.4, 0.4, 0.4]
impute_expected = [1, 1, 0]
expected = self.make_expected(index, values_expected, impute_expected)
actual = util.mean_impute_series(self.make_input(index, values))
assert_frame_equal(expected, actual)
def test_impute_two(self):
index = ["bla", "blue", "blo"]
values = [np.nan, np.nan, 0.4]
values_expected = [0.4, 0.4, 0.4]
impute_expected = [1, 1, 0]
expected = self.make_expected(index, values_expected, impute_expected)
actual = util.mean_impute_series(self.make_input(index, values))
assert_frame_equal(expected, actual)
def test_all_missing(self):
index = ["bla", "blue", "blo"]
values = [np.nan, np.nan, np.nan]
util.mean_impute_series(self.make_input(index, values))
def test_all_missing(self):
index = ["bla", "blue", "blo"]
values = [np.nan, np.nan, np.nan]
util.mean_impute_series(self.make_input(index, values))
