def test_GivenMissingValueDic(self): df = pd.DataFrame({"a": [1, 2, 3], "b": [1, np.NaN, 4]}) MissingValues.add_miss_val_indicator_from_dic(df, miss_dic={"a": 1}) self.assertTrue((df.a_is_nan.values == [1, 0, 0]).all())
def test_GivenMissingValueDic(self): df = pd.DataFrame({'a': [1, 2, 3], 'b': [1, np.NaN, 4]}) MissingValues.add_miss_val_indicator_from_dic(df, miss_dic={'a': 1}) self.assertTrue((df.a_is_nan.values == [1, 0, 0]).all())