update unit tests to put setup inside it block