[FEATURE]Add more tests