modify unit test of get_id function