Make unit tests more configurable