def METHOD_NAME(self):
    output = preprocess(pd.DataFrame(columns=keep_cols), "state")
    expected = pd.DataFrame(columns=new_keep_cols)
    assert_frame_equal(output, expected, check_dtype = False)