Add more test case for each function