IW: Add some basic unit tests