update the test.sh for unit test