[IMP]:Improve test cases