add simple unit tests