Add test suite and first unit tests