unit test optimisation