test_preprocess.py 文件源码

def test_filter_on_flag_column_nothing_left():
    bad_df = pd.DataFrame({'spkitemid': ['a1', 'b1', 'c1', 'd1'],
                           'sc1': [1, 2, 1, 3],
                           'feature': [2, 3, 4, 5],
                           'flag1': [1, 0, 20, 14],
                           'flag2': [1, 1.0, 'TD', '03']})

    flag_dict = {'flag1': [1, 0, 14], 'flag2': ['TD']}

    df_new, df_excluded = filter_on_flag_columns(bad_df, flag_dict)
    import ipdb
    ipdb.set_trace()