Some improvements on unit test framework.