keep unit test close to function