Revise input test cases after unifying the interface of robust multinomial