Refine the unit tests.