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